persisted_data_unittest.cc 10.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209
  1. // Copyright 2016 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include <map>
  5. #include <memory>
  6. #include <string>
  7. #include <vector>
  8. #include "base/test/task_environment.h"
  9. #include "base/version.h"
  10. #include "components/prefs/testing_pref_service.h"
  11. #include "components/update_client/persisted_data.h"
  12. #include "components/update_client/test_activity_data_service.h"
  13. #include "testing/gtest/include/gtest/gtest.h"
  14. namespace update_client {
  15. TEST(PersistedDataTest, Simple) {
  16. base::test::TaskEnvironment env;
  17. auto pref = std::make_unique<TestingPrefServiceSimple>();
  18. PersistedData::RegisterPrefs(pref->registry());
  19. auto metadata = std::make_unique<PersistedData>(pref.get(), nullptr);
  20. EXPECT_EQ(-2, metadata->GetDateLastRollCall("someappid"));
  21. EXPECT_EQ(-2, metadata->GetDateLastActive("someappid"));
  22. EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someappid"));
  23. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid"));
  24. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid.withdot"));
  25. EXPECT_EQ(-2, metadata->GetInstallDate("someappid"));
  26. EXPECT_EQ(-2, metadata->GetInstallDate("someappid.withdot"));
  27. std::vector<std::string> items;
  28. items.push_back("someappid");
  29. items.push_back("someappid.withdot");
  30. test::SetDateLastData(metadata.get(), items, 3383);
  31. EXPECT_EQ(3383, metadata->GetDateLastRollCall("someappid"));
  32. EXPECT_EQ(3383, metadata->GetDateLastRollCall("someappid.withdot"));
  33. EXPECT_EQ(3383, metadata->GetInstallDate("someappid"));
  34. EXPECT_EQ(3383, metadata->GetInstallDate("someappid.withdot"));
  35. EXPECT_EQ(-2, metadata->GetDateLastActive("someappid"));
  36. EXPECT_EQ(-2, metadata->GetDateLastActive("someappid.withdot"));
  37. EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someappid"));
  38. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid"));
  39. EXPECT_EQ(-2, metadata->GetDateLastRollCall("someotherappid"));
  40. EXPECT_EQ(-2, metadata->GetDateLastActive("someotherappid"));
  41. EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someotherappid"));
  42. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someotherappid"));
  43. const std::string pf1 = metadata->GetPingFreshness("someappid");
  44. EXPECT_FALSE(pf1.empty());
  45. test::SetDateLastData(metadata.get(), items, 3386);
  46. EXPECT_EQ(3386, metadata->GetDateLastRollCall("someappid"));
  47. EXPECT_EQ(3383, metadata->GetInstallDate("someappid"));
  48. EXPECT_EQ(-2, metadata->GetDateLastActive("someappid"));
  49. EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someappid"));
  50. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid"));
  51. EXPECT_EQ(-2, metadata->GetDateLastRollCall("someotherappid"));
  52. EXPECT_EQ(-2, metadata->GetDateLastActive("someotherappid"));
  53. EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someotherappid"));
  54. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someotherappid"));
  55. EXPECT_EQ(-2, metadata->GetInstallDate("someotherappid"));
  56. const std::string pf2 = metadata->GetPingFreshness("someappid");
  57. EXPECT_FALSE(pf2.empty());
  58. // The following has a 1 / 2^128 chance of being flaky.
  59. EXPECT_NE(pf1, pf2);
  60. EXPECT_FALSE(metadata->GetProductVersion("someappid").IsValid());
  61. metadata->SetProductVersion("someappid", base::Version("1.0"));
  62. EXPECT_EQ(base::Version("1.0"), metadata->GetProductVersion("someappid"));
  63. EXPECT_TRUE(metadata->GetFingerprint("someappid").empty());
  64. metadata->SetFingerprint("someappid", "somefingerprint");
  65. EXPECT_STREQ("somefingerprint",
  66. metadata->GetFingerprint("someappid").c_str());
  67. }
  68. TEST(PersistedDataTest, SharedPref) {
  69. base::test::TaskEnvironment env;
  70. auto pref = std::make_unique<TestingPrefServiceSimple>();
  71. PersistedData::RegisterPrefs(pref->registry());
  72. auto metadata = std::make_unique<PersistedData>(pref.get(), nullptr);
  73. EXPECT_EQ(-2, metadata->GetDateLastRollCall("someappid"));
  74. EXPECT_EQ(-2, metadata->GetDateLastActive("someappid"));
  75. EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someappid"));
  76. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid"));
  77. EXPECT_EQ(-2, metadata->GetInstallDate("someappid"));
  78. std::vector<std::string> items;
  79. items.push_back("someappid");
  80. test::SetDateLastData(metadata.get(), items, 3383);
  81. // Now, create a new PersistedData reading from the same path, verify
  82. // that it loads the value.
  83. metadata = std::make_unique<PersistedData>(pref.get(), nullptr);
  84. EXPECT_EQ(3383, metadata->GetDateLastRollCall("someappid"));
  85. EXPECT_EQ(-2, metadata->GetDateLastActive("someappid"));
  86. EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someappid"));
  87. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someappid"));
  88. EXPECT_EQ(3383, metadata->GetInstallDate("someappid"));
  89. EXPECT_EQ(-2, metadata->GetDateLastRollCall("someotherappid"));
  90. EXPECT_EQ(-2, metadata->GetDateLastActive("someotherappid"));
  91. EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall("someotherappid"));
  92. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("someotherappid"));
  93. EXPECT_EQ(-2, metadata->GetInstallDate("someotherappid"));
  94. }
  95. TEST(PersistedDataTest, SimpleCohort) {
  96. base::test::TaskEnvironment env;
  97. auto pref = std::make_unique<TestingPrefServiceSimple>();
  98. PersistedData::RegisterPrefs(pref->registry());
  99. auto metadata = std::make_unique<PersistedData>(pref.get(), nullptr);
  100. EXPECT_EQ("", metadata->GetCohort("someappid"));
  101. EXPECT_EQ("", metadata->GetCohort("someotherappid"));
  102. EXPECT_EQ("", metadata->GetCohortHint("someappid"));
  103. EXPECT_EQ("", metadata->GetCohortHint("someotherappid"));
  104. EXPECT_EQ("", metadata->GetCohortName("someappid"));
  105. EXPECT_EQ("", metadata->GetCohortName("someotherappid"));
  106. metadata->SetCohort("someappid", "c1");
  107. metadata->SetCohort("someotherappid", "c2");
  108. metadata->SetCohortHint("someappid", "ch1");
  109. metadata->SetCohortHint("someotherappid", "ch2");
  110. metadata->SetCohortName("someappid", "cn1");
  111. metadata->SetCohortName("someotherappid", "cn2");
  112. EXPECT_EQ("c1", metadata->GetCohort("someappid"));
  113. EXPECT_EQ("c2", metadata->GetCohort("someotherappid"));
  114. EXPECT_EQ("ch1", metadata->GetCohortHint("someappid"));
  115. EXPECT_EQ("ch2", metadata->GetCohortHint("someotherappid"));
  116. EXPECT_EQ("cn1", metadata->GetCohortName("someappid"));
  117. EXPECT_EQ("cn2", metadata->GetCohortName("someotherappid"));
  118. metadata->SetCohort("someappid", "oc1");
  119. metadata->SetCohort("someotherappid", "oc2");
  120. metadata->SetCohortHint("someappid", "och1");
  121. metadata->SetCohortHint("someotherappid", "och2");
  122. metadata->SetCohortName("someappid", "ocn1");
  123. metadata->SetCohortName("someotherappid", "ocn2");
  124. EXPECT_EQ("oc1", metadata->GetCohort("someappid"));
  125. EXPECT_EQ("oc2", metadata->GetCohort("someotherappid"));
  126. EXPECT_EQ("och1", metadata->GetCohortHint("someappid"));
  127. EXPECT_EQ("och2", metadata->GetCohortHint("someotherappid"));
  128. EXPECT_EQ("ocn1", metadata->GetCohortName("someappid"));
  129. EXPECT_EQ("ocn2", metadata->GetCohortName("someotherappid"));
  130. }
  131. TEST(PersistedDataTest, ActivityData) {
  132. base::test::TaskEnvironment env;
  133. auto pref = std::make_unique<TestingPrefServiceSimple>();
  134. auto activity_service = std::make_unique<TestActivityDataService>();
  135. PersistedData::RegisterPrefs(pref->registry());
  136. auto metadata =
  137. std::make_unique<PersistedData>(pref.get(), activity_service.get());
  138. std::vector<std::string> items({"id1", "id2", "id3"});
  139. for (const auto& item : items) {
  140. EXPECT_EQ(-2, metadata->GetDateLastActive(item));
  141. EXPECT_EQ(-2, metadata->GetDateLastRollCall(item));
  142. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive(item));
  143. EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall(item));
  144. EXPECT_EQ(-2, metadata->GetInstallDate(item));
  145. EXPECT_EQ(false, test::GetActiveBit(metadata.get(), item));
  146. }
  147. test::SetDateLastData(metadata.get(), items, 1234);
  148. for (const auto& item : items) {
  149. EXPECT_EQ(false, test::GetActiveBit(metadata.get(), item));
  150. EXPECT_EQ(-2, metadata->GetDateLastActive(item));
  151. EXPECT_EQ(1234, metadata->GetDateLastRollCall(item));
  152. EXPECT_EQ(1234, metadata->GetInstallDate(item));
  153. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive(item));
  154. EXPECT_EQ(-2, metadata->GetDaysSinceLastRollCall(item));
  155. }
  156. activity_service->SetActiveBit("id1", true);
  157. activity_service->SetDaysSinceLastActive("id1", 3);
  158. activity_service->SetDaysSinceLastRollCall("id1", 2);
  159. activity_service->SetDaysSinceLastRollCall("id2", 3);
  160. activity_service->SetDaysSinceLastRollCall("id3", 4);
  161. EXPECT_EQ(true, test::GetActiveBit(metadata.get(), "id1"));
  162. EXPECT_EQ(false, test::GetActiveBit(metadata.get(), "id2"));
  163. EXPECT_EQ(false, test::GetActiveBit(metadata.get(), "id2"));
  164. EXPECT_EQ(3, metadata->GetDaysSinceLastActive("id1"));
  165. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("id2"));
  166. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("id3"));
  167. EXPECT_EQ(2, metadata->GetDaysSinceLastRollCall("id1"));
  168. EXPECT_EQ(3, metadata->GetDaysSinceLastRollCall("id2"));
  169. EXPECT_EQ(4, metadata->GetDaysSinceLastRollCall("id3"));
  170. test::SetDateLastData(metadata.get(), items, 5678);
  171. activity_service->SetActiveBit("id2", true);
  172. test::SetDateLastData(metadata.get(), items, 6789);
  173. EXPECT_EQ(false, test::GetActiveBit(metadata.get(), "id1"));
  174. EXPECT_EQ(false, test::GetActiveBit(metadata.get(), "id2"));
  175. EXPECT_EQ(false, test::GetActiveBit(metadata.get(), "id3"));
  176. EXPECT_EQ(3, metadata->GetDaysSinceLastActive("id1"));
  177. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("id2"));
  178. EXPECT_EQ(-2, metadata->GetDaysSinceLastActive("id3"));
  179. EXPECT_EQ(2, metadata->GetDaysSinceLastRollCall("id1"));
  180. EXPECT_EQ(3, metadata->GetDaysSinceLastRollCall("id2"));
  181. EXPECT_EQ(4, metadata->GetDaysSinceLastRollCall("id3"));
  182. EXPECT_EQ(5678, metadata->GetDateLastActive("id1"));
  183. EXPECT_EQ(6789, metadata->GetDateLastActive("id2"));
  184. EXPECT_EQ(-2, metadata->GetDateLastActive("id3"));
  185. EXPECT_EQ(6789, metadata->GetDateLastRollCall("id1"));
  186. EXPECT_EQ(6789, metadata->GetDateLastRollCall("id2"));
  187. EXPECT_EQ(6789, metadata->GetDateLastRollCall("id3"));
  188. EXPECT_EQ(1234, metadata->GetInstallDate("id1"));
  189. EXPECT_EQ(1234, metadata->GetInstallDate("id2"));
  190. EXPECT_EQ(1234, metadata->GetInstallDate("id3"));
  191. }
  192. } // namespace update_client