Add counts of User data to ProfileInfoCache

chrome/browser/profiles/profile_attributes_storage_unittest.cc

 // Copyright 2015 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/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
+#include "base/values.h"
 #include "chrome/browser/profiles/profile_info_cache.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/supervised_user/supervised_user_constants.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile_manager.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 // The ProfileMetadataEntry accessors aren't just plain old accessors to local
@@ skipping 27 matching lines @@
 void TestAccessors(ProfileAttributesEntry** entry,
     TGetter getter_func,
     TSetter setter_func,
     TValue first_value,
     TValue second_value) {
   (*entry->*setter_func)(first_value);
   EXPECT_EQ(first_value, (*entry->*getter_func)());
   (*entry->*setter_func)(second_value);
   EXPECT_EQ(second_value, (*entry->*getter_func)());
 }
+
+void VerifyStatistics(const ProfileAttributesEntry * const entry,
+                      const std::map<std::string, int>& expect_map,
+                      const std::vector<std::string>& categories_to_check) {
+  // Check GetStatistic().
+  for (const auto& category : categories_to_check) {
+    bool has_category = expect_map.count(category);
+    int actual_value;
+    ASSERT_EQ(has_category, entry->GetStatistic(category, &actual_value));
+    if (has_category)
+      EXPECT_EQ(expect_map.at(category), actual_value);
+  }
+
+  // Check GetAllStatistics()
+  const std::map<std::string, int> actual_map = entry->GetAllStatistics();
+  EXPECT_EQ(expect_map.size(), actual_map.size());
+  for (const auto& item : expect_map) {
+    ASSERT_TRUE(actual_map.count(item.first));
+    ASSERT_EQ(item.second, actual_map.at(item.first));
+  }
+}
 }  // namespace
 
 class ProfileAttributesStorageTest : public testing::Test {
  public:
   ProfileAttributesStorageTest()
       : testing_profile_manager_(TestingBrowserProcess::GetGlobal()) {}
   ~ProfileAttributesStorageTest() override {}
 
  protected:
   void SetUp() override {
@@ skipping 199 matching lines @@
   ASSERT_TRUE(entry->IsSupervised());
   ASSERT_FALSE(entry->IsChild());
   ASSERT_TRUE(entry->IsLegacySupervised());
 
   entry->SetSupervisedUserId(supervised_users::kChildAccountSUID);
   ASSERT_TRUE(entry->IsSupervised());
   ASSERT_TRUE(entry->IsChild());
   ASSERT_FALSE(entry->IsLegacySupervised());
 }
 
+TEST_F(ProfileAttributesStorageTest, StatisticsAccessors) {
+  AddTestingProfile();
+
+  ProfileAttributesEntry* entry;
+  ASSERT_TRUE(storage()->GetProfileAttributesWithPath(
+      GetProfilePath("testing_profile_path0"), &entry));
+
+  EXPECT_EQ(GetProfilePath("testing_profile_path0"), entry->GetPath());
+
+  std::vector<std::string> categories_to_check{"One", "Two", "Three"};
+  std::map<std::string, int> expect_map;
+  std::vector<std::pair<std::string, int>> insertions{
+    std::make_pair("One", 3),  // Insert data
+    std::make_pair("Two", 4),  // Insert data
+    std::make_pair("One", 5)  // Overwrite data
+  };
+
+  // Now no keys are set.
+  VerifyStatistics(entry, expect_map, categories_to_check);
+  // Insert items and test after each insert.
+  for (const auto& item : insertions) {
+    entry->SetStatistic(item.first, item.second);
+    expect_map[item.first] = item.second;
+    VerifyStatistics(entry, expect_map, categories_to_check);
+  }
+}
+
 TEST_F(ProfileAttributesStorageTest, ReSortTriggered) {
   storage()->AddProfile(GetProfilePath("alpha_path"),
                         base::ASCIIToUTF16("alpha"),
                         std::string("alpha_gaia"),
                         base::ASCIIToUTF16("alpha_username"),
                         1,
                         std::string(""));
 
   storage()->AddProfile(GetProfilePath("lima_path"),
                         base::ASCIIToUTF16("lima"),
@@ skipping 64 matching lines @@
   // should be reflected by the ProfileAttributesStorage.
   size_t index = profile_info_cache()->GetIndexOfProfileWithPath(
       GetProfilePath("testing_profile_path0"));
   EXPECT_EQ(base::ASCIIToUTF16("NewName"),
       profile_info_cache()->GetNameOfProfileAtIndex(index));
 
   profile_info_cache()->SetNameOfProfileAtIndex(
       index, base::ASCIIToUTF16("OtherNewName"));
   EXPECT_EQ(base::ASCIIToUTF16("OtherNewName"), first_entry->GetName());
 }