[Offline pages] Use the new policy bits

components/offline_pages/downloads/download_ui_adapter_unittest.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/downloads/download_ui_adapter.h"
 
 #include <stdint.h>
 
 #include <map>
 #include <memory>
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/files/file_path.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/test_mock_time_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "base/time/time.h"
 #include "components/offline_pages/client_namespace_constants.h"
+#include "components/offline_pages/client_policy_controller.h"
 #include "components/offline_pages/stub_offline_page_model.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace offline_pages {
 
 namespace {
 // Constants for a test OfflinePageItem.
 static const int kTestOfflineId1 = 1;
 static const int kTestOfflineId2 = 2;
 static const int kTestOfflineId3 = 3;
@@ skipping 10 matching lines @@
 static const int kFileSize = 1000;
 static const base::Time kTestCreationTime = base::Time::Now();
 static const base::string16 kTestTitle = base::ASCIIToUTF16("test title");
 }  // namespace
 
 // Mock OfflinePageModel for testing the SavePage calls.
 class MockOfflinePageModel : public StubOfflinePageModel {
  public:
   MockOfflinePageModel(base::TestMockTimeTaskRunner* task_runner)
       : observer_(nullptr),
-        task_runner_(task_runner) {
+        task_runner_(task_runner),
+        policy_controller_(new ClientPolicyController()) {
     adapter.reset(new DownloadUIAdapter(this));
     // Add one page.
     OfflinePageItem page(GURL(kTestUrl),
                          kTestOfflineId1,
                          kTestClientId1,
                          kTestFilePath,
                          kFileSize,
                          kTestCreationTime);
     page.title = kTestTitle;
     pages[kTestOfflineId1] = page;
@@ skipping 37 matching lines @@
       }
     }
   }
 
   void AddPageAndNotifyAdapter(const OfflinePageItem& page) {
     EXPECT_EQ(pages.end(), pages.find(page.offline_id));
     pages[page.offline_id] = page;
     observer_->OfflinePageModelChanged(this);
   }
 
+  ClientPolicyController* GetPolicyController() override {
+    return policy_controller_.get();
+  }
+
   // Normally, OfflinePageModel owns this adapter, so lets test it this way.
   std::unique_ptr<DownloadUIAdapter> adapter;
 
   std::map<int64_t, OfflinePageItem> pages;
 
  private:
   OfflinePageModel::Observer* observer_;
   base::TestMockTimeTaskRunner* task_runner_;
+  // Normally owned by OfflinePageModel.
+  std::unique_ptr<ClientPolicyController> policy_controller_;
 
   DISALLOW_COPY_AND_ASSIGN(MockOfflinePageModel);
 };
 
 class DownloadUIAdapterTest
     : public testing::Test,
       public DownloadUIAdapter::Observer {
  public:
   DownloadUIAdapterTest();
   ~DownloadUIAdapterTest() override;
@@ skipping 158 matching lines @@
   model->adapter->RemoveObserver(this);
   // This will complete async fetch of items, but...
   PumpLoop();
   // items should not be loaded when there is no observers!
   EXPECT_FALSE(items_loaded);
   // This should not crash.
   model->adapter->AddObserver(this);
 }
 
 }  // namespace offline_pages