| // Copyright 2016 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 <map> |
| #include <memory> |
| #include <string> |
| #include <vector> |
| |
| #include "base/test/task_environment.h" |
| #include "base/version.h" |
| #include "components/prefs/testing_pref_service.h" |
| #include "components/update_client/persisted_data.h" |
| #include "components/update_client/test_activity_data_service.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace update_client { |
| |
| TEST(PersistedDataTest, Simple) { |
| base::test::TaskEnvironment env; |
| auto pref = std::make_unique<TestingPrefServiceSimple>(); |
| PersistedData::RegisterPrefs(pref->registry()); |
| auto metadata = std::make_unique<PersistedData>(pref.get(), nullptr); |
| EXPECT_EQ(-2, metadata->GetDateLastRollCall("someappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastActive("someappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid.withdot")); |
| std::vector<std::string> items; |
| items.push_back("someappid"); |
| items.push_back("someappid.withdot"); |
| test::SetDateLastData(metadata.get(), items, 3383); |
| EXPECT_EQ(3383, metadata->GetDateLastRollCall("someappid")); |
| EXPECT_EQ(3383, metadata->GetDateLastRollCall("someappid.withdot")); |
| EXPECT_EQ(-2, metadata->GetDateLastActive("someappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastActive("someappid.withdot")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastRollCall("someotherappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastActive("someotherappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someotherappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someotherappid")); |
| const std::string pf1 = metadata->GetPingFreshness("someappid"); |
| EXPECT_FALSE(pf1.empty()); |
| test::SetDateLastData(metadata.get(), items, 3386); |
| EXPECT_EQ(3386, metadata->GetDateLastRollCall("someappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastActive("someappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastRollCall("someotherappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastActive("someotherappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someotherappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someotherappid")); |
| const std::string pf2 = metadata->GetPingFreshness("someappid"); |
| EXPECT_FALSE(pf2.empty()); |
| // The following has a 1 / 2^128 chance of being flaky. |
| EXPECT_NE(pf1, pf2); |
| |
| EXPECT_FALSE(metadata->GetProductVersion("someappid").IsValid()); |
| metadata->SetProductVersion("someappid", base::Version("1.0")); |
| EXPECT_EQ(base::Version("1.0"), metadata->GetProductVersion("someappid")); |
| |
| EXPECT_TRUE(metadata->GetFingerprint("someappid").empty()); |
| metadata->SetFingerprint("someappid", "somefingerprint"); |
| EXPECT_STREQ("somefingerprint", |
| metadata->GetFingerprint("someappid").c_str()); |
| } |
| |
| TEST(PersistedDataTest, SharedPref) { |
| base::test::TaskEnvironment env; |
| auto pref = std::make_unique<TestingPrefServiceSimple>(); |
| PersistedData::RegisterPrefs(pref->registry()); |
| auto metadata = std::make_unique<PersistedData>(pref.get(), nullptr); |
| EXPECT_EQ(-2, metadata->GetDateLastRollCall("someappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastActive("someappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid")); |
| std::vector<std::string> items; |
| items.push_back("someappid"); |
| test::SetDateLastData(metadata.get(), items, 3383); |
| |
| // Now, create a new PersistedData reading from the same path, verify |
| // that it loads the value. |
| metadata = std::make_unique<PersistedData>(pref.get(), nullptr); |
| EXPECT_EQ(3383, metadata->GetDateLastRollCall("someappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastActive("someappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastRollCall("someotherappid")); |
| EXPECT_EQ(-2, metadata->GetDateLastActive("someotherappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someotherappid")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someotherappid")); |
| } |
| |
| TEST(PersistedDataTest, SimpleCohort) { |
| base::test::TaskEnvironment env; |
| auto pref = std::make_unique<TestingPrefServiceSimple>(); |
| PersistedData::RegisterPrefs(pref->registry()); |
| auto metadata = std::make_unique<PersistedData>(pref.get(), nullptr); |
| EXPECT_EQ("", metadata->GetCohort("someappid")); |
| EXPECT_EQ("", metadata->GetCohort("someotherappid")); |
| EXPECT_EQ("", metadata->GetCohortHint("someappid")); |
| EXPECT_EQ("", metadata->GetCohortHint("someotherappid")); |
| EXPECT_EQ("", metadata->GetCohortName("someappid")); |
| EXPECT_EQ("", metadata->GetCohortName("someotherappid")); |
| metadata->SetCohort("someappid", "c1"); |
| metadata->SetCohort("someotherappid", "c2"); |
| metadata->SetCohortHint("someappid", "ch1"); |
| metadata->SetCohortHint("someotherappid", "ch2"); |
| metadata->SetCohortName("someappid", "cn1"); |
| metadata->SetCohortName("someotherappid", "cn2"); |
| EXPECT_EQ("c1", metadata->GetCohort("someappid")); |
| EXPECT_EQ("c2", metadata->GetCohort("someotherappid")); |
| EXPECT_EQ("ch1", metadata->GetCohortHint("someappid")); |
| EXPECT_EQ("ch2", metadata->GetCohortHint("someotherappid")); |
| EXPECT_EQ("cn1", metadata->GetCohortName("someappid")); |
| EXPECT_EQ("cn2", metadata->GetCohortName("someotherappid")); |
| metadata->SetCohort("someappid", "oc1"); |
| metadata->SetCohort("someotherappid", "oc2"); |
| metadata->SetCohortHint("someappid", "och1"); |
| metadata->SetCohortHint("someotherappid", "och2"); |
| metadata->SetCohortName("someappid", "ocn1"); |
| metadata->SetCohortName("someotherappid", "ocn2"); |
| EXPECT_EQ("oc1", metadata->GetCohort("someappid")); |
| EXPECT_EQ("oc2", metadata->GetCohort("someotherappid")); |
| EXPECT_EQ("och1", metadata->GetCohortHint("someappid")); |
| EXPECT_EQ("och2", metadata->GetCohortHint("someotherappid")); |
| EXPECT_EQ("ocn1", metadata->GetCohortName("someappid")); |
| EXPECT_EQ("ocn2", metadata->GetCohortName("someotherappid")); |
| } |
| |
| TEST(PersistedDataTest, ActivityData) { |
| base::test::TaskEnvironment env; |
| auto pref = std::make_unique<TestingPrefServiceSimple>(); |
| auto activity_service = std::make_unique<TestActivityDataService>(); |
| PersistedData::RegisterPrefs(pref->registry()); |
| auto metadata = |
| std::make_unique<PersistedData>(pref.get(), activity_service.get()); |
| |
| std::vector<std::string> items({"id1", "id2", "id3"}); |
| |
| for (const auto& item : items) { |
| EXPECT_EQ(-2, metadata->GetDateLastActive(item)); |
| EXPECT_EQ(-2, metadata->GetDateLastRollCall(item)); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive(item)); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall(item)); |
| EXPECT_EQ(false, test::GetActiveBit(metadata.get(), item)); |
| } |
| |
| test::SetDateLastData(metadata.get(), items, 1234); |
| for (const auto& item : items) { |
| EXPECT_EQ(false, test::GetActiveBit(metadata.get(), item)); |
| EXPECT_EQ(-2, metadata->GetDateLastActive(item)); |
| EXPECT_EQ(1234, metadata->GetDateLastRollCall(item)); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive(item)); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall(item)); |
| } |
| |
| activity_service->SetActiveBit("id1", true); |
| activity_service->SetDaysSinceLastActive("id1", 3); |
| activity_service->SetDaysSinceLastRollCall("id1", 2); |
| activity_service->SetDaysSinceLastRollCall("id2", 3); |
| activity_service->SetDaysSinceLastRollCall("id3", 4); |
| EXPECT_EQ(true, test::GetActiveBit(metadata.get(), "id1")); |
| EXPECT_EQ(false, test::GetActiveBit(metadata.get(), "id2")); |
| EXPECT_EQ(false, test::GetActiveBit(metadata.get(), "id2")); |
| EXPECT_EQ(3, metadata->GetDaysSinceLastActive("id1")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("id2")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("id3")); |
| EXPECT_EQ(2, metadata->GetDaysSinceLastRollCall("id1")); |
| EXPECT_EQ(3, metadata->GetDaysSinceLastRollCall("id2")); |
| EXPECT_EQ(4, metadata->GetDaysSinceLastRollCall("id3")); |
| |
| test::SetDateLastData(metadata.get(), items, 5678); |
| activity_service->SetActiveBit("id2", true); |
| test::SetDateLastData(metadata.get(), items, 6789); |
| EXPECT_EQ(false, test::GetActiveBit(metadata.get(), "id1")); |
| EXPECT_EQ(false, test::GetActiveBit(metadata.get(), "id2")); |
| EXPECT_EQ(false, test::GetActiveBit(metadata.get(), "id3")); |
| EXPECT_EQ(3, metadata->GetDaysSinceLastActive("id1")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("id2")); |
| EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("id3")); |
| EXPECT_EQ(2, metadata->GetDaysSinceLastRollCall("id1")); |
| EXPECT_EQ(3, metadata->GetDaysSinceLastRollCall("id2")); |
| EXPECT_EQ(4, metadata->GetDaysSinceLastRollCall("id3")); |
| EXPECT_EQ(5678, metadata->GetDateLastActive("id1")); |
| EXPECT_EQ(6789, metadata->GetDateLastActive("id2")); |
| EXPECT_EQ(-2, metadata->GetDateLastActive("id3")); |
| EXPECT_EQ(6789, metadata->GetDateLastRollCall("id1")); |
| EXPECT_EQ(6789, metadata->GetDateLastRollCall("id2")); |
| EXPECT_EQ(6789, metadata->GetDateLastRollCall("id3")); |
| } |
| |
| } // namespace update_client |