Index: components/ntp_snippets/remote/remote_suggestions_provider_unittest.cc |
diff --git a/components/ntp_snippets/remote/remote_suggestions_provider_unittest.cc b/components/ntp_snippets/remote/remote_suggestions_provider_unittest.cc |
deleted file mode 100644 |
index 1e14f2aa51643d05ca840421aa93635f790d068a..0000000000000000000000000000000000000000 |
--- a/components/ntp_snippets/remote/remote_suggestions_provider_unittest.cc |
+++ /dev/null |
@@ -1,1829 +0,0 @@ |
-// Copyright 2015 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/ntp_snippets/remote/remote_suggestions_provider.h" |
- |
-#include <memory> |
-#include <utility> |
-#include <vector> |
- |
-#include "base/command_line.h" |
-#include "base/files/file_path.h" |
-#include "base/files/scoped_temp_dir.h" |
-#include "base/json/json_reader.h" |
-#include "base/macros.h" |
-#include "base/memory/ptr_util.h" |
-#include "base/message_loop/message_loop.h" |
-#include "base/run_loop.h" |
-#include "base/strings/string_number_conversions.h" |
-#include "base/strings/string_util.h" |
-#include "base/strings/stringprintf.h" |
-#include "base/test/histogram_tester.h" |
-#include "base/test/simple_test_clock.h" |
-#include "base/threading/thread_task_runner_handle.h" |
-#include "base/time/time.h" |
-#include "components/image_fetcher/image_decoder.h" |
-#include "components/image_fetcher/image_fetcher.h" |
-#include "components/image_fetcher/image_fetcher_delegate.h" |
-#include "components/ntp_snippets/category.h" |
-#include "components/ntp_snippets/category_info.h" |
-#include "components/ntp_snippets/category_rankers/category_ranker.h" |
-#include "components/ntp_snippets/category_rankers/constant_category_ranker.h" |
-#include "components/ntp_snippets/category_rankers/mock_category_ranker.h" |
-#include "components/ntp_snippets/ntp_snippets_constants.h" |
-#include "components/ntp_snippets/pref_names.h" |
-#include "components/ntp_snippets/remote/ntp_snippet.h" |
-#include "components/ntp_snippets/remote/ntp_snippets_fetcher.h" |
-#include "components/ntp_snippets/remote/ntp_snippets_scheduler.h" |
-#include "components/ntp_snippets/remote/remote_suggestions_database.h" |
-#include "components/ntp_snippets/remote/test_utils.h" |
-#include "components/ntp_snippets/user_classifier.h" |
-#include "components/prefs/testing_pref_service.h" |
-#include "components/signin/core/browser/fake_profile_oauth2_token_service.h" |
-#include "components/signin/core/browser/fake_signin_manager.h" |
-#include "components/variations/variations_params_manager.h" |
-#include "net/url_request/test_url_fetcher_factory.h" |
-#include "net/url_request/url_request_test_util.h" |
-#include "testing/gmock/include/gmock/gmock.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
-#include "ui/gfx/image/image.h" |
-#include "ui/gfx/image/image_unittest_util.h" |
- |
-using image_fetcher::ImageFetcher; |
-using image_fetcher::ImageFetcherDelegate; |
-using testing::_; |
-using testing::ElementsAre; |
-using testing::Eq; |
-using testing::InSequence; |
-using testing::Invoke; |
-using testing::IsEmpty; |
-using testing::Mock; |
-using testing::MockFunction; |
-using testing::NiceMock; |
-using testing::Not; |
-using testing::SaveArg; |
-using testing::SizeIs; |
-using testing::StartsWith; |
-using testing::WithArgs; |
- |
-namespace ntp_snippets { |
- |
-namespace { |
- |
-MATCHER_P(IdEq, value, "") { |
- return arg->id() == value; |
-} |
- |
-MATCHER_P(IdWithinCategoryEq, expected_id, "") { |
- return arg.id().id_within_category() == expected_id; |
-} |
- |
-MATCHER_P(IsCategory, id, "") { |
- return arg.id() == static_cast<int>(id); |
-} |
- |
-MATCHER_P(HasCode, code, "") { |
- return arg.code == code; |
-} |
- |
-const base::Time::Exploded kDefaultCreationTime = {2015, 11, 4, 25, 13, 46, 45}; |
-const char kTestContentSuggestionsServerEndpoint[] = |
- "https://localunittest-chromecontentsuggestions-pa.googleapis.com/v1/" |
- "suggestions/fetch"; |
-const char kAPIKey[] = "fakeAPIkey"; |
-const char kTestContentSuggestionsServerWithAPIKey[] = |
- "https://localunittest-chromecontentsuggestions-pa.googleapis.com/v1/" |
- "suggestions/fetch?key=fakeAPIkey"; |
- |
-const char kSnippetUrl[] = "http://localhost/foobar"; |
-const char kSnippetTitle[] = "Title"; |
-const char kSnippetText[] = "Snippet"; |
-const char kSnippetSalientImage[] = "http://localhost/salient_image"; |
-const char kSnippetPublisherName[] = "Foo News"; |
-const char kSnippetAmpUrl[] = "http://localhost/amp"; |
- |
-const char kSnippetUrl2[] = "http://foo.com/bar"; |
- |
-const char kTestJsonDefaultCategoryTitle[] = "Some title"; |
- |
-const int kUnknownRemoteCategoryId = 1234; |
- |
-base::Time GetDefaultCreationTime() { |
- base::Time out_time; |
- EXPECT_TRUE(base::Time::FromUTCExploded(kDefaultCreationTime, &out_time)); |
- return out_time; |
-} |
- |
-base::Time GetDefaultExpirationTime() { |
- return base::Time::Now() + base::TimeDelta::FromHours(1); |
-} |
- |
-std::string GetCategoryJson(const std::vector<std::string>& snippets, |
- int remote_category_id, |
- const std::string& category_title) { |
- return base::StringPrintf( |
- " {\n" |
- " \"id\": %d,\n" |
- " \"localizedTitle\": \"%s\",\n" |
- " \"suggestions\": [%s]\n" |
- " }\n", |
- remote_category_id, category_title.c_str(), |
- base::JoinString(snippets, ", ").c_str()); |
-} |
- |
-class MultiCategoryJsonBuilder { |
- public: |
- MultiCategoryJsonBuilder() {} |
- |
- MultiCategoryJsonBuilder& AddCategoryWithCustomTitle( |
- const std::vector<std::string>& snippets, |
- int remote_category_id, |
- const std::string& category_title) { |
- category_json_.push_back( |
- GetCategoryJson(snippets, remote_category_id, category_title)); |
- return *this; |
- } |
- |
- MultiCategoryJsonBuilder& AddCategory( |
- const std::vector<std::string>& snippets, |
- int remote_category_id) { |
- return AddCategoryWithCustomTitle( |
- snippets, remote_category_id, |
- "Title" + base::IntToString(remote_category_id)); |
- } |
- |
- std::string Build() { |
- return base::StringPrintf( |
- "{\n" |
- " \"categories\": [\n" |
- "%s\n" |
- " ]\n" |
- "}\n", |
- base::JoinString(category_json_, " ,\n").c_str()); |
- } |
- |
- private: |
- std::vector<std::string> category_json_; |
-}; |
- |
-// TODO(vitaliii): Remove these convenience functions as they do not provide |
-// that much value and add additional redirections obscuring the code. |
-std::string GetTestJson(const std::vector<std::string>& snippets, |
- const std::string& category_title) { |
- return MultiCategoryJsonBuilder() |
- .AddCategoryWithCustomTitle(snippets, /*remote_category_id=*/1, |
- category_title) |
- .Build(); |
-} |
- |
-std::string GetTestJson(const std::vector<std::string>& snippets) { |
- return GetTestJson(snippets, kTestJsonDefaultCategoryTitle); |
-} |
- |
-std::string FormatTime(const base::Time& t) { |
- base::Time::Exploded x; |
- t.UTCExplode(&x); |
- return base::StringPrintf("%04d-%02d-%02dT%02d:%02d:%02dZ", x.year, x.month, |
- x.day_of_month, x.hour, x.minute, x.second); |
-} |
- |
-std::string GetSnippetWithUrlAndTimesAndSource( |
- const std::vector<std::string>& ids, |
- const std::string& url, |
- const base::Time& creation_time, |
- const base::Time& expiry_time, |
- const std::string& publisher, |
- const std::string& amp_url) { |
- const std::string ids_string = base::JoinString(ids, "\",\n \""); |
- return base::StringPrintf( |
- "{\n" |
- " \"ids\": [\n" |
- " \"%s\"\n" |
- " ],\n" |
- " \"title\": \"%s\",\n" |
- " \"snippet\": \"%s\",\n" |
- " \"fullPageUrl\": \"%s\",\n" |
- " \"creationTime\": \"%s\",\n" |
- " \"expirationTime\": \"%s\",\n" |
- " \"attribution\": \"%s\",\n" |
- " \"imageUrl\": \"%s\",\n" |
- " \"ampUrl\": \"%s\"\n" |
- " }", |
- ids_string.c_str(), kSnippetTitle, kSnippetText, url.c_str(), |
- FormatTime(creation_time).c_str(), FormatTime(expiry_time).c_str(), |
- publisher.c_str(), kSnippetSalientImage, amp_url.c_str()); |
-} |
- |
-std::string GetSnippetWithSources(const std::string& source_url, |
- const std::string& publisher, |
- const std::string& amp_url) { |
- return GetSnippetWithUrlAndTimesAndSource( |
- {kSnippetUrl}, source_url, GetDefaultCreationTime(), |
- GetDefaultExpirationTime(), publisher, amp_url); |
-} |
- |
-std::string GetSnippetWithUrlAndTimes(const std::string& url, |
- const base::Time& content_creation_time, |
- const base::Time& expiry_time) { |
- return GetSnippetWithUrlAndTimesAndSource({url}, url, content_creation_time, |
- expiry_time, kSnippetPublisherName, |
- kSnippetAmpUrl); |
-} |
- |
-std::string GetSnippetWithTimes(const base::Time& content_creation_time, |
- const base::Time& expiry_time) { |
- return GetSnippetWithUrlAndTimes(kSnippetUrl, content_creation_time, |
- expiry_time); |
-} |
- |
-std::string GetSnippetWithUrl(const std::string& url) { |
- return GetSnippetWithUrlAndTimes(url, GetDefaultCreationTime(), |
- GetDefaultExpirationTime()); |
-} |
- |
-std::string GetSnippet() { |
- return GetSnippetWithUrlAndTimes(kSnippetUrl, GetDefaultCreationTime(), |
- GetDefaultExpirationTime()); |
-} |
- |
-std::string GetSnippetN(int n) { |
- return GetSnippetWithUrlAndTimes(base::StringPrintf("%s/%d", kSnippetUrl, n), |
- GetDefaultCreationTime(), |
- GetDefaultExpirationTime()); |
-} |
- |
-std::string GetExpiredSnippet() { |
- return GetSnippetWithTimes(GetDefaultCreationTime(), base::Time::Now()); |
-} |
- |
-std::string GetInvalidSnippet() { |
- std::string json_str = GetSnippet(); |
- // Make the json invalid by removing the final closing brace. |
- return json_str.substr(0, json_str.size() - 1); |
-} |
- |
-std::string GetIncompleteSnippet() { |
- std::string json_str = GetSnippet(); |
- // Rename the "url" entry. The result is syntactically valid json that will |
- // fail to parse as snippets. |
- size_t pos = json_str.find("\"fullPageUrl\""); |
- if (pos == std::string::npos) { |
- NOTREACHED(); |
- return std::string(); |
- } |
- json_str[pos + 1] = 'x'; |
- return json_str; |
-} |
- |
-using ServeImageCallback = base::Callback<void( |
- const std::string&, |
- base::Callback<void(const std::string&, const gfx::Image&)>)>; |
- |
-void ServeOneByOneImage( |
- image_fetcher::ImageFetcherDelegate* notify, |
- const std::string& id, |
- base::Callback<void(const std::string&, const gfx::Image&)> callback) { |
- base::ThreadTaskRunnerHandle::Get()->PostTask( |
- FROM_HERE, base::Bind(callback, id, gfx::test::CreateImage(1, 1))); |
- notify->OnImageDataFetched(id, "1-by-1-image-data"); |
-} |
- |
-gfx::Image FetchImage(RemoteSuggestionsProvider* service, |
- const ContentSuggestion::ID& suggestion_id) { |
- gfx::Image result; |
- base::RunLoop run_loop; |
- service->FetchSuggestionImage(suggestion_id, |
- base::Bind( |
- [](base::Closure signal, gfx::Image* output, |
- const gfx::Image& loaded) { |
- *output = loaded; |
- signal.Run(); |
- }, |
- run_loop.QuitClosure(), &result)); |
- run_loop.Run(); |
- return result; |
-} |
- |
-void ParseJson(const std::string& json, |
- const SuccessCallback& success_callback, |
- const ErrorCallback& error_callback) { |
- base::JSONReader json_reader; |
- std::unique_ptr<base::Value> value = json_reader.ReadToValue(json); |
- if (value) { |
- success_callback.Run(std::move(value)); |
- } else { |
- error_callback.Run(json_reader.GetErrorMessage()); |
- } |
-} |
- |
-// Factory for FakeURLFetcher objects that always generate errors. |
-class FailingFakeURLFetcherFactory : public net::URLFetcherFactory { |
- public: |
- std::unique_ptr<net::URLFetcher> CreateURLFetcher( |
- int id, |
- const GURL& url, |
- net::URLFetcher::RequestType request_type, |
- net::URLFetcherDelegate* d) override { |
- return base::MakeUnique<net::FakeURLFetcher>( |
- url, d, /*response_data=*/std::string(), net::HTTP_NOT_FOUND, |
- net::URLRequestStatus::FAILED); |
- } |
-}; |
- |
-class MockScheduler : public NTPSnippetsScheduler { |
- public: |
- MOCK_METHOD2(Schedule, |
- bool(base::TimeDelta period_wifi, |
- base::TimeDelta period_fallback)); |
- MOCK_METHOD0(Unschedule, bool()); |
-}; |
- |
-class MockImageFetcher : public ImageFetcher { |
- public: |
- MOCK_METHOD1(SetImageFetcherDelegate, void(ImageFetcherDelegate*)); |
- MOCK_METHOD1(SetDataUseServiceName, void(DataUseServiceName)); |
- MOCK_METHOD3( |
- StartOrQueueNetworkRequest, |
- void(const std::string&, |
- const GURL&, |
- base::Callback<void(const std::string&, const gfx::Image&)>)); |
-}; |
- |
-class FakeContentSuggestionsProviderObserver |
- : public ContentSuggestionsProvider::Observer { |
- public: |
- FakeContentSuggestionsProviderObserver() = default; |
- |
- void OnNewSuggestions(ContentSuggestionsProvider* provider, |
- Category category, |
- std::vector<ContentSuggestion> suggestions) override { |
- suggestions_[category] = std::move(suggestions); |
- } |
- |
- void OnCategoryStatusChanged(ContentSuggestionsProvider* provider, |
- Category category, |
- CategoryStatus new_status) override { |
- statuses_[category] = new_status; |
- } |
- |
- void OnSuggestionInvalidated( |
- ContentSuggestionsProvider* provider, |
- const ContentSuggestion::ID& suggestion_id) override {} |
- |
- const std::map<Category, CategoryStatus, Category::CompareByID>& statuses() |
- const { |
- return statuses_; |
- } |
- |
- CategoryStatus StatusForCategory(Category category) const { |
- auto it = statuses_.find(category); |
- if (it == statuses_.end()) { |
- return CategoryStatus::NOT_PROVIDED; |
- } |
- return it->second; |
- } |
- |
- const std::vector<ContentSuggestion>& SuggestionsForCategory( |
- Category category) { |
- return suggestions_[category]; |
- } |
- |
- private: |
- std::map<Category, CategoryStatus, Category::CompareByID> statuses_; |
- std::map<Category, std::vector<ContentSuggestion>, Category::CompareByID> |
- suggestions_; |
- |
- DISALLOW_COPY_AND_ASSIGN(FakeContentSuggestionsProviderObserver); |
-}; |
- |
-class FakeImageDecoder : public image_fetcher::ImageDecoder { |
- public: |
- FakeImageDecoder() {} |
- ~FakeImageDecoder() override = default; |
- void DecodeImage( |
- const std::string& image_data, |
- const image_fetcher::ImageDecodedCallback& callback) override { |
- callback.Run(decoded_image_); |
- } |
- |
- void SetDecodedImage(const gfx::Image& image) { decoded_image_ = image; } |
- |
- private: |
- gfx::Image decoded_image_; |
-}; |
- |
-} // namespace |
- |
-class RemoteSuggestionsProviderTest : public ::testing::Test { |
- public: |
- RemoteSuggestionsProviderTest() |
- : params_manager_(ntp_snippets::kStudyName, |
- {{"content_suggestions_backend", |
- kTestContentSuggestionsServerEndpoint}, |
- {"fetching_personalization", "non_personal"}}), |
- fake_url_fetcher_factory_( |
- /*default_factory=*/&failing_url_fetcher_factory_), |
- test_url_(kTestContentSuggestionsServerWithAPIKey), |
- category_ranker_(base::MakeUnique<ConstantCategoryRanker>()), |
- user_classifier_(/*pref_service=*/nullptr), |
- image_fetcher_(nullptr), |
- image_decoder_(nullptr), |
- database_(nullptr) { |
- RemoteSuggestionsProvider::RegisterProfilePrefs( |
- utils_.pref_service()->registry()); |
- RequestThrottler::RegisterProfilePrefs(utils_.pref_service()->registry()); |
- |
- EXPECT_TRUE(database_dir_.CreateUniqueTempDir()); |
- } |
- |
- ~RemoteSuggestionsProviderTest() override { |
- // We need to run the message loop after deleting the database, because |
- // ProtoDatabaseImpl deletes the actual LevelDB asynchronously on the task |
- // runner. Without this, we'd get reports of memory leaks. |
- base::RunLoop().RunUntilIdle(); |
- } |
- |
- // TODO(vitaliii): Rewrite this function to initialize a test class member |
- // instead of creating a new service. |
- std::unique_ptr<RemoteSuggestionsProvider> MakeSnippetsService( |
- bool set_empty_response = true) { |
- auto service = MakeSnippetsServiceWithoutInitialization(); |
- WaitForSnippetsServiceInitialization(service.get(), set_empty_response); |
- return service; |
- } |
- |
- std::unique_ptr<RemoteSuggestionsProvider> |
- MakeSnippetsServiceWithoutInitialization() { |
- scoped_refptr<base::SingleThreadTaskRunner> task_runner( |
- base::ThreadTaskRunnerHandle::Get()); |
- scoped_refptr<net::TestURLRequestContextGetter> request_context_getter = |
- new net::TestURLRequestContextGetter(task_runner.get()); |
- |
- utils_.ResetSigninManager(); |
- std::unique_ptr<NTPSnippetsFetcher> snippets_fetcher = |
- base::MakeUnique<NTPSnippetsFetcher>( |
- utils_.fake_signin_manager(), fake_token_service_.get(), |
- std::move(request_context_getter), utils_.pref_service(), nullptr, |
- base::Bind(&ParseJson), kAPIKey, &user_classifier_); |
- |
- utils_.fake_signin_manager()->SignIn("foo@bar.com"); |
- |
- auto image_fetcher = base::MakeUnique<NiceMock<MockImageFetcher>>(); |
- |
- image_fetcher_ = image_fetcher.get(); |
- EXPECT_CALL(*image_fetcher, SetImageFetcherDelegate(_)); |
- auto image_decoder = base::MakeUnique<FakeImageDecoder>(); |
- image_decoder_ = image_decoder.get(); |
- EXPECT_FALSE(observer_); |
- observer_ = base::MakeUnique<FakeContentSuggestionsProviderObserver>(); |
- auto database = base::MakeUnique<RemoteSuggestionsDatabase>( |
- database_dir_.GetPath(), task_runner); |
- database_ = database.get(); |
- return base::MakeUnique<RemoteSuggestionsProvider>( |
- observer_.get(), utils_.pref_service(), "fr", category_ranker_.get(), |
- &user_classifier_, &scheduler_, std::move(snippets_fetcher), |
- std::move(image_fetcher), std::move(image_decoder), std::move(database), |
- base::MakeUnique<RemoteSuggestionsStatusService>( |
- utils_.fake_signin_manager(), utils_.pref_service())); |
- } |
- |
- void WaitForSnippetsServiceInitialization(RemoteSuggestionsProvider* service, |
- bool set_empty_response) { |
- EXPECT_EQ(RemoteSuggestionsProvider::State::NOT_INITED, service->state_); |
- |
- // Add an initial fetch response, as the service tries to fetch when there |
- // is nothing in the DB. |
- if (set_empty_response) { |
- SetUpFetchResponse(GetTestJson(std::vector<std::string>())); |
- } |
- |
- // TODO(treib): Find a better way to wait for initialization to finish. |
- base::RunLoop().RunUntilIdle(); |
- EXPECT_NE(RemoteSuggestionsProvider::State::NOT_INITED, service->state_); |
- } |
- |
- void ResetSnippetsService(std::unique_ptr<RemoteSuggestionsProvider>* service, |
- bool set_empty_response) { |
- service->reset(); |
- observer_.reset(); |
- *service = MakeSnippetsService(set_empty_response); |
- } |
- |
- void SetCategoryRanker(std::unique_ptr<CategoryRanker> category_ranker) { |
- category_ranker_ = std::move(category_ranker); |
- } |
- |
- ContentSuggestion::ID MakeArticleID(const std::string& id_within_category) { |
- return ContentSuggestion::ID(articles_category(), id_within_category); |
- } |
- |
- Category articles_category() { |
- return Category::FromKnownCategory(KnownCategories::ARTICLES); |
- } |
- |
- ContentSuggestion::ID MakeOtherID(const std::string& id_within_category) { |
- return ContentSuggestion::ID(other_category(), id_within_category); |
- } |
- |
- // TODO(tschumann): Get rid of the convenience other_category() and |
- // unknown_category() helpers -- tests can just define their own. |
- Category other_category() { return Category::FromRemoteCategory(2); } |
- |
- Category unknown_category() { |
- return Category::FromRemoteCategory(kUnknownRemoteCategoryId); |
- } |
- |
- protected: |
- const GURL& test_url() { return test_url_; } |
- FakeContentSuggestionsProviderObserver& observer() { return *observer_; } |
- MockScheduler& mock_scheduler() { return scheduler_; } |
- // TODO(tschumann): Make this a strict-mock. We want to avoid unneccesary |
- // network requests. |
- NiceMock<MockImageFetcher>* image_fetcher() { return image_fetcher_; } |
- FakeImageDecoder* image_decoder() { return image_decoder_; } |
- PrefService* pref_service() { return utils_.pref_service(); } |
- RemoteSuggestionsDatabase* database() { return database_; } |
- |
- // Provide the json to be returned by the fake fetcher. |
- void SetUpFetchResponse(const std::string& json) { |
- fake_url_fetcher_factory_.SetFakeResponse(test_url_, json, net::HTTP_OK, |
- net::URLRequestStatus::SUCCESS); |
- } |
- |
- // Have the fake fetcher fail due to a HTTP error like a 404. |
- void SetUpHttpError() { |
- fake_url_fetcher_factory_.SetFakeResponse(test_url_, /*json=*/std::string(), |
- net::HTTP_NOT_FOUND, |
- net::URLRequestStatus::SUCCESS); |
- } |
- |
- void LoadFromJSONString(RemoteSuggestionsProvider* service, |
- const std::string& json) { |
- SetUpFetchResponse(json); |
- service->FetchSnippets(true); |
- base::RunLoop().RunUntilIdle(); |
- } |
- |
- void LoadMoreFromJSONString(RemoteSuggestionsProvider* service, |
- const Category& category, |
- const std::string& json, |
- const std::set<std::string>& known_ids, |
- FetchDoneCallback callback) { |
- SetUpFetchResponse(json); |
- service->Fetch(category, known_ids, callback); |
- base::RunLoop().RunUntilIdle(); |
- } |
- |
- private: |
- variations::testing::VariationParamsManager params_manager_; |
- test::RemoteSuggestionsTestUtils utils_; |
- base::MessageLoop message_loop_; |
- FailingFakeURLFetcherFactory failing_url_fetcher_factory_; |
- // Instantiation of factory automatically sets itself as URLFetcher's factory. |
- net::FakeURLFetcherFactory fake_url_fetcher_factory_; |
- const GURL test_url_; |
- std::unique_ptr<OAuth2TokenService> fake_token_service_; |
- std::unique_ptr<CategoryRanker> category_ranker_; |
- UserClassifier user_classifier_; |
- NiceMock<MockScheduler> scheduler_; |
- std::unique_ptr<FakeContentSuggestionsProviderObserver> observer_; |
- NiceMock<MockImageFetcher>* image_fetcher_; |
- FakeImageDecoder* image_decoder_; |
- |
- base::ScopedTempDir database_dir_; |
- RemoteSuggestionsDatabase* database_; |
- |
- DISALLOW_COPY_AND_ASSIGN(RemoteSuggestionsProviderTest); |
-}; |
- |
-TEST_F(RemoteSuggestionsProviderTest, ScheduleOnStart) { |
- // We should get two |Schedule| calls: The first when initialization |
- // completes, the second one after the automatic (since the service doesn't |
- // have any data yet) fetch finishes. |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(2); |
- EXPECT_CALL(mock_scheduler(), Unschedule()).Times(0); |
- auto service = MakeSnippetsService(); |
- |
- // When we have no snippets are all, loading the service initiates a fetch. |
- EXPECT_EQ("OK", service->snippets_fetcher()->last_status()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, DontRescheduleOnStart) { |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(2); |
- EXPECT_CALL(mock_scheduler(), Unschedule()).Times(0); |
- SetUpFetchResponse(GetTestJson({GetSnippet()})); |
- auto service = MakeSnippetsService(/*set_empty_response=*/false); |
- |
- // When recreating the service, we should not get any |Schedule| calls: |
- // The tasks are already scheduled with the correct intervals, so nothing on |
- // initialization, and the service has data from the DB, so no automatic fetch |
- // should happen. |
- Mock::VerifyAndClearExpectations(&mock_scheduler()); |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(0); |
- EXPECT_CALL(mock_scheduler(), Unschedule()).Times(0); |
- ResetSnippetsService(&service, /*set_empty_response=*/true); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, RescheduleAfterSuccessfulFetch) { |
- // We should get two |Schedule| calls: The first when initialization |
- // completes, the second one after the automatic (since the service doesn't |
- // have any data yet) fetch finishes. |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(2); |
- auto service = MakeSnippetsService(); |
- |
- // A successful fetch should trigger another |Schedule|. |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)); |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippet()})); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, DontRescheduleAfterFailedFetch) { |
- // We should get two |Schedule| calls: The first when initialization |
- // completes, the second one after the automatic (since the service doesn't |
- // have any data yet) fetch finishes. |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(2); |
- auto service = MakeSnippetsService(); |
- |
- // A failed fetch should NOT trigger another |Schedule|. |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(0); |
- LoadFromJSONString(service.get(), GetTestJson({GetInvalidSnippet()})); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, IgnoreRescheduleBeforeInit) { |
- // We should get two |Schedule| calls: The first when initialization |
- // completes, the second one after the automatic (since the service doesn't |
- // have any data yet) fetch finishes. |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(2); |
- // The |RescheduleFetching| call shouldn't do anything (in particular not |
- // result in an |Unschedule|), since the service isn't initialized yet. |
- EXPECT_CALL(mock_scheduler(), Unschedule()).Times(0); |
- auto service = MakeSnippetsServiceWithoutInitialization(); |
- service->RescheduleFetching(false); |
- WaitForSnippetsServiceInitialization(service.get(), |
- /*set_empty_response=*/true); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, HandleForcedRescheduleBeforeInit) { |
- { |
- InSequence s; |
- // The |RescheduleFetching| call with force=true should result in an |
- // |Unschedule|, since the service isn't initialized yet. |
- EXPECT_CALL(mock_scheduler(), Unschedule()).Times(1); |
- // We should get two |Schedule| calls: The first when initialization |
- // completes, the second one after the automatic (since the service doesn't |
- // have any data yet) fetch finishes. |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(2); |
- } |
- auto service = MakeSnippetsServiceWithoutInitialization(); |
- service->RescheduleFetching(true); |
- WaitForSnippetsServiceInitialization(service.get(), |
- /*set_empty_response=*/true); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, RescheduleOnStateChange) { |
- { |
- InSequence s; |
- // Initial startup. |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(2); |
- // Service gets disabled. |
- EXPECT_CALL(mock_scheduler(), Unschedule()); |
- // Service gets enabled again. |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(2); |
- } |
- auto service = MakeSnippetsService(); |
- ASSERT_TRUE(service->ready()); |
- |
- service->OnStatusChanged(RemoteSuggestionsStatus::ENABLED_AND_SIGNED_IN, |
- RemoteSuggestionsStatus::EXPLICITLY_DISABLED); |
- ASSERT_FALSE(service->ready()); |
- base::RunLoop().RunUntilIdle(); |
- |
- service->OnStatusChanged(RemoteSuggestionsStatus::EXPLICITLY_DISABLED, |
- RemoteSuggestionsStatus::ENABLED_AND_SIGNED_OUT); |
- ASSERT_TRUE(service->ready()); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, DontUnscheduleOnShutdown) { |
- EXPECT_CALL(mock_scheduler(), Schedule(_, _)).Times(2); |
- EXPECT_CALL(mock_scheduler(), Unschedule()).Times(0); |
- |
- auto service = MakeSnippetsService(); |
- |
- service.reset(); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, Full) { |
- std::string json_str(GetTestJson({GetSnippet()})); |
- |
- auto service = MakeSnippetsService(); |
- |
- LoadFromJSONString(service.get(), json_str); |
- |
- ASSERT_THAT(observer().SuggestionsForCategory(articles_category()), |
- SizeIs(1)); |
- ASSERT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- |
- const ContentSuggestion& suggestion = |
- observer().SuggestionsForCategory(articles_category()).front(); |
- |
- EXPECT_EQ(MakeArticleID(kSnippetUrl), suggestion.id()); |
- EXPECT_EQ(kSnippetTitle, base::UTF16ToUTF8(suggestion.title())); |
- EXPECT_EQ(kSnippetText, base::UTF16ToUTF8(suggestion.snippet_text())); |
- EXPECT_EQ(GetDefaultCreationTime(), suggestion.publish_date()); |
- EXPECT_EQ(kSnippetPublisherName, |
- base::UTF16ToUTF8(suggestion.publisher_name())); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, CategoryTitle) { |
- const base::string16 test_default_title = |
- base::UTF8ToUTF16(kTestJsonDefaultCategoryTitle); |
- |
- // Don't send an initial response -- we want to test what happens without any |
- // server status. |
- auto service = MakeSnippetsService(/*set_empty_response=*/false); |
- |
- // The articles category should be there by default, and have a title. |
- CategoryInfo info_before = service->GetCategoryInfo(articles_category()); |
- ASSERT_THAT(info_before.title(), Not(IsEmpty())); |
- ASSERT_THAT(info_before.title(), Not(Eq(test_default_title))); |
- EXPECT_THAT(info_before.has_more_action(), Eq(true)); |
- EXPECT_THAT(info_before.has_reload_action(), Eq(true)); |
- EXPECT_THAT(info_before.has_view_all_action(), Eq(false)); |
- EXPECT_THAT(info_before.show_if_empty(), Eq(true)); |
- |
- std::string json_str_with_title(GetTestJson({GetSnippet()})); |
- LoadFromJSONString(service.get(), json_str_with_title); |
- |
- ASSERT_THAT(observer().SuggestionsForCategory(articles_category()), |
- SizeIs(1)); |
- ASSERT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- |
- // The response contained a title, |kTestJsonDefaultCategoryTitle|. |
- // Make sure we updated the title in the CategoryInfo. |
- CategoryInfo info_with_title = service->GetCategoryInfo(articles_category()); |
- EXPECT_THAT(info_before.title(), Not(Eq(info_with_title.title()))); |
- EXPECT_THAT(test_default_title, Eq(info_with_title.title())); |
- EXPECT_THAT(info_before.has_more_action(), Eq(true)); |
- EXPECT_THAT(info_before.has_reload_action(), Eq(true)); |
- EXPECT_THAT(info_before.has_view_all_action(), Eq(false)); |
- EXPECT_THAT(info_before.show_if_empty(), Eq(true)); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, MultipleCategories) { |
- auto service = MakeSnippetsService(); |
- std::string json_str = |
- MultiCategoryJsonBuilder() |
- .AddCategory({GetSnippetN(0)}, /*remote_category_id=*/1) |
- .AddCategory({GetSnippetN(1)}, /*remote_category_id=*/2) |
- .Build(); |
- LoadFromJSONString(service.get(), json_str); |
- |
- ASSERT_THAT(observer().statuses(), |
- Eq(std::map<Category, CategoryStatus, Category::CompareByID>{ |
- {articles_category(), CategoryStatus::AVAILABLE}, |
- {other_category(), CategoryStatus::AVAILABLE}, |
- })); |
- |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- EXPECT_THAT(service->GetSnippetsForTesting(other_category()), SizeIs(1)); |
- |
- ASSERT_THAT(observer().SuggestionsForCategory(articles_category()), |
- SizeIs(1)); |
- |
- ASSERT_THAT(observer().SuggestionsForCategory(other_category()), SizeIs(1)); |
- |
- { |
- const ContentSuggestion& suggestion = |
- observer().SuggestionsForCategory(articles_category()).front(); |
- EXPECT_EQ(MakeArticleID(std::string(kSnippetUrl) + "/0"), suggestion.id()); |
- EXPECT_EQ(kSnippetTitle, base::UTF16ToUTF8(suggestion.title())); |
- EXPECT_EQ(kSnippetText, base::UTF16ToUTF8(suggestion.snippet_text())); |
- EXPECT_EQ(GetDefaultCreationTime(), suggestion.publish_date()); |
- EXPECT_EQ(kSnippetPublisherName, |
- base::UTF16ToUTF8(suggestion.publisher_name())); |
- } |
- |
- { |
- const ContentSuggestion& suggestion = |
- observer().SuggestionsForCategory(other_category()).front(); |
- EXPECT_EQ(MakeOtherID(std::string(kSnippetUrl) + "/1"), suggestion.id()); |
- EXPECT_EQ(kSnippetTitle, base::UTF16ToUTF8(suggestion.title())); |
- EXPECT_EQ(kSnippetText, base::UTF16ToUTF8(suggestion.snippet_text())); |
- EXPECT_EQ(GetDefaultCreationTime(), suggestion.publish_date()); |
- EXPECT_EQ(kSnippetPublisherName, |
- base::UTF16ToUTF8(suggestion.publisher_name())); |
- } |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ArticleCategoryInfo) { |
- auto service = MakeSnippetsService(); |
- CategoryInfo article_info = service->GetCategoryInfo(articles_category()); |
- EXPECT_THAT(article_info.has_more_action(), Eq(true)); |
- EXPECT_THAT(article_info.has_reload_action(), Eq(true)); |
- EXPECT_THAT(article_info.has_view_all_action(), Eq(false)); |
- EXPECT_THAT(article_info.show_if_empty(), Eq(true)); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ExperimentalCategoryInfo) { |
- auto service = MakeSnippetsService(); |
- std::string json_str = |
- MultiCategoryJsonBuilder() |
- .AddCategory({GetSnippetN(0)}, /*remote_category_id=*/1) |
- .AddCategory({GetSnippetN(1)}, kUnknownRemoteCategoryId) |
- .Build(); |
- // Load data with multiple categories so that a new experimental category gets |
- // registered. |
- LoadFromJSONString(service.get(), json_str); |
- |
- CategoryInfo info = service->GetCategoryInfo(unknown_category()); |
- EXPECT_THAT(info.has_more_action(), Eq(false)); |
- EXPECT_THAT(info.has_reload_action(), Eq(false)); |
- EXPECT_THAT(info.has_view_all_action(), Eq(false)); |
- EXPECT_THAT(info.show_if_empty(), Eq(false)); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, AddRemoteCategoriesToCategoryRanker) { |
- auto mock_ranker = base::MakeUnique<MockCategoryRanker>(); |
- MockCategoryRanker* raw_mock_ranker = mock_ranker.get(); |
- SetCategoryRanker(std::move(mock_ranker)); |
- std::string json_str = |
- MultiCategoryJsonBuilder() |
- .AddCategory({GetSnippetN(0)}, /*remote_category_id=*/11) |
- .AddCategory({GetSnippetN(1)}, /*remote_category_id=*/13) |
- .AddCategory({GetSnippetN(2)}, /*remote_category_id=*/12) |
- .Build(); |
- SetUpFetchResponse(json_str); |
- { |
- // The order of categories is determined by the order in which they are |
- // added. Thus, the latter is tested here. |
- InSequence s; |
- EXPECT_CALL(*raw_mock_ranker, |
- AppendCategoryIfNecessary(Category::FromRemoteCategory(11))); |
- EXPECT_CALL(*raw_mock_ranker, |
- AppendCategoryIfNecessary(Category::FromRemoteCategory(13))); |
- EXPECT_CALL(*raw_mock_ranker, |
- AppendCategoryIfNecessary(Category::FromRemoteCategory(12))); |
- } |
- auto service = MakeSnippetsService(/*set_empty_response=*/false); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, PersistCategoryInfos) { |
- auto service = MakeSnippetsService(); |
- // TODO(vitaliii): Use |articles_category()| instead of constant ID below. |
- std::string json_str = |
- MultiCategoryJsonBuilder() |
- .AddCategoryWithCustomTitle( |
- {GetSnippetN(0)}, /*remote_category_id=*/1, "Articles for You") |
- .AddCategoryWithCustomTitle({GetSnippetN(1)}, |
- kUnknownRemoteCategoryId, "Other Things") |
- .Build(); |
- LoadFromJSONString(service.get(), json_str); |
- |
- ASSERT_EQ(observer().StatusForCategory(articles_category()), |
- CategoryStatus::AVAILABLE); |
- ASSERT_EQ(observer().StatusForCategory(unknown_category()), |
- CategoryStatus::AVAILABLE); |
- |
- CategoryInfo info_articles_before = |
- service->GetCategoryInfo(articles_category()); |
- CategoryInfo info_unknown_before = |
- service->GetCategoryInfo(unknown_category()); |
- |
- // Recreate the service to simulate a Chrome restart. |
- ResetSnippetsService(&service, /*set_empty_response=*/true); |
- |
- // The categories should have been restored. |
- ASSERT_NE(observer().StatusForCategory(articles_category()), |
- CategoryStatus::NOT_PROVIDED); |
- ASSERT_NE(observer().StatusForCategory(unknown_category()), |
- CategoryStatus::NOT_PROVIDED); |
- |
- EXPECT_EQ(observer().StatusForCategory(articles_category()), |
- CategoryStatus::AVAILABLE); |
- EXPECT_EQ(observer().StatusForCategory(unknown_category()), |
- CategoryStatus::AVAILABLE); |
- |
- CategoryInfo info_articles_after = |
- service->GetCategoryInfo(articles_category()); |
- CategoryInfo info_unknown_after = |
- service->GetCategoryInfo(unknown_category()); |
- |
- EXPECT_EQ(info_articles_before.title(), info_articles_after.title()); |
- EXPECT_EQ(info_unknown_before.title(), info_unknown_after.title()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, PersistRemoteCategoryOrder) { |
- // We create a service with a normal ranker to store the order. |
- std::string json_str = |
- MultiCategoryJsonBuilder() |
- .AddCategory({GetSnippetN(0)}, /*remote_category_id=*/11) |
- .AddCategory({GetSnippetN(1)}, /*remote_category_id=*/13) |
- .AddCategory({GetSnippetN(2)}, /*remote_category_id=*/12) |
- .Build(); |
- SetUpFetchResponse(json_str); |
- auto service = MakeSnippetsService(/*set_empty_response=*/false); |
- |
- // We manually recreate the service to simulate Chrome restart and enforce a |
- // mock ranker. The response is cleared to ensure that the order is not |
- // fetched. |
- SetUpFetchResponse(""); |
- auto mock_ranker = base::MakeUnique<MockCategoryRanker>(); |
- MockCategoryRanker* raw_mock_ranker = mock_ranker.get(); |
- SetCategoryRanker(std::move(mock_ranker)); |
- { |
- // The order of categories is determined by the order in which they are |
- // added. Thus, the latter is tested here. |
- InSequence s; |
- // Article category always exists and, therefore, it is stored in prefs too. |
- EXPECT_CALL(*raw_mock_ranker, |
- AppendCategoryIfNecessary(articles_category())); |
- |
- EXPECT_CALL(*raw_mock_ranker, |
- AppendCategoryIfNecessary(Category::FromRemoteCategory(11))); |
- EXPECT_CALL(*raw_mock_ranker, |
- AppendCategoryIfNecessary(Category::FromRemoteCategory(13))); |
- EXPECT_CALL(*raw_mock_ranker, |
- AppendCategoryIfNecessary(Category::FromRemoteCategory(12))); |
- } |
- ResetSnippetsService(&service, /*set_empty_response=*/false); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, PersistSuggestions) { |
- auto service = MakeSnippetsService(); |
- std::string json_str = |
- MultiCategoryJsonBuilder() |
- .AddCategory({GetSnippetN(0)}, /*remote_category_id=*/1) |
- .AddCategory({GetSnippetN(2)}, /*remote_category_id=*/2) |
- .Build(); |
- LoadFromJSONString(service.get(), json_str); |
- |
- ASSERT_THAT(observer().SuggestionsForCategory(articles_category()), |
- SizeIs(1)); |
- ASSERT_THAT(observer().SuggestionsForCategory(other_category()), SizeIs(1)); |
- |
- // Recreate the service to simulate a Chrome restart. |
- ResetSnippetsService(&service, /*set_empty_response=*/true); |
- |
- // The suggestions in both categories should have been restored. |
- EXPECT_THAT(observer().SuggestionsForCategory(articles_category()), |
- SizeIs(1)); |
- EXPECT_THAT(observer().SuggestionsForCategory(other_category()), SizeIs(1)); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, DontNotifyIfNotAvailable) { |
- // Get some suggestions into the database. |
- auto service = MakeSnippetsService(); |
- std::string json_str = |
- MultiCategoryJsonBuilder() |
- .AddCategory({GetSnippetN(0)}, |
- /*remote_category_id=*/1) |
- .AddCategory({GetSnippetN(1)}, /*remote_category_id=*/2) |
- .Build(); |
- LoadFromJSONString(service.get(), json_str); |
- |
- ASSERT_THAT(observer().SuggestionsForCategory(articles_category()), |
- SizeIs(1)); |
- ASSERT_THAT(observer().SuggestionsForCategory(other_category()), SizeIs(1)); |
- |
- service.reset(); |
- |
- // Set the pref that disables remote suggestions. |
- pref_service()->SetBoolean(prefs::kEnableSnippets, false); |
- |
- // Recreate the service to simulate a Chrome start. |
- ResetSnippetsService(&service, /*set_empty_response=*/true); |
- |
- ASSERT_THAT(RemoteSuggestionsProvider::State::DISABLED, Eq(service->state_)); |
- |
- // Now the observer should not have received any suggestions. |
- EXPECT_THAT(observer().SuggestionsForCategory(articles_category()), |
- IsEmpty()); |
- EXPECT_THAT(observer().SuggestionsForCategory(other_category()), IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, Clear) { |
- auto service = MakeSnippetsService(); |
- |
- std::string json_str(GetTestJson({GetSnippet()})); |
- |
- LoadFromJSONString(service.get(), json_str); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- |
- service->ClearCachedSuggestions(articles_category()); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ReplaceSnippets) { |
- auto service = MakeSnippetsService(); |
- |
- std::string first("http://first"); |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippetWithUrl(first)})); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), |
- ElementsAre(IdEq(first))); |
- |
- std::string second("http://second"); |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippetWithUrl(second)})); |
- // The snippets loaded last replace all that was loaded previously. |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), |
- ElementsAre(IdEq(second))); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, LoadsAdditionalSnippets) { |
- auto service = MakeSnippetsService(); |
- |
- LoadFromJSONString(service.get(), |
- GetTestJson({GetSnippetWithUrl("http://first")})); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), |
- ElementsAre(IdEq("http://first"))); |
- |
- auto expect_only_second_suggestion_received = base::Bind([]( |
- Status status, std::vector<ContentSuggestion> suggestions) { |
- EXPECT_THAT(suggestions, SizeIs(1)); |
- EXPECT_THAT(suggestions[0].id().id_within_category(), Eq("http://second")); |
- }); |
- LoadMoreFromJSONString(service.get(), articles_category(), |
- GetTestJson({GetSnippetWithUrl("http://second")}), |
- /*known_ids=*/std::set<std::string>(), |
- expect_only_second_suggestion_received); |
- |
- // Verify we can resolve the image of the new snippets. |
- ServeImageCallback cb = |
- base::Bind(&ServeOneByOneImage, &service->GetImageFetcherForTesting()); |
- EXPECT_CALL(*image_fetcher(), StartOrQueueNetworkRequest(_, _, _)) |
- .Times(2) |
- .WillRepeatedly(WithArgs<0, 2>(Invoke(&cb, &ServeImageCallback::Run))); |
- image_decoder()->SetDecodedImage(gfx::test::CreateImage(1, 1)); |
- gfx::Image image = FetchImage(service.get(), MakeArticleID("http://first")); |
- EXPECT_FALSE(image.IsEmpty()); |
- EXPECT_EQ(1, image.Width()); |
- |
- image = FetchImage(service.get(), MakeArticleID("http://second")); |
- EXPECT_FALSE(image.IsEmpty()); |
- EXPECT_EQ(1, image.Width()); |
- |
- // Verify that the observer received the update as well. We should see the |
- // newly-fetched items filled up with existing ones. |
- EXPECT_THAT(observer().SuggestionsForCategory(articles_category()), |
- ElementsAre(IdWithinCategoryEq("http://first"), |
- IdWithinCategoryEq("http://second"))); |
-} |
- |
-// The tests TestMergingFetchedMoreSnippetsFillup and |
-// TestMergingFetchedMoreSnippetsReplaceAll simulate the following user story: |
-// 1) fetch suggestions in NTP A |
-// 2) fetch more suggestions in NTP A. |
-// 3) open new NTP B: See the last 10 results visible in step 2). |
-// 4) fetch more suggestions in NTP B. Make sure no results from step 1) which |
-// were superseded in step 2) get merged back in again. |
-// TODO(tschumann): Test step 4) on a higher level instead of peeking into the |
-// internal 'dismissed' data. The proper check is to make sure we tell the |
-// backend to exclude these snippets. |
-TEST_F(RemoteSuggestionsProviderTest, TestMergingFetchedMoreSnippetsFillup) { |
- auto service = MakeSnippetsService(/*set_empty_response=*/false); |
- LoadFromJSONString( |
- service.get(), |
- GetTestJson( |
- {GetSnippetWithUrl("http://id-1"), GetSnippetWithUrl("http://id-2"), |
- GetSnippetWithUrl("http://id-3"), GetSnippetWithUrl("http://id-4"), |
- GetSnippetWithUrl("http://id-5"), GetSnippetWithUrl("http://id-6"), |
- GetSnippetWithUrl("http://id-7"), GetSnippetWithUrl("http://id-8"), |
- GetSnippetWithUrl("http://id-9"), |
- GetSnippetWithUrl("http://id-10")})); |
- EXPECT_THAT( |
- observer().SuggestionsForCategory(articles_category()), |
- ElementsAre( |
- IdWithinCategoryEq("http://id-1"), IdWithinCategoryEq("http://id-2"), |
- IdWithinCategoryEq("http://id-3"), IdWithinCategoryEq("http://id-4"), |
- IdWithinCategoryEq("http://id-5"), IdWithinCategoryEq("http://id-6"), |
- IdWithinCategoryEq("http://id-7"), IdWithinCategoryEq("http://id-8"), |
- IdWithinCategoryEq("http://id-9"), |
- IdWithinCategoryEq("http://id-10"))); |
- |
- auto expect_receiving_two_new_snippets = |
- base::Bind([](Status status, std::vector<ContentSuggestion> suggestions) { |
- ASSERT_THAT(suggestions, SizeIs(2)); |
- EXPECT_THAT(suggestions[0], IdWithinCategoryEq("http://more-id-1")); |
- EXPECT_THAT(suggestions[1], IdWithinCategoryEq("http://more-id-2")); |
- }); |
- LoadMoreFromJSONString( |
- service.get(), articles_category(), |
- GetTestJson({GetSnippetWithUrl("http://more-id-1"), |
- GetSnippetWithUrl("http://more-id-2")}), |
- /*known_ids=*/{"http://id-1", "http://id-2", "http://id-3", "http://id-4", |
- "http://id-5", "http://id-6", "http://id-7", "http://id-8", |
- "http://id-9", "http://id-10"}, |
- expect_receiving_two_new_snippets); |
- |
- // Verify that the observer received the update as well. We should see the |
- // newly-fetched items filled up with existing ones. The merging is done |
- // mimicking a scrolling behavior. |
- EXPECT_THAT( |
- observer().SuggestionsForCategory(articles_category()), |
- ElementsAre( |
- IdWithinCategoryEq("http://id-3"), IdWithinCategoryEq("http://id-4"), |
- IdWithinCategoryEq("http://id-5"), IdWithinCategoryEq("http://id-6"), |
- IdWithinCategoryEq("http://id-7"), IdWithinCategoryEq("http://id-8"), |
- IdWithinCategoryEq("http://id-9"), IdWithinCategoryEq("http://id-10"), |
- IdWithinCategoryEq("http://more-id-1"), |
- IdWithinCategoryEq("http://more-id-2"))); |
- // Verify the superseded suggestions got marked as dismissed. |
- EXPECT_THAT(service->GetDismissedSnippetsForTesting(articles_category()), |
- ElementsAre(IdEq("http://id-1"), IdEq("http://id-2"))); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, |
- TestMergingFetchedMoreSnippetsReplaceAll) { |
- auto service = MakeSnippetsService(/*set_empty_response=*/false); |
- LoadFromJSONString( |
- service.get(), |
- GetTestJson( |
- {GetSnippetWithUrl("http://id-1"), GetSnippetWithUrl("http://id-2"), |
- GetSnippetWithUrl("http://id-3"), GetSnippetWithUrl("http://id-4"), |
- GetSnippetWithUrl("http://id-5"), GetSnippetWithUrl("http://id-6"), |
- GetSnippetWithUrl("http://id-7"), GetSnippetWithUrl("http://id-8"), |
- GetSnippetWithUrl("http://id-9"), |
- GetSnippetWithUrl("http://id-10")})); |
- EXPECT_THAT( |
- observer().SuggestionsForCategory(articles_category()), |
- ElementsAre( |
- IdWithinCategoryEq("http://id-1"), IdWithinCategoryEq("http://id-2"), |
- IdWithinCategoryEq("http://id-3"), IdWithinCategoryEq("http://id-4"), |
- IdWithinCategoryEq("http://id-5"), IdWithinCategoryEq("http://id-6"), |
- IdWithinCategoryEq("http://id-7"), IdWithinCategoryEq("http://id-8"), |
- IdWithinCategoryEq("http://id-9"), |
- IdWithinCategoryEq("http://id-10"))); |
- |
- auto expect_receiving_ten_new_snippets = |
- base::Bind([](Status status, std::vector<ContentSuggestion> suggestions) { |
- EXPECT_THAT(suggestions, ElementsAre( |
- IdWithinCategoryEq("http://more-id-1"), |
- IdWithinCategoryEq("http://more-id-2"), |
- IdWithinCategoryEq("http://more-id-3"), |
- IdWithinCategoryEq("http://more-id-4"), |
- IdWithinCategoryEq("http://more-id-5"), |
- IdWithinCategoryEq("http://more-id-6"), |
- IdWithinCategoryEq("http://more-id-7"), |
- IdWithinCategoryEq("http://more-id-8"), |
- IdWithinCategoryEq("http://more-id-9"), |
- IdWithinCategoryEq("http://more-id-10"))); |
- }); |
- LoadMoreFromJSONString( |
- service.get(), articles_category(), |
- GetTestJson({GetSnippetWithUrl("http://more-id-1"), |
- GetSnippetWithUrl("http://more-id-2"), |
- GetSnippetWithUrl("http://more-id-3"), |
- GetSnippetWithUrl("http://more-id-4"), |
- GetSnippetWithUrl("http://more-id-5"), |
- GetSnippetWithUrl("http://more-id-6"), |
- GetSnippetWithUrl("http://more-id-7"), |
- GetSnippetWithUrl("http://more-id-8"), |
- GetSnippetWithUrl("http://more-id-9"), |
- GetSnippetWithUrl("http://more-id-10")}), |
- /*known_ids=*/{"http://id-1", "http://id-2", "http://id-3", "http://id-4", |
- "http://id-5", "http://id-6", "http://id-7", "http://id-8", |
- "http://id-9", "http://id-10"}, |
- expect_receiving_ten_new_snippets); |
- EXPECT_THAT(observer().SuggestionsForCategory(articles_category()), |
- ElementsAre(IdWithinCategoryEq("http://more-id-1"), |
- IdWithinCategoryEq("http://more-id-2"), |
- IdWithinCategoryEq("http://more-id-3"), |
- IdWithinCategoryEq("http://more-id-4"), |
- IdWithinCategoryEq("http://more-id-5"), |
- IdWithinCategoryEq("http://more-id-6"), |
- IdWithinCategoryEq("http://more-id-7"), |
- IdWithinCategoryEq("http://more-id-8"), |
- IdWithinCategoryEq("http://more-id-9"), |
- IdWithinCategoryEq("http://more-id-10"))); |
- // Verify the superseded suggestions got marked as dismissed. |
- EXPECT_THAT( |
- service->GetDismissedSnippetsForTesting(articles_category()), |
- ElementsAre(IdEq("http://id-1"), IdEq("http://id-2"), IdEq("http://id-3"), |
- IdEq("http://id-4"), IdEq("http://id-5"), IdEq("http://id-6"), |
- IdEq("http://id-7"), IdEq("http://id-8"), IdEq("http://id-9"), |
- IdEq("http://id-10"))); |
-} |
- |
-// TODO(tschumann): We don't have test making sure the NTPSnippetsFetcher |
-// actually gets the proper parameters. Add tests with an injected |
-// NTPSnippetsFetcher to verify the parameters, including proper handling of |
-// dismissed and known_ids. |
- |
-namespace { |
- |
-// Workaround for gMock's lack of support for movable types. |
-void SuggestionsLoaded( |
- MockFunction<void(Status, const std::vector<ContentSuggestion>&)>* loaded, |
- Status status, |
- std::vector<ContentSuggestion> suggestions) { |
- loaded->Call(status, suggestions); |
-} |
- |
-} // namespace |
- |
-TEST_F(RemoteSuggestionsProviderTest, ReturnFetchRequestEmptyBeforeInit) { |
- auto service = MakeSnippetsServiceWithoutInitialization(); |
- MockFunction<void(Status, const std::vector<ContentSuggestion>&)> loaded; |
- EXPECT_CALL(loaded, Call(HasCode(StatusCode::TEMPORARY_ERROR), IsEmpty())); |
- service->Fetch(articles_category(), std::set<std::string>(), |
- base::Bind(&SuggestionsLoaded, &loaded)); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ReturnTemporaryErrorForInvalidJson) { |
- auto service = MakeSnippetsService(); |
- |
- MockFunction<void(Status, const std::vector<ContentSuggestion>&)> loaded; |
- EXPECT_CALL(loaded, Call(HasCode(StatusCode::TEMPORARY_ERROR), IsEmpty())); |
- LoadMoreFromJSONString(service.get(), articles_category(), |
- "invalid json string}]}", |
- /*known_ids=*/std::set<std::string>(), |
- base::Bind(&SuggestionsLoaded, &loaded)); |
- EXPECT_THAT(service->snippets_fetcher()->last_status(), |
- StartsWith("Received invalid JSON")); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ReturnTemporaryErrorForInvalidSnippet) { |
- auto service = MakeSnippetsService(); |
- |
- MockFunction<void(Status, const std::vector<ContentSuggestion>&)> loaded; |
- EXPECT_CALL(loaded, Call(HasCode(StatusCode::TEMPORARY_ERROR), IsEmpty())); |
- LoadMoreFromJSONString(service.get(), articles_category(), |
- GetTestJson({GetIncompleteSnippet()}), |
- /*known_ids=*/std::set<std::string>(), |
- base::Bind(&SuggestionsLoaded, &loaded)); |
- EXPECT_THAT(service->snippets_fetcher()->last_status(), |
- StartsWith("Invalid / empty list")); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ReturnTemporaryErrorForRequestFailure) { |
- // Created SnippetsService will fail by default with unsuccessful request. |
- auto service = MakeSnippetsService(/*set_empty_response=*/false); |
- |
- MockFunction<void(Status, const std::vector<ContentSuggestion>&)> loaded; |
- EXPECT_CALL(loaded, Call(HasCode(StatusCode::TEMPORARY_ERROR), IsEmpty())); |
- service->Fetch(articles_category(), |
- /*known_ids=*/std::set<std::string>(), |
- base::Bind(&SuggestionsLoaded, &loaded)); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ReturnTemporaryErrorForHttpFailure) { |
- auto service = MakeSnippetsService(); |
- SetUpHttpError(); |
- |
- MockFunction<void(Status, const std::vector<ContentSuggestion>&)> loaded; |
- EXPECT_CALL(loaded, Call(HasCode(StatusCode::TEMPORARY_ERROR), IsEmpty())); |
- service->Fetch(articles_category(), |
- /*known_ids=*/std::set<std::string>(), |
- base::Bind(&SuggestionsLoaded, &loaded)); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, LoadInvalidJson) { |
- auto service = MakeSnippetsService(); |
- |
- LoadFromJSONString(service.get(), GetTestJson({GetInvalidSnippet()})); |
- EXPECT_THAT(service->snippets_fetcher()->last_status(), |
- StartsWith("Received invalid JSON")); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, LoadInvalidJsonWithExistingSnippets) { |
- auto service = MakeSnippetsService(); |
- |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippet()})); |
- ASSERT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- ASSERT_EQ("OK", service->snippets_fetcher()->last_status()); |
- |
- LoadFromJSONString(service.get(), GetTestJson({GetInvalidSnippet()})); |
- EXPECT_THAT(service->snippets_fetcher()->last_status(), |
- StartsWith("Received invalid JSON")); |
- // This should not have changed the existing snippets. |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, LoadIncompleteJson) { |
- auto service = MakeSnippetsService(); |
- |
- LoadFromJSONString(service.get(), GetTestJson({GetIncompleteSnippet()})); |
- EXPECT_EQ("Invalid / empty list.", |
- service->snippets_fetcher()->last_status()); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, LoadIncompleteJsonWithExistingSnippets) { |
- auto service = MakeSnippetsService(); |
- |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippet()})); |
- ASSERT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- |
- LoadFromJSONString(service.get(), GetTestJson({GetIncompleteSnippet()})); |
- EXPECT_EQ("Invalid / empty list.", |
- service->snippets_fetcher()->last_status()); |
- // This should not have changed the existing snippets. |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, Dismiss) { |
- auto service = MakeSnippetsService(); |
- |
- std::string json_str( |
- GetTestJson({GetSnippetWithSources("http://site.com", "Source 1", "")})); |
- |
- LoadFromJSONString(service.get(), json_str); |
- |
- ASSERT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- // Load the image to store it in the database. |
- ServeImageCallback cb = |
- base::Bind(&ServeOneByOneImage, &service->GetImageFetcherForTesting()); |
- EXPECT_CALL(*image_fetcher(), StartOrQueueNetworkRequest(_, _, _)) |
- .WillOnce(WithArgs<0, 2>(Invoke(&cb, &ServeImageCallback::Run))); |
- image_decoder()->SetDecodedImage(gfx::test::CreateImage(1, 1)); |
- gfx::Image image = FetchImage(service.get(), MakeArticleID(kSnippetUrl)); |
- EXPECT_FALSE(image.IsEmpty()); |
- EXPECT_EQ(1, image.Width()); |
- |
- // Dismissing a non-existent snippet shouldn't do anything. |
- service->DismissSuggestion(MakeArticleID("http://othersite.com")); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- |
- // Dismiss the snippet. |
- service->DismissSuggestion(MakeArticleID(kSnippetUrl)); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
- |
- // Verify we can still load the image of the discarded snippet (other NTPs |
- // might still reference it). This should come from the database -- no network |
- // fetch necessary. |
- image_decoder()->SetDecodedImage(gfx::test::CreateImage(1, 1)); |
- image = FetchImage(service.get(), MakeArticleID(kSnippetUrl)); |
- EXPECT_FALSE(image.IsEmpty()); |
- EXPECT_EQ(1, image.Width()); |
- |
- // Make sure that fetching the same snippet again does not re-add it. |
- LoadFromJSONString(service.get(), json_str); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
- |
- // The snippet should stay dismissed even after re-creating the service. |
- ResetSnippetsService(&service, /*set_empty_response=*/true); |
- LoadFromJSONString(service.get(), json_str); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
- |
- // The snippet can be added again after clearing dismissed snippets. |
- service->ClearDismissedSuggestionsForDebugging(articles_category()); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
- LoadFromJSONString(service.get(), json_str); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, GetDismissed) { |
- auto service = MakeSnippetsService(); |
- |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippet()})); |
- |
- service->DismissSuggestion(MakeArticleID(kSnippetUrl)); |
- |
- service->GetDismissedSuggestionsForDebugging( |
- articles_category(), |
- base::Bind( |
- [](RemoteSuggestionsProvider* service, |
- RemoteSuggestionsProviderTest* test, |
- std::vector<ContentSuggestion> dismissed_suggestions) { |
- EXPECT_EQ(1u, dismissed_suggestions.size()); |
- for (auto& suggestion : dismissed_suggestions) { |
- EXPECT_EQ(test->MakeArticleID(kSnippetUrl), suggestion.id()); |
- } |
- }, |
- service.get(), this)); |
- base::RunLoop().RunUntilIdle(); |
- |
- // There should be no dismissed snippet after clearing the list. |
- service->ClearDismissedSuggestionsForDebugging(articles_category()); |
- service->GetDismissedSuggestionsForDebugging( |
- articles_category(), |
- base::Bind( |
- [](RemoteSuggestionsProvider* service, |
- RemoteSuggestionsProviderTest* test, |
- std::vector<ContentSuggestion> dismissed_suggestions) { |
- EXPECT_EQ(0u, dismissed_suggestions.size()); |
- }, |
- service.get(), this)); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, CreationTimestampParseFail) { |
- auto service = MakeSnippetsService(); |
- |
- std::string json = |
- GetSnippetWithTimes(GetDefaultCreationTime(), GetDefaultExpirationTime()); |
- base::ReplaceFirstSubstringAfterOffset( |
- &json, 0, FormatTime(GetDefaultCreationTime()), "aaa1448459205"); |
- std::string json_str(GetTestJson({json})); |
- |
- LoadFromJSONString(service.get(), json_str); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, RemoveExpiredDismissedContent) { |
- auto service = MakeSnippetsService(); |
- |
- std::string json_str1(GetTestJson({GetExpiredSnippet()})); |
- // Load it. |
- LoadFromJSONString(service.get(), json_str1); |
- // Load the image to store it in the database. |
- // TODO(tschumann): Introduce some abstraction to nicely work with image |
- // fetching expectations. |
- ServeImageCallback cb = |
- base::Bind(&ServeOneByOneImage, &service->GetImageFetcherForTesting()); |
- EXPECT_CALL(*image_fetcher(), StartOrQueueNetworkRequest(_, _, _)) |
- .WillOnce(WithArgs<0, 2>(Invoke(&cb, &ServeImageCallback::Run))); |
- image_decoder()->SetDecodedImage(gfx::test::CreateImage(1, 1)); |
- gfx::Image image = FetchImage(service.get(), MakeArticleID(kSnippetUrl)); |
- EXPECT_FALSE(image.IsEmpty()); |
- EXPECT_EQ(1, image.Width()); |
- |
- // Dismiss the suggestion |
- service->DismissSuggestion( |
- ContentSuggestion::ID(articles_category(), kSnippetUrl)); |
- |
- // Load a different snippet - this will clear the expired dismissed ones. |
- std::string json_str2(GetTestJson({GetSnippetWithUrl(kSnippetUrl2)})); |
- LoadFromJSONString(service.get(), json_str2); |
- |
- EXPECT_THAT(service->GetDismissedSnippetsForTesting(articles_category()), |
- IsEmpty()); |
- |
- // Verify the image got removed, too. |
- EXPECT_TRUE(FetchImage(service.get(), MakeArticleID(kSnippetUrl)).IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ExpiredContentNotRemoved) { |
- auto service = MakeSnippetsService(); |
- |
- std::string json_str(GetTestJson({GetExpiredSnippet()})); |
- |
- LoadFromJSONString(service.get(), json_str); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, TestSingleSource) { |
- auto service = MakeSnippetsService(); |
- |
- std::string json_str(GetTestJson({GetSnippetWithSources( |
- "http://source1.com", "Source 1", "http://source1.amp.com")})); |
- |
- LoadFromJSONString(service.get(), json_str); |
- ASSERT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- const NTPSnippet& snippet = |
- *service->GetSnippetsForTesting(articles_category()).front(); |
- EXPECT_EQ(snippet.id(), kSnippetUrl); |
- EXPECT_EQ(snippet.url(), GURL("http://source1.com")); |
- EXPECT_EQ(snippet.publisher_name(), std::string("Source 1")); |
- EXPECT_EQ(snippet.amp_url(), GURL("http://source1.amp.com")); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, TestSingleSourceWithMalformedUrl) { |
- auto service = MakeSnippetsService(); |
- |
- std::string json_str(GetTestJson({GetSnippetWithSources( |
- "ceci n'est pas un url", "Source 1", "http://source1.amp.com")})); |
- |
- LoadFromJSONString(service.get(), json_str); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, TestSingleSourceWithMissingData) { |
- auto service = MakeSnippetsService(); |
- |
- std::string json_str( |
- GetTestJson({GetSnippetWithSources("http://source1.com", "", "")})); |
- |
- LoadFromJSONString(service.get(), json_str); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, LogNumArticlesHistogram) { |
- auto service = MakeSnippetsService(); |
- |
- base::HistogramTester tester; |
- LoadFromJSONString(service.get(), GetTestJson({GetInvalidSnippet()})); |
- |
- EXPECT_THAT(tester.GetAllSamples("NewTabPage.Snippets.NumArticles"), |
- ElementsAre(base::Bucket(/*min=*/0, /*count=*/1))); |
- |
- // Invalid JSON shouldn't contribute to NumArticlesFetched. |
- EXPECT_THAT(tester.GetAllSamples("NewTabPage.Snippets.NumArticlesFetched"), |
- IsEmpty()); |
- |
- // Valid JSON with empty list. |
- LoadFromJSONString(service.get(), GetTestJson(std::vector<std::string>())); |
- EXPECT_THAT(tester.GetAllSamples("NewTabPage.Snippets.NumArticles"), |
- ElementsAre(base::Bucket(/*min=*/0, /*count=*/2))); |
- EXPECT_THAT(tester.GetAllSamples("NewTabPage.Snippets.NumArticlesFetched"), |
- ElementsAre(base::Bucket(/*min=*/0, /*count=*/1))); |
- |
- // Snippet list should be populated with size 1. |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippet()})); |
- EXPECT_THAT(tester.GetAllSamples("NewTabPage.Snippets.NumArticles"), |
- ElementsAre(base::Bucket(/*min=*/0, /*count=*/2), |
- base::Bucket(/*min=*/1, /*count=*/1))); |
- EXPECT_THAT(tester.GetAllSamples("NewTabPage.Snippets.NumArticlesFetched"), |
- ElementsAre(base::Bucket(/*min=*/0, /*count=*/1), |
- base::Bucket(/*min=*/1, /*count=*/1))); |
- |
- // Duplicate snippet shouldn't increase the list size. |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippet()})); |
- EXPECT_THAT(tester.GetAllSamples("NewTabPage.Snippets.NumArticles"), |
- ElementsAre(base::Bucket(/*min=*/0, /*count=*/2), |
- base::Bucket(/*min=*/1, /*count=*/2))); |
- EXPECT_THAT(tester.GetAllSamples("NewTabPage.Snippets.NumArticlesFetched"), |
- ElementsAre(base::Bucket(/*min=*/0, /*count=*/1), |
- base::Bucket(/*min=*/1, /*count=*/2))); |
- EXPECT_THAT( |
- tester.GetAllSamples("NewTabPage.Snippets.NumArticlesZeroDueToDiscarded"), |
- IsEmpty()); |
- |
- // Dismissing a snippet should decrease the list size. This will only be |
- // logged after the next fetch. |
- service->DismissSuggestion(MakeArticleID(kSnippetUrl)); |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippet()})); |
- EXPECT_THAT(tester.GetAllSamples("NewTabPage.Snippets.NumArticles"), |
- ElementsAre(base::Bucket(/*min=*/0, /*count=*/3), |
- base::Bucket(/*min=*/1, /*count=*/2))); |
- // Dismissed snippets shouldn't influence NumArticlesFetched. |
- EXPECT_THAT(tester.GetAllSamples("NewTabPage.Snippets.NumArticlesFetched"), |
- ElementsAre(base::Bucket(/*min=*/0, /*count=*/1), |
- base::Bucket(/*min=*/1, /*count=*/3))); |
- EXPECT_THAT( |
- tester.GetAllSamples("NewTabPage.Snippets.NumArticlesZeroDueToDiscarded"), |
- ElementsAre(base::Bucket(/*min=*/1, /*count=*/1))); |
- |
- // There is only a single, dismissed snippet in the database, so recreating |
- // the service will require us to re-fetch. |
- tester.ExpectTotalCount("NewTabPage.Snippets.NumArticlesFetched", 4); |
- ResetSnippetsService(&service, /*set_empty_response=*/true); |
- EXPECT_EQ(observer().StatusForCategory(articles_category()), |
- CategoryStatus::AVAILABLE); |
- tester.ExpectTotalCount("NewTabPage.Snippets.NumArticlesFetched", 5); |
- EXPECT_THAT( |
- tester.GetAllSamples("NewTabPage.Snippets.NumArticlesZeroDueToDiscarded"), |
- ElementsAre(base::Bucket(/*min=*/1, /*count=*/2))); |
- |
- // But if there's a non-dismissed snippet in the database, recreating it |
- // shouldn't trigger a fetch. |
- LoadFromJSONString( |
- service.get(), |
- GetTestJson({GetSnippetWithUrl("http://not-dismissed.com")})); |
- tester.ExpectTotalCount("NewTabPage.Snippets.NumArticlesFetched", 6); |
- ResetSnippetsService(&service, /*set_empty_response=*/true); |
- tester.ExpectTotalCount("NewTabPage.Snippets.NumArticlesFetched", 6); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, DismissShouldRespectAllKnownUrls) { |
- auto service = MakeSnippetsService(); |
- |
- const base::Time creation = GetDefaultCreationTime(); |
- const base::Time expiry = GetDefaultExpirationTime(); |
- const std::vector<std::string> source_urls = { |
- "http://mashable.com/2016/05/11/stolen", |
- "http://www.aol.com/article/2016/05/stolen-doggie"}; |
- const std::vector<std::string> publishers = {"Mashable", "AOL"}; |
- const std::vector<std::string> amp_urls = { |
- "http://mashable-amphtml.googleusercontent.com/1", |
- "http://t2.gstatic.com/images?q=tbn:3"}; |
- |
- // Add the snippet from the mashable domain. |
- LoadFromJSONString(service.get(), |
- GetTestJson({GetSnippetWithUrlAndTimesAndSource( |
- source_urls, source_urls[0], creation, expiry, |
- publishers[0], amp_urls[0])})); |
- ASSERT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- // Dismiss the snippet via the mashable source corpus ID. |
- service->DismissSuggestion(MakeArticleID(source_urls[0])); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
- |
- // The same article from the AOL domain should now be detected as dismissed. |
- LoadFromJSONString(service.get(), |
- GetTestJson({GetSnippetWithUrlAndTimesAndSource( |
- source_urls, source_urls[1], creation, expiry, |
- publishers[1], amp_urls[1])})); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, StatusChanges) { |
- auto service = MakeSnippetsService(); |
- |
- // Simulate user signed out |
- SetUpFetchResponse(GetTestJson({GetSnippet()})); |
- service->OnStatusChanged(RemoteSuggestionsStatus::ENABLED_AND_SIGNED_IN, |
- RemoteSuggestionsStatus::SIGNED_OUT_AND_DISABLED); |
- |
- base::RunLoop().RunUntilIdle(); |
- EXPECT_THAT(observer().StatusForCategory(articles_category()), |
- Eq(CategoryStatus::SIGNED_OUT)); |
- EXPECT_THAT(RemoteSuggestionsProvider::State::DISABLED, Eq(service->state_)); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), |
- IsEmpty()); // No fetch should be made. |
- |
- // Simulate user sign in. The service should be ready again and load snippets. |
- SetUpFetchResponse(GetTestJson({GetSnippet()})); |
- service->OnStatusChanged(RemoteSuggestionsStatus::SIGNED_OUT_AND_DISABLED, |
- RemoteSuggestionsStatus::ENABLED_AND_SIGNED_IN); |
- EXPECT_THAT(observer().StatusForCategory(articles_category()), |
- Eq(CategoryStatus::AVAILABLE_LOADING)); |
- |
- base::RunLoop().RunUntilIdle(); |
- EXPECT_THAT(observer().StatusForCategory(articles_category()), |
- Eq(CategoryStatus::AVAILABLE)); |
- EXPECT_THAT(RemoteSuggestionsProvider::State::READY, Eq(service->state_)); |
- EXPECT_FALSE(service->GetSnippetsForTesting(articles_category()).empty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ImageReturnedWithTheSameId) { |
- auto service = MakeSnippetsService(); |
- |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippet()})); |
- |
- gfx::Image image; |
- MockFunction<void(const gfx::Image&)> image_fetched; |
- ServeImageCallback cb = |
- base::Bind(&ServeOneByOneImage, &service->GetImageFetcherForTesting()); |
- { |
- InSequence s; |
- EXPECT_CALL(*image_fetcher(), StartOrQueueNetworkRequest(_, _, _)) |
- .WillOnce(WithArgs<0, 2>(Invoke(&cb, &ServeImageCallback::Run))); |
- EXPECT_CALL(image_fetched, Call(_)).WillOnce(SaveArg<0>(&image)); |
- } |
- |
- service->FetchSuggestionImage( |
- MakeArticleID(kSnippetUrl), |
- base::Bind(&MockFunction<void(const gfx::Image&)>::Call, |
- base::Unretained(&image_fetched))); |
- base::RunLoop().RunUntilIdle(); |
- // Check that the image by ServeOneByOneImage is really served. |
- EXPECT_EQ(1, image.Width()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, EmptyImageReturnedForNonExistentId) { |
- auto service = MakeSnippetsService(); |
- |
- // Create a non-empty image so that we can test the image gets updated. |
- gfx::Image image = gfx::test::CreateImage(1, 1); |
- MockFunction<void(const gfx::Image&)> image_fetched; |
- EXPECT_CALL(image_fetched, Call(_)).WillOnce(SaveArg<0>(&image)); |
- |
- service->FetchSuggestionImage( |
- MakeArticleID(kSnippetUrl2), |
- base::Bind(&MockFunction<void(const gfx::Image&)>::Call, |
- base::Unretained(&image_fetched))); |
- |
- base::RunLoop().RunUntilIdle(); |
- EXPECT_TRUE(image.IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, |
- FetchingUnknownImageIdShouldNotHitDatabase) { |
- // Testing that the provider is not accessing the database is tricky. |
- // Therefore, we simply put in some data making sure that if the provider asks |
- // the database, it will get a wrong answer. |
- auto service = MakeSnippetsService(); |
- |
- ContentSuggestion::ID unknown_id = MakeArticleID(kSnippetUrl2); |
- database()->SaveImage(unknown_id.id_within_category(), "some image blob"); |
- // Set up the image decoder to always return the 1x1 test image. |
- image_decoder()->SetDecodedImage(gfx::test::CreateImage(1, 1)); |
- |
- // Create a non-empty image so that we can test the image gets updated. |
- gfx::Image image = gfx::test::CreateImage(2, 2); |
- MockFunction<void(const gfx::Image&)> image_fetched; |
- EXPECT_CALL(image_fetched, Call(_)).WillOnce(SaveArg<0>(&image)); |
- |
- service->FetchSuggestionImage( |
- MakeArticleID(kSnippetUrl2), |
- base::Bind(&MockFunction<void(const gfx::Image&)>::Call, |
- base::Unretained(&image_fetched))); |
- |
- base::RunLoop().RunUntilIdle(); |
- EXPECT_TRUE(image.IsEmpty()) << "got image with width: " << image.Width(); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ClearHistoryRemovesAllSuggestions) { |
- auto service = MakeSnippetsService(); |
- |
- std::string first_snippet = GetSnippetWithUrl("http://url1.com"); |
- std::string second_snippet = GetSnippetWithUrl("http://url2.com"); |
- std::string json_str = GetTestJson({first_snippet, second_snippet}); |
- LoadFromJSONString(service.get(), json_str); |
- ASSERT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(2)); |
- |
- service->DismissSuggestion(MakeArticleID("http://url1.com")); |
- ASSERT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- ASSERT_THAT(service->GetDismissedSnippetsForTesting(articles_category()), |
- SizeIs(1)); |
- |
- base::Time begin = base::Time::FromTimeT(123), |
- end = base::Time::FromTimeT(456); |
- base::Callback<bool(const GURL& url)> filter; |
- service->ClearHistory(begin, end, filter); |
- |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
- EXPECT_THAT(service->GetDismissedSnippetsForTesting(articles_category()), |
- IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, SuggestionsFetchedOnSignInAndSignOut) { |
- auto service = MakeSnippetsService(); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), IsEmpty()); |
- |
- // |MakeSnippetsService()| creates a service where user is signed in already, |
- // so we start by signing out. |
- SetUpFetchResponse(GetTestJson({GetSnippetN(1)})); |
- service->OnStatusChanged(RemoteSuggestionsStatus::ENABLED_AND_SIGNED_IN, |
- RemoteSuggestionsStatus::ENABLED_AND_SIGNED_OUT); |
- base::RunLoop().RunUntilIdle(); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(1)); |
- |
- // Sign in to check a transition from signed out to signed in. |
- SetUpFetchResponse(GetTestJson({GetSnippetN(1), GetSnippetN(2)})); |
- service->OnStatusChanged(RemoteSuggestionsStatus::ENABLED_AND_SIGNED_OUT, |
- RemoteSuggestionsStatus::ENABLED_AND_SIGNED_IN); |
- base::RunLoop().RunUntilIdle(); |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), SizeIs(2)); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, ShouldClearOrphanedImagesOnRestart) { |
- auto service = MakeSnippetsService(); |
- |
- LoadFromJSONString(service.get(), GetTestJson({GetSnippet()})); |
- ServeImageCallback cb = |
- base::Bind(&ServeOneByOneImage, &service->GetImageFetcherForTesting()); |
- |
- EXPECT_CALL(*image_fetcher(), StartOrQueueNetworkRequest(_, _, _)) |
- .WillOnce(WithArgs<0, 2>(Invoke(&cb, &ServeImageCallback::Run))); |
- image_decoder()->SetDecodedImage(gfx::test::CreateImage(1, 1)); |
- |
- gfx::Image image = FetchImage(service.get(), MakeArticleID(kSnippetUrl)); |
- EXPECT_EQ(1, image.Width()); |
- EXPECT_FALSE(image.IsEmpty()); |
- |
- // Send new suggestion which don't include the snippet referencing the image. |
- LoadFromJSONString(service.get(), |
- GetTestJson({GetSnippetWithUrl( |
- "http://something.com/pletely/unrelated")})); |
- // The image should still be available until a restart happens. |
- EXPECT_FALSE(FetchImage(service.get(), MakeArticleID(kSnippetUrl)).IsEmpty()); |
- ResetSnippetsService(&service, /*set_empty_response=*/true); |
- // After the restart, the image should be garbage collected. |
- EXPECT_TRUE(FetchImage(service.get(), MakeArticleID(kSnippetUrl)).IsEmpty()); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, |
- ShouldHandleMoreThanMaxSnippetsInResponse) { |
- auto service = MakeSnippetsService(); |
- |
- std::vector<std::string> suggestions; |
- for (int i = 0; i < service->GetMaxSnippetCountForTesting() + 1; ++i) { |
- suggestions.push_back(GetSnippetWithUrl( |
- base::StringPrintf("http://localhost/snippet-id-%d", i))); |
- } |
- LoadFromJSONString(service.get(), GetTestJson(suggestions)); |
- // TODO(tschumann): We should probably trim out any additional results and |
- // only serve the MaxSnippetCount items. |
- EXPECT_THAT(service->GetSnippetsForTesting(articles_category()), |
- SizeIs(service->GetMaxSnippetCountForTesting() + 1)); |
-} |
- |
-TEST_F(RemoteSuggestionsProviderTest, StoreLastSuccessfullBackgroundFetchTime) { |
- // On initialization of the RemoteSuggestionsProvider a background fetch is |
- // triggered since the snippets DB is empty. Therefore the service must not be |
- // initialized until the test clock is set. |
- auto service = MakeSnippetsServiceWithoutInitialization(); |
- |
- auto simple_test_clock = base::MakeUnique<base::SimpleTestClock>(); |
- base::SimpleTestClock* simple_test_clock_ptr = simple_test_clock.get(); |
- service->SetClockForTesting(std::move(simple_test_clock)); |
- |
- // Test that the preference is correctly initialized with the default value 0. |
- EXPECT_EQ( |
- 0, pref_service()->GetInt64(prefs::kLastSuccessfulBackgroundFetchTime)); |
- |
- WaitForSnippetsServiceInitialization(service.get(), |
- /*set_empty_response=*/true); |
- EXPECT_EQ( |
- simple_test_clock_ptr->Now().ToInternalValue(), |
- pref_service()->GetInt64(prefs::kLastSuccessfulBackgroundFetchTime)); |
- |
- // Advance the time and check whether the time was updated correctly after the |
- // background fetch. |
- simple_test_clock_ptr->Advance(TimeDelta::FromHours(1)); |
- service->FetchSnippetsInTheBackground(); |
- base::RunLoop().RunUntilIdle(); |
- EXPECT_EQ( |
- simple_test_clock_ptr->Now().ToInternalValue(), |
- pref_service()->GetInt64(prefs::kLastSuccessfulBackgroundFetchTime)); |
- // TODO(markusheintz): Add a test that simulates a browser restart once the |
- // scheduler refactoring is done (crbug.com/672434). |
-} |
- |
-} // namespace ntp_snippets |