Keep WKWebView control group users in the control group

ios/chrome/browser/experimental_flags.mm

 // 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.
 
 // This file can be empty. Its purpose is to contain the relatively short lived
 // definitions required for experimental flags.
 
 #include "ios/chrome/browser/experimental_flags.h"
 
 #include <dispatch/dispatch.h>
 #import <Foundation/Foundation.h>
 
 #include <string>
 
 #include "base/command_line.h"
 #include "base/metrics/field_trial.h"
 #include "base/strings/string_util.h"
 #include "components/autofill/core/common/autofill_switches.h"
 #include "components/enhanced_bookmarks/enhanced_bookmark_features.h"
 #include "components/variations/variations_associated_data.h"
 #include "ios/chrome/browser/chrome_switches.h"
 #include "ios/web/public/web_view_creation_util.h"
 
 namespace {
 NSString* const kEnableAlertOnBackgroundUpload =
     @"EnableAlertsOnBackgroundUpload";
 NSString* const kEnableViewCopyPasswords = @"EnableViewCopyPasswords";
 NSString* const kHeuristicsForPasswordGeneration =
     @"HeuristicsForPasswordGeneration";
+const char* const kWKWebViewTrialName = "IOSUseWKWebView";
 
 enum class WKWebViewEligibility {
   // UNSET indicates that no explicit call to set eligibility has been made,
   // nor has a default value been assumed due to checking eligibility.
   UNSET,
   ELIGIBLE,
   INELIGIBLE
 };
 WKWebViewEligibility g_wkwebview_trial_eligibility =
     WKWebViewEligibility::UNSET;
@@ skipping 81 matching lines @@
   });
 
   return ok_to_check_finch;
 }
 
 bool IsWKWebViewEnabled() {
   if (!CanCheckWKWebViewExperiment()) {
     return false;
   }
 
+  // Now that it's been established that user is a candidate, set up the trial
+  // by checking the group.
+  std::string group_name =
+      base::FieldTrialList::FindFullName(kWKWebViewTrialName);
+
   // Check if the experimental flag is turned on.
   base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
   if (command_line->HasSwitch(switches::kEnableIOSWKWebView))
     return true;
   else if (command_line->HasSwitch(switches::kDisableIOSWKWebView))
     return false;
 
-  // Now that it's been established that user is a candidate, set up the trial
-  // by checking the group.
-  std::string group_name =
-      base::FieldTrialList::FindFullName("IOSUseWKWebView");
-
   // Check if the finch experiment is turned on.
   return base::StartsWith(group_name, "Enabled",
                           base::CompareCase::INSENSITIVE_ASCII);
 }
 
+bool IsTargetedToWKWebViewExperimentControlGroup() {
+  base::FieldTrial* trial = base::FieldTrialList::Find(kWKWebViewTrialName);
+  if (!trial)
+    return false;
+  std::string group_name = trial->GetGroupNameWithoutActivation();
+  return base::StartsWith(group_name, "Control",
+                          base::CompareCase::INSENSITIVE_ASCII);
+}
+
+bool IsInWKWebViewExperimentControlGroup() {
+  if (!CanCheckWKWebViewExperiment()) {
+    return false;
+  }
+  std::string group_name =
+      base::FieldTrialList::FindFullName(kWKWebViewTrialName);
+  return base::StartsWith(group_name, "Control",
+                          base::CompareCase::INSENSITIVE_ASCII);
+}
+
 std::string GetWKWebViewSearchParams() {
   if (!CanCheckWKWebViewExperiment()) {
     return std::string();
   }
 
-  return variations::GetVariationParamValue("IOSUseWKWebView", "esrch");
+  return variations::GetVariationParamValue(kWKWebViewTrialName, "esrch");
 }
 
 bool AreKeyboardCommandsEnabled() {
   return !base::CommandLine::ForCurrentProcess()->HasSwitch(
       switches::kDisableKeyboardCommands);
 }
 
 bool IsViewCopyPasswordsEnabled() {
   NSString* viewCopyPasswordFlag = [[NSUserDefaults standardUserDefaults]
       objectForKey:kEnableViewCopyPasswords];
@@ skipping 19 matching lines @@
   if ([[NSUserDefaults standardUserDefaults]
           boolForKey:kHeuristicsForPasswordGeneration]) {
     return true;
   }
   base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
   return command_line->HasSwitch(
       autofill::switches::kLocalHeuristicsOnlyForPasswordGeneration);
 }
 
 }  // namespace experimental_flags