Query API: Introduces an OfflinePageModelQuery object.

components/offline_pages/client_policy_controller.cc

 // 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 "components/offline_pages/client_policy_controller.h"
 
+#include <set>

[fgorski, 2016/10/26 18:05:41]

remove?

[dewittj, 2016/10/27 22:49:17]

Done.

 #include <utility>
 
 #include "base/memory/ptr_util.h"
 #include "base/time/time.h"
 #include "components/offline_pages/client_namespace_constants.h"
 
 using LifetimeType = offline_pages::LifetimePolicy::LifetimeType;
 
 namespace offline_pages {
 
@@ skipping 60 matching lines @@
 
 const OfflinePageClientPolicy& ClientPolicyController::GetPolicy(
     const std::string& name_space) const {
   const auto& iter = policies_.find(name_space);
   if (iter != policies_.end())
     return iter->second;
   // Fallback when the namespace isn't defined.
   return policies_.at(kDefaultNamespace);
 }
 
+std::vector<std::string> ClientPolicyController::GetAllNamespaces() const {
+  std::vector<std::string> result;
+  for (const auto& policy_item : policies_)
+    result.emplace_back(policy_item.first);
+
+  return result;
+}
+
 bool ClientPolicyController::IsRemovedOnCacheReset(
     const std::string& name_space) const {
   return GetPolicy(name_space).feature_policy.is_removed_on_cache_reset;
 }
 
 bool ClientPolicyController::IsSupportedByDownload(
     const std::string& name_space) const {
   return GetPolicy(name_space).feature_policy.is_supported_by_download;
 }
 
@@ skipping 41 matching lines @@
 
   show_in_original_tab_cache_ = base::MakeUnique<std::vector<std::string>>();
   for (const auto& policy_item : policies_) {
     if (policy_item.second.feature_policy.only_shown_in_original_tab)
       show_in_original_tab_cache_->emplace_back(policy_item.first);
   }
 
   return *show_in_original_tab_cache_;
 }
 
+void ClientPolicyController::AddPolicyForTest(
+    std::string name_space,
+    const OfflinePageClientPolicyBuilder& builder) {
+  policies_.insert(std::make_pair(name_space, builder.Build()));
+}
+
 }  // namespace offline_pages