Create skeleton for the Prefetching store and initial pipeline step.

components/offline_pages/content/suggested_articles_observer_unittest.cc

 // Copyright 2017 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/content/suggested_articles_observer.h"
 
 #include "base/run_loop.h"
 #include "base/test/test_simple_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "components/offline_pages/core/client_namespace_constants.h"
@@ skipping 14 matching lines @@
 ContentSuggestion ContentSuggestionFromTestURL(const GURL& test_url) {
   auto category =
       Category::FromKnownCategory(ntp_snippets::KnownCategories::ARTICLES);
   return ContentSuggestion(category, test_url.spec(), test_url);
 }
 
 class TestingPrefetchDispatcher : public PrefetchDispatcher {
  public:
   TestingPrefetchDispatcher() = default;
 
+  void SetService(PrefetchService* service) override{};
+
   void AddCandidatePrefetchURLs(
-      const std::vector<PrefetchURL>& suggested_urls) override {
-    latest_prefetch_urls = suggested_urls;
+      const std::string& name_space,
+      const std::vector<PrefetchURL>& prefetch_urls) override {
+    latest_name_space = name_space;
+    latest_prefetch_urls = prefetch_urls;
     new_suggestions_count++;
   }
 
   void RemoveAllUnprocessedPrefetchURLs(
       const std::string& name_space) override {
-    DCHECK_EQ(name_space, kSuggestedArticlesNamespace);
+    DCHECK_EQ(kSuggestedArticlesNamespace, name_space);
     latest_prefetch_urls.clear();
     remove_all_suggestions_count++;
   }
 
-  void RemovePrefetchURLsByClientId(const ClientId& client_id) override {
-    DCHECK_EQ(client_id.name_space, kSuggestedArticlesNamespace);
+  void RemovePrefetchURLsByClientId(const std::string& name_space,
+                                    const std::string& client_id) override {
+    DCHECK_EQ(kSuggestedArticlesNamespace, name_space);
     remove_by_client_id_count++;
-    last_removed_client_id = base::MakeUnique<ClientId>(client_id);
+    last_removed_name_space = name_space;
+    last_removed_client_id = client_id;
   }
 
   void BeginBackgroundTask(
       std::unique_ptr<ScopedBackgroundTask> task) override {}
   void StopBackgroundTask(ScopedBackgroundTask* task) override {}
 
+  std::string latest_name_space;
   std::vector<PrefetchURL> latest_prefetch_urls;
-  std::unique_ptr<ClientId> last_removed_client_id;
+  std::string last_removed_name_space;
+  std::string last_removed_client_id;
 
   int new_suggestions_count = 0;
   int remove_all_suggestions_count = 0;
   int remove_by_client_id_count = 0;
 };
 
 class TestingPrefetchService : public PrefetchService {
  public:
   TestingPrefetchService() = default;
 
   PrefetchDispatcher* GetDispatcher() override { return &dispatcher; };
+  PrefetchStore* GetStore() override {
+    CHECK(false);
+    return nullptr;
+  };
 
   TestingPrefetchDispatcher dispatcher;
 };
 
 class TestDelegate : public SuggestedArticlesObserver::Delegate {
  public:
   TestDelegate() = default;
   ~TestDelegate() override = default;
 
   const std::vector<ContentSuggestion>& GetSuggestions(
@@ skipping 80 matching lines @@
   test_delegate()->suggestions.push_back(
       ContentSuggestionFromTestURL(test_url_1));
 
   observer()->OnCategoryStatusChanged(category,
                                       ntp_snippets::CategoryStatus::AVAILABLE);
   observer()->OnNewSuggestions(category);
   EXPECT_EQ(1, test_prefetch_dispatcher()->new_suggestions_count);
   EXPECT_EQ(1U, test_prefetch_dispatcher()->latest_prefetch_urls.size());
   EXPECT_EQ(test_url_1,
             test_prefetch_dispatcher()->latest_prefetch_urls[0].url);
-  EXPECT_EQ(
-      kSuggestedArticlesNamespace,
-      test_prefetch_dispatcher()->latest_prefetch_urls[0].client_id.name_space);
+  EXPECT_EQ(kSuggestedArticlesNamespace,
+            test_prefetch_dispatcher()->latest_name_space);
 }
 
 TEST_F(OfflinePageSuggestedArticlesObserverTest, RemovesAllOnBadStatus) {
   const GURL test_url_1("https://www.example.com/1");
   const GURL test_url_2("https://www.example.com/2");
   test_delegate()->suggestions.push_back(
       ContentSuggestionFromTestURL(test_url_1));
   test_delegate()->suggestions.push_back(
       ContentSuggestionFromTestURL(test_url_2));
 
@@ skipping 17 matching lines @@
       ContentSuggestionFromTestURL(test_url_1));
   observer()->OnCategoryStatusChanged(category,
                                       ntp_snippets::CategoryStatus::AVAILABLE);
   observer()->OnNewSuggestions(category);
   ASSERT_EQ(1U, test_prefetch_dispatcher()->latest_prefetch_urls.size());
 
   observer()->OnSuggestionInvalidated(
       ntp_snippets::ContentSuggestion::ID(category, test_url_1.spec()));
 
   EXPECT_EQ(1, test_prefetch_dispatcher()->remove_by_client_id_count);
-  EXPECT_EQ(ClientId(kSuggestedArticlesNamespace, test_url_1.spec()),
-            *test_prefetch_dispatcher()->last_removed_client_id);
+  EXPECT_EQ(test_url_1.spec(),
+            test_prefetch_dispatcher()->last_removed_client_id);
+  EXPECT_EQ(kSuggestedArticlesNamespace,
+            test_prefetch_dispatcher()->last_removed_name_space);
 }
 
 }  // namespace offline_pages