Retrieve article original URL from entryID

components/dom_distiller/core/viewer_unittest.cc

 // Copyright 2014 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/dom_distiller/core/viewer.h"
 
 #include "components/dom_distiller/core/distilled_page_prefs.h"
 #include "components/dom_distiller/core/dom_distiller_service.h"
 #include "components/dom_distiller/core/dom_distiller_test_util.h"
 #include "components/dom_distiller/core/task_tracker.h"
@@ skipping 21 matching lines @@
   MOCK_METHOD3(AddToList,
                const std::string(const GURL&,
                                  DistillerPage*,
                                  const ArticleAvailableCallback&));
   virtual const std::string AddToList(
       const GURL& url,
       scoped_ptr<DistillerPage> distiller_page,
       const ArticleAvailableCallback& article_cb) {
     return AddToList(url, distiller_page.get(), article_cb);
   }
+  MOCK_METHOD1(HasEntry, bool(const std::string&));
+  MOCK_METHOD1(GetUrlForEntry, std::string(const std::string&));
   MOCK_CONST_METHOD0(GetEntries, std::vector<ArticleEntry>());
   MOCK_METHOD1(AddObserver, void(DomDistillerObserver*));
   MOCK_METHOD1(RemoveObserver, void(DomDistillerObserver*));
   MOCK_METHOD0(ViewUrlImpl, ViewerHandle*());
   virtual scoped_ptr<ViewerHandle> ViewUrl(
       ViewRequestDelegate*,
       scoped_ptr<DistillerPage> distiller_page,
       const GURL&) {
     return scoped_ptr<ViewerHandle>(ViewUrlImpl());
   }
@@ skipping 93 matching lines @@
             0);
   EXPECT_EQ(kLightJs.compare(viewer::GetDistilledPageThemeJs(
                 DistilledPagePrefs::LIGHT)),
             0);
   EXPECT_EQ(kSepiaJs.compare(viewer::GetDistilledPageThemeJs(
                 DistilledPagePrefs::SEPIA)),
             0);
 }
 
 }  // namespace dom_distiller