[Doodle] Allow overriding the API URL via a fieldtrial param

components/doodle/doodle_fetcher_impl_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/doodle/doodle_fetcher_impl.h"
 
 #include <memory>
 #include <string>
 #include <utility>
 
@@ skipping 52 matching lines @@
     success.Run(std::move(value));
   } else {
     error.Run(json_reader.GetErrorMessage());
   }
 }
 
 }  // namespace
 
 class DoodleFetcherImplTestBase : public testing::Test {
  public:
-  DoodleFetcherImplTestBase(bool gray_background)
+  DoodleFetcherImplTestBase(bool gray_background,
+                            const base::Optional<std::string>& override_url)
       : google_url_tracker_(base::MakeUnique<GoogleURLTrackerClientStub>(),
                             GoogleURLTracker::UNIT_TEST_MODE),
         doodle_fetcher_(
             new net::TestURLRequestContextGetter(message_loop_.task_runner()),
             &google_url_tracker_,
             base::Bind(ParseJson),
-            gray_background) {}
+            gray_background,
+            override_url) {}
 
   void RespondWithData(const std::string& data) {
     net::TestURLFetcher* url_fetcher = GetRunningFetcher();
     url_fetcher->set_status(net::URLRequestStatus());
     url_fetcher->set_response_code(net::HTTP_OK);
     url_fetcher->SetResponseString(data);
     // Call the URLFetcher delegate to continue the test.
     url_fetcher->delegate()->OnURLFetchComplete(url_fetcher);
   }
 
@@ skipping 23 matching lines @@
  private:
   base::MessageLoop message_loop_;
   net::TestURLFetcherFactory url_fetcher_factory_;
   GoogleURLTracker google_url_tracker_;
   DoodleFetcherImpl doodle_fetcher_;
 };
 
 class DoodleFetcherImplTest : public DoodleFetcherImplTestBase {
  public:
   DoodleFetcherImplTest()
-      : DoodleFetcherImplTestBase(/*gray_background=*/true) {}
+      : DoodleFetcherImplTestBase(/*gray_background=*/true,
+                                  /*override_url=*/base::nullopt) {}
 };
 
 TEST_F(DoodleFetcherImplTest, ReturnsFromFetchWithoutError) {
   base::MockCallback<DoodleFetcherImpl::FinishedCallback> callback;
   doodle_fetcher()->FetchDoodle(callback.Get());
 
   DoodleState state = DoodleState::NO_DOODLE;
   base::Optional<DoodleConfig> response;
   EXPECT_CALL(callback, Run(_, _, _))
       .WillOnce(DoAll(SaveArg<0>(&state), SaveArg<2>(&response)));
@@ skipping 280 matching lines @@
   base::MockCallback<DoodleFetcherImpl::FinishedCallback> callback;
   doodle_fetcher()->FetchDoodle(callback.Get());
 
   EXPECT_THAT(GetRunningFetcher()->GetOriginalURL(),
               Eq(GURL("http://www.google.kz").Resolve(kDoodleConfigPath)));
 }
 
 class DoodleFetcherImplNoGrayBgTest : public DoodleFetcherImplTestBase {
  public:
   DoodleFetcherImplNoGrayBgTest()
-      : DoodleFetcherImplTestBase(/*gray_background=*/false) {}
+      : DoodleFetcherImplTestBase(/*gray_background=*/false,
+                                  /*override_url=*/base::nullopt) {}
 };
 
 TEST_F(DoodleFetcherImplNoGrayBgTest, PassesNoGrayBgParam) {
   base::MockCallback<DoodleFetcherImpl::FinishedCallback> callback;
   doodle_fetcher()->FetchDoodle(callback.Get());
 
   EXPECT_THAT(GetRunningFetcher()->GetOriginalURL(),
               Eq(Resolve(kDoodleConfigPathNoGrayBg)));
 }
 
+class DoodleFetcherImplRelativeOverrideUrlTest
+    : public DoodleFetcherImplTestBase {
+ public:
+  DoodleFetcherImplRelativeOverrideUrlTest()
+      : DoodleFetcherImplTestBase(/*gray_background=*/false,
+                                  /*override_url=*/std::string("/different")) {}
+};
+
+TEST_F(DoodleFetcherImplRelativeOverrideUrlTest, OverridesWithRelativeUrl) {
+  base::MockCallback<DoodleFetcherImpl::FinishedCallback> callback;
+  doodle_fetcher()->FetchDoodle(callback.Get());
+
+  EXPECT_THAT(GetRunningFetcher()->GetOriginalURL(), Eq(Resolve("/different")));
+}
+
+class DoodleFetcherImplAbsoluteOverrideUrlTest
+    : public DoodleFetcherImplTestBase {
+ public:
+  DoodleFetcherImplAbsoluteOverrideUrlTest()
+      : DoodleFetcherImplTestBase(
+            /*gray_background=*/false,
+            /*override_url=*/std::string("http://host.com/ddl")) {}
+};
+
+TEST_F(DoodleFetcherImplAbsoluteOverrideUrlTest, OverridesWithAbsoluteUrl) {
+  base::MockCallback<DoodleFetcherImpl::FinishedCallback> callback;
+  doodle_fetcher()->FetchDoodle(callback.Get());
+
+  EXPECT_THAT(GetRunningFetcher()->GetOriginalURL(),
+              Eq(GURL("http://host.com/ddl")));
+}
+
 }  // namespace doodle