Add a minimum byte count to metrics log serialization

chrome/browser/metrics/metrics_log_serializer_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/base64.h"
 #include "base/md5.h"
 #include "base/values.h"
 #include "chrome/browser/metrics/metrics_log_serializer.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 typedef MetricsLogManager::SerializedLog SerializedLog;
 
 namespace {
 
-const size_t kMaxLocalListSize = 3;
+const size_t kMaxLogByteTotal = 1000;
 
 }  // namespace
 
 class MetricsLogSerializerTest : public ::testing::Test {
 };
 
 // Store and retrieve empty list.
 TEST(MetricsLogSerializerTest, EmptyLogList) {
   ListValue list;
   std::vector<SerializedLog> local_list;
 
-  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLocalListSize,
+  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLogByteTotal,
                                             &list);
   EXPECT_EQ(0U, list.GetSize());
 
   local_list.clear();  // ReadLogsFromPrefList() expects empty |local_list|.
   EXPECT_EQ(
       MetricsLogSerializer::LIST_EMPTY,
       MetricsLogSerializer::ReadLogsFromPrefList(list, true, &local_list));
   EXPECT_EQ(0U, local_list.size());
 }
 
 // Store and retrieve a single log value.
 TEST(MetricsLogSerializerTest, SingleElementLogList) {
   ListValue list;
 
   std::vector<SerializedLog> local_list(1);
   local_list[0].xml = "Hello world!";
 
-  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLocalListSize,
+  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLogByteTotal,
                                             &list);
 
   // |list| will now contain the following:
   // [1, Base64Encode("Hello world!"), MD5("Hello world!")].
   ASSERT_EQ(3U, list.GetSize());
 
   // Examine each element.
   ListValue::const_iterator it = list.begin();
   int size = 0;
   (*it)->GetAsInteger(&size);
@@ skipping 13 matching lines @@
   ++it;
   EXPECT_TRUE(it == list.end());  // Reached end of list.
 
   local_list.clear();
   EXPECT_EQ(
       MetricsLogSerializer::RECALL_SUCCESS,
       MetricsLogSerializer::ReadLogsFromPrefList(list, true, &local_list));
   EXPECT_EQ(1U, local_list.size());
 }
 
-// Store elements greater than the limit.
+// Store a set of logs larger than the limit.
 TEST(MetricsLogSerializerTest, OverLimitLogList) {
   ListValue list;
 
+  // Make log_count logs each slightly larger that TotalByteLimit / log_count
+  // so that the oldest (first) log won't fit.
   std::vector<SerializedLog> local_list(4);
-  local_list[0].proto = "one";
-  local_list[1].proto = "two";
-  local_list[2].proto = "three";
-  local_list[3].proto = "four";
+  size_t log_size = (kMaxLogByteTotal / local_list.size()) + 2;
+  local_list[0].xml = "one";
+  local_list[1].xml = "two";
+  local_list[2].xml = "three";
+  local_list[3].xml = "four";
+  for (size_t i = 0; i < local_list.size(); ++i) {
+    local_list[i].xml.resize(log_size, ' ');
+  }
 
-  std::string expected_first;
-  base::Base64Encode(local_list[local_list.size() - kMaxLocalListSize].proto,
-                     &expected_first);
-  std::string expected_last;
-  base::Base64Encode(local_list[local_list.size() - 1].proto,
-                     &expected_last);
-
-  MetricsLogSerializer::WriteLogsToPrefList(local_list, false,
-                                            kMaxLocalListSize, &list);
-  EXPECT_EQ(kMaxLocalListSize + 2, list.GetSize());
-
-  std::string actual_first;
-  EXPECT_TRUE((*(list.begin() + 1))->GetAsString(&actual_first));
-  EXPECT_EQ(expected_first, actual_first);
-
-  std::string actual_last;
-  EXPECT_TRUE((*(list.end() - 2))->GetAsString(&actual_last));
-  EXPECT_EQ(expected_last, actual_last);
-
-  local_list.clear();
+  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLogByteTotal,
+                                            &list);
+  std::vector<SerializedLog> result_list;
   EXPECT_EQ(
       MetricsLogSerializer::RECALL_SUCCESS,
-      MetricsLogSerializer::ReadLogsFromPrefList(list, true, &local_list));
-  EXPECT_EQ(kMaxLocalListSize, local_list.size());
+      MetricsLogSerializer::ReadLogsFromPrefList(list, true, &result_list));
+  EXPECT_EQ(local_list.size() - 1, result_list.size());
+
+  EXPECT_TRUE(result_list.front().xml.find("two") == 0);
+  EXPECT_TRUE(result_list.back().xml.find("four") == 0);
+}
+
+// Store a log larger than the limit.
+TEST(MetricsLogSerializerTest, OverLimitIndividualLog) {
+  ListValue list;
+
+  std::vector<SerializedLog> local_list(4);
+  local_list[0].xml = "one";
+  local_list[1].xml = "two";
+  local_list[2].xml = "three";
+  local_list[2].xml.resize(kMaxLogByteTotal + 1, ' ');
+  local_list[3].xml = "four";
+
+  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLogByteTotal,
+                                            &list);
+  std::vector<SerializedLog> result_list;
+  EXPECT_EQ(
+      MetricsLogSerializer::RECALL_SUCCESS,
+      MetricsLogSerializer::ReadLogsFromPrefList(list, true, &result_list));
+  EXPECT_EQ(local_list.size() - 1, result_list.size());
+
+  for (size_t i = 0; i < result_list.size(); ++i) {
+    EXPECT_LT(result_list[i].xml.size(), kMaxLogByteTotal);
+  }
+}
+
+// Store a set of logs each of which is larger than the limit.
+TEST(MetricsLogSerializerTest, OverLimitAllLogs) {
+  ListValue list;
+
+  std::vector<SerializedLog> local_list(2);
+  size_t log_size = kMaxLogByteTotal + 1;
+  for (size_t i = 0; i < local_list.size(); ++i) {
+    local_list[i].xml.resize(log_size, ' ');
+  }
+
+  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLogByteTotal,
+                                            &list);
+  std::vector<SerializedLog> result_list;
+  EXPECT_EQ(
+      MetricsLogSerializer::LIST_EMPTY,
+      MetricsLogSerializer::ReadLogsFromPrefList(list, true, &result_list));
+  EXPECT_EQ(0U, result_list.size());
 }
 
 // Induce LIST_SIZE_TOO_SMALL corruption
 TEST(MetricsLogSerializerTest, SmallRecoveredListSize) {
   ListValue list;
 
   std::vector<SerializedLog> local_list(1);
   local_list[0].xml = "Hello world!";
 
-  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLocalListSize,
+  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLogByteTotal,
                                             &list);
   EXPECT_EQ(3U, list.GetSize());
 
   // Remove last element.
   list.Remove(list.GetSize() - 1, NULL);
   EXPECT_EQ(2U, list.GetSize());
 
   local_list.clear();
   EXPECT_EQ(
       MetricsLogSerializer::LIST_SIZE_TOO_SMALL,
       MetricsLogSerializer::ReadLogsFromPrefList(list, true, &local_list));
 }
 
 // Remove size from the stored list.
 TEST(MetricsLogSerializerTest, RemoveSizeFromLogList) {
   ListValue list;
 
   std::vector<SerializedLog> local_list(2);
   local_list[0].xml = "one";
   local_list[1].xml = "two";
   EXPECT_EQ(2U, local_list.size());
-  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLocalListSize,
+  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLogByteTotal,
                                             &list);
   EXPECT_EQ(4U, list.GetSize());
 
   list.Remove(0, NULL);  // Delete size (1st element).
   EXPECT_EQ(3U, list.GetSize());
 
   local_list.clear();
   EXPECT_EQ(
       MetricsLogSerializer::LIST_SIZE_MISSING,
       MetricsLogSerializer::ReadLogsFromPrefList(list, true, &local_list));
 }
 
 // Corrupt size of stored list.
 TEST(MetricsLogSerializerTest, CorruptSizeOfLogList) {
   ListValue list;
 
   std::vector<SerializedLog> local_list(1);
   local_list[0].xml = "Hello world!";
 
-  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLocalListSize,
+  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLogByteTotal,
                                             &list);
   EXPECT_EQ(3U, list.GetSize());
 
   // Change list size from 1 to 2.
   EXPECT_TRUE(list.Set(0, Value::CreateIntegerValue(2)));
   EXPECT_EQ(3U, list.GetSize());
 
   local_list.clear();
   EXPECT_EQ(
       MetricsLogSerializer::LIST_SIZE_CORRUPTION,
       MetricsLogSerializer::ReadLogsFromPrefList(list, true, &local_list));
 }
 
 // Corrupt checksum of stored list.
 TEST(MetricsLogSerializerTest, CorruptChecksumOfLogList) {
   ListValue list;
 
   std::vector<SerializedLog> local_list(1);
   local_list[0].xml = "Hello world!";
 
-  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLocalListSize,
+  MetricsLogSerializer::WriteLogsToPrefList(local_list, true, kMaxLogByteTotal,
                                             &list);
   EXPECT_EQ(3U, list.GetSize());
 
   // Fetch checksum (last element) and change it.
   std::string checksum;
   EXPECT_TRUE((*(list.end() - 1))->GetAsString(&checksum));
   checksum[0] = (checksum[0] == 'a') ? 'b' : 'a';
   EXPECT_TRUE(list.Set(2, Value::CreateStringValue(checksum)));
   EXPECT_EQ(3U, list.GetSize());
 
   local_list.clear();
   EXPECT_EQ(
       MetricsLogSerializer::CHECKSUM_CORRUPTION,
       MetricsLogSerializer::ReadLogsFromPrefList(list, true, &local_list));
 }