Add finch gate for captive portal interstitial.

chrome/browser/ssl/ssl_error_handler_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 "chrome/browser/ssl/ssl_error_handler.h"
 
 #include "base/callback.h"
 #include "base/message_loop/message_loop.h"
+#include "base/metrics/field_trial.h"
 #include "base/run_loop.h"
 #include "base/time/time.h"
 #include "chrome/browser/captive_portal/captive_portal_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/captive_portal/captive_portal_testing_utils.h"
 #include "content/public/browser/notification_service.h"
 #include "net/base/net_errors.h"
 #include "net/ssl/ssl_info.h"
@@ skipping 70 matching lines @@
   Profile* profile_;
   bool captive_portal_checked_;
   bool ssl_interstitial_shown_;
   bool captive_portal_interstitial_shown_;
 
   DISALLOW_COPY_AND_ASSIGN(TestSSLErrorHandler);
 };
 
 class SSLErrorHandlerTest : public ChromeRenderViewHostTestHarness {
  public:
+  SSLErrorHandlerTest()
+      : field_trial_list_(NULL) {
+  }
+
   void SetUp() override {
     ChromeRenderViewHostTestHarness::SetUp();
     SSLErrorHandler::SetInterstitialDelayTypeForTest(SSLErrorHandler::NONE);
     error_handler_.reset(new TestSSLErrorHandler(profile(),
                                                  web_contents(),
                                                  ssl_info_));
-  }
+    // Enable finch experiment for captive portal interstitials.
+    ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial(
+                    "CaptivePortalInterstitial", "Enabled"));
+}
 
   void TearDown() override {
     EXPECT_FALSE(error_handler()->IsTimerRunning());
     error_handler_.reset(NULL);
     ChromeRenderViewHostTestHarness::TearDown();
   }
 
   TestSSLErrorHandler* error_handler() { return error_handler_.get(); }
 
  private:
   net::SSLInfo ssl_info_;
   scoped_ptr<TestSSLErrorHandler> error_handler_;
+  base::FieldTrialList field_trial_list_;
 };
 
 #if defined(ENABLE_CAPTIVE_PORTAL_DETECTION)
 
 TEST_F(SSLErrorHandlerTest,
        ShouldShowSSLInterstitialOnTimerExpired) {
   EXPECT_FALSE(error_handler()->IsTimerRunning());
   error_handler()->StartHandlingError();
 
   EXPECT_TRUE(error_handler()->IsTimerRunning());
@@ skipping 60 matching lines @@
        ShouldShowSSLInterstitialOnCaptivePortalDetectionDisabled) {
   EXPECT_FALSE(error_handler()->IsTimerRunning());
   error_handler()->StartHandlingError();
   EXPECT_FALSE(error_handler()->IsTimerRunning());
   EXPECT_FALSE(error_handler()->captive_portal_checked());
   EXPECT_TRUE(error_handler()->ssl_interstitial_shown());
   EXPECT_FALSE(error_handler()->captive_portal_interstitial_shown());
 }
 
 #endif  // defined(ENABLE_CAPTIVE_PORTAL_DETECTION)