client_policy_controller_unittest.cc 8.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207
  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 "components/offline_pages/core/offline_page_client_policy.h"
  5. #include <algorithm>
  6. #include <memory>
  7. #include <set>
  8. #include "base/containers/contains.h"
  9. #include "components/offline_pages/core/client_namespace_constants.h"
  10. #include "components/offline_pages/core/offline_page_feature.h"
  11. #include "testing/gtest/include/gtest/gtest.h"
  12. namespace offline_pages {
  13. namespace {
  14. const char kUndefinedNamespace[] = "undefined";
  15. bool isTemporary(const OfflinePageClientPolicy& policy) {
  16. return policy.lifetime_type == LifetimeType::TEMPORARY;
  17. }
  18. } // namespace
  19. class ClientPolicyControllerTest : public testing::Test {
  20. public:
  21. // testing::Test
  22. void SetUp() override {}
  23. void TearDown() override {}
  24. protected:
  25. void ExpectTemporary(std::string name_space);
  26. void ExpectDownloadSupport(std::string name_space, bool expectation);
  27. void ExpectPersistent(std::string name_space);
  28. void ExpectRestrictedToTabFromClientId(std::string name_space,
  29. bool expectation);
  30. void ExpectRequiresSpecificUserSettings(std::string name_space,
  31. bool expectation);
  32. };
  33. void ClientPolicyControllerTest::ExpectTemporary(std::string name_space) {
  34. EXPECT_TRUE(base::Contains(GetTemporaryPolicyNamespaces(), name_space))
  35. << "Namespace " << name_space
  36. << " had incorrect lifetime type when getting temporary namespaces.";
  37. EXPECT_EQ(GetPolicy(name_space).lifetime_type, LifetimeType::TEMPORARY)
  38. << "Namespace " << name_space
  39. << " had incorrect lifetime type setting when directly checking"
  40. " if it is temporary.";
  41. EXPECT_FALSE(base::Contains(GetPersistentPolicyNamespaces(), name_space))
  42. << "Namespace " << name_space
  43. << " had incorrect lifetime type when getting persistent namespaces.";
  44. }
  45. void ClientPolicyControllerTest::ExpectDownloadSupport(std::string name_space,
  46. bool expectation) {
  47. EXPECT_EQ(expectation, GetPolicy(name_space).is_supported_by_download)
  48. << "Namespace " << name_space
  49. << " had incorrect download support when directly checking if supported"
  50. " by download.";
  51. }
  52. void ClientPolicyControllerTest::ExpectPersistent(std::string name_space) {
  53. EXPECT_FALSE(base::Contains(GetTemporaryPolicyNamespaces(), name_space))
  54. << "Namespace " << name_space
  55. << " had incorrect lifetime type when getting temporary namespaces.";
  56. EXPECT_EQ(GetPolicy(name_space).lifetime_type, LifetimeType::PERSISTENT)
  57. << "Namespace " << name_space
  58. << " had incorrect lifetime type setting when directly checking"
  59. " if it is temporary.";
  60. EXPECT_TRUE(base::Contains(GetPersistentPolicyNamespaces(), name_space))
  61. << "Namespace " << name_space
  62. << " had incorrect lifetime type when getting persistent namespaces.";
  63. }
  64. void ClientPolicyControllerTest::ExpectRestrictedToTabFromClientId(
  65. std::string name_space,
  66. bool expectation) {
  67. EXPECT_EQ(expectation,
  68. GetPolicy(name_space).is_restricted_to_tab_from_client_id)
  69. << "Namespace " << name_space
  70. << " had incorrect restriction when directly checking if the namespace"
  71. " is restricted to the tab from the client id field";
  72. }
  73. void ClientPolicyControllerTest::ExpectRequiresSpecificUserSettings(
  74. std::string name_space,
  75. bool expectation) {
  76. EXPECT_EQ(expectation, GetPolicy(name_space).requires_specific_user_settings)
  77. << "Namespace " << name_space
  78. << " had incorrect download support when directly checking if disabled"
  79. " when prefetch settings are disabled.";
  80. }
  81. TEST_F(ClientPolicyControllerTest, FallbackTest) {
  82. const OfflinePageClientPolicy& policy = GetPolicy(kUndefinedNamespace);
  83. EXPECT_EQ(policy.name_space, kDefaultNamespace);
  84. EXPECT_TRUE(isTemporary(policy));
  85. ExpectTemporary(kDefaultNamespace);
  86. EXPECT_FALSE(
  87. base::Contains(GetTemporaryPolicyNamespaces(), kUndefinedNamespace));
  88. EXPECT_EQ(GetPolicy(kUndefinedNamespace).lifetime_type,
  89. LifetimeType::TEMPORARY);
  90. ExpectDownloadSupport(kUndefinedNamespace, false);
  91. ExpectDownloadSupport(kDefaultNamespace, false);
  92. ExpectRestrictedToTabFromClientId(kUndefinedNamespace, false);
  93. ExpectRestrictedToTabFromClientId(kDefaultNamespace, false);
  94. ExpectRequiresSpecificUserSettings(kUndefinedNamespace, false);
  95. ExpectRequiresSpecificUserSettings(kDefaultNamespace, false);
  96. }
  97. TEST_F(ClientPolicyControllerTest, CheckBookmarkDefined) {
  98. OfflinePageClientPolicy policy = GetPolicy(kBookmarkNamespace);
  99. EXPECT_EQ(policy.name_space, kBookmarkNamespace);
  100. EXPECT_TRUE(isTemporary(policy));
  101. ExpectTemporary(kBookmarkNamespace);
  102. ExpectDownloadSupport(kBookmarkNamespace, false);
  103. ExpectRestrictedToTabFromClientId(kBookmarkNamespace, false);
  104. ExpectRequiresSpecificUserSettings(kBookmarkNamespace, false);
  105. }
  106. TEST_F(ClientPolicyControllerTest, CheckLastNDefined) {
  107. OfflinePageClientPolicy policy = GetPolicy(kLastNNamespace);
  108. EXPECT_EQ(policy.name_space, kLastNNamespace);
  109. EXPECT_TRUE(isTemporary(policy));
  110. ExpectTemporary(kLastNNamespace);
  111. ExpectDownloadSupport(kLastNNamespace, false);
  112. ExpectRestrictedToTabFromClientId(kLastNNamespace, true);
  113. ExpectRequiresSpecificUserSettings(kLastNNamespace, false);
  114. }
  115. TEST_F(ClientPolicyControllerTest, CheckAsyncDefined) {
  116. OfflinePageClientPolicy policy = GetPolicy(kAsyncNamespace);
  117. EXPECT_EQ(policy.name_space, kAsyncNamespace);
  118. EXPECT_FALSE(isTemporary(policy));
  119. ExpectDownloadSupport(kAsyncNamespace, true);
  120. ExpectPersistent(kAsyncNamespace);
  121. ExpectRestrictedToTabFromClientId(kAsyncNamespace, false);
  122. ExpectRequiresSpecificUserSettings(kAsyncNamespace, false);
  123. }
  124. TEST_F(ClientPolicyControllerTest, CheckCCTDefined) {
  125. OfflinePageClientPolicy policy = GetPolicy(kCCTNamespace);
  126. EXPECT_EQ(policy.name_space, kCCTNamespace);
  127. EXPECT_TRUE(isTemporary(policy));
  128. ExpectTemporary(kCCTNamespace);
  129. ExpectDownloadSupport(kCCTNamespace, false);
  130. ExpectRestrictedToTabFromClientId(kCCTNamespace, false);
  131. ExpectRequiresSpecificUserSettings(kCCTNamespace, true);
  132. }
  133. TEST_F(ClientPolicyControllerTest, CheckDownloadDefined) {
  134. OfflinePageClientPolicy policy = GetPolicy(kDownloadNamespace);
  135. EXPECT_EQ(policy.name_space, kDownloadNamespace);
  136. EXPECT_FALSE(isTemporary(policy));
  137. ExpectDownloadSupport(kDownloadNamespace, true);
  138. ExpectPersistent(kDownloadNamespace);
  139. ExpectRestrictedToTabFromClientId(kDownloadNamespace, false);
  140. ExpectRequiresSpecificUserSettings(kDownloadNamespace, false);
  141. }
  142. TEST_F(ClientPolicyControllerTest, CheckNTPSuggestionsDefined) {
  143. OfflinePageClientPolicy policy = GetPolicy(kNTPSuggestionsNamespace);
  144. EXPECT_EQ(policy.name_space, kNTPSuggestionsNamespace);
  145. EXPECT_FALSE(isTemporary(policy));
  146. ExpectDownloadSupport(kNTPSuggestionsNamespace, true);
  147. ExpectPersistent(kNTPSuggestionsNamespace);
  148. ExpectRestrictedToTabFromClientId(kNTPSuggestionsNamespace, false);
  149. ExpectRequiresSpecificUserSettings(kNTPSuggestionsNamespace, false);
  150. }
  151. TEST_F(ClientPolicyControllerTest, CheckSuggestedArticlesDefined) {
  152. OfflinePageClientPolicy policy = GetPolicy(kSuggestedArticlesNamespace);
  153. EXPECT_EQ(policy.name_space, kSuggestedArticlesNamespace);
  154. EXPECT_TRUE(isTemporary(policy));
  155. ExpectTemporary(kSuggestedArticlesNamespace);
  156. ExpectDownloadSupport(kSuggestedArticlesNamespace, true);
  157. ExpectRestrictedToTabFromClientId(kSuggestedArticlesNamespace, false);
  158. ExpectRequiresSpecificUserSettings(kSuggestedArticlesNamespace, false);
  159. }
  160. TEST_F(ClientPolicyControllerTest, CheckLivePageSharingDefined) {
  161. OfflinePageClientPolicy policy = GetPolicy(kLivePageSharingNamespace);
  162. EXPECT_EQ(policy.name_space, kLivePageSharingNamespace);
  163. EXPECT_TRUE(isTemporary(policy));
  164. ExpectTemporary(kLivePageSharingNamespace);
  165. ExpectDownloadSupport(kLivePageSharingNamespace, false);
  166. ExpectRestrictedToTabFromClientId(kLivePageSharingNamespace, true);
  167. ExpectRequiresSpecificUserSettings(kLivePageSharingNamespace, false);
  168. }
  169. TEST_F(ClientPolicyControllerTest, AllTemporaryNamespaces) {
  170. std::vector<std::string> all_namespaces = GetAllPolicyNamespaces();
  171. const std::vector<std::string>& cache_reset_namespaces_list =
  172. GetTemporaryPolicyNamespaces();
  173. std::set<std::string> cache_reset_namespaces(
  174. cache_reset_namespaces_list.begin(), cache_reset_namespaces_list.end());
  175. for (auto name_space : cache_reset_namespaces) {
  176. if (cache_reset_namespaces.count(name_space) > 0)
  177. ExpectTemporary(name_space);
  178. else
  179. ExpectPersistent(name_space);
  180. }
  181. }
  182. } // namespace offline_pages