Add disabled origin trial feature list

chrome/common/origin_trials/chrome_origin_trial_policy_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 "chrome/common/origin_trials/chrome_origin_trial_policy.h"
 
 #include <memory>
 
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
@@ skipping 10 matching lines @@
 const char kTestPublicKeyString[] =
     "dRCs+TocuKkocNKa0AtZ4awrt9XKH2SQCI6o4FY6BNA=";
 const char kBadEncodingPublicKeyString[] = "Not even base64!";
 // Base64-encoded, 31 bytes long
 const char kTooShortPublicKeyString[] =
     "dRCs+TocuKkocNKa0AtZ4awrt9XKH2SQCI6o4FY6BN==";
 // Base64-encoded, 33 bytes long
 const char kTooLongPublicKeyString[] =
     "dRCs+TocuKkocNKa0AtZ4awrt9XKH2SQCI6o4FY6BNAA";
 
+const char kOneDisabledFeature[] = "A";
+const char kTwoDisabledFeatures[] = "A|B";
+const char kThreeDisabledFeatures[] = "A|B|C";
+const char kSpacesInDisabledFeatures[] = "A|B C";
+
 class ChromeOriginTrialPolicyTest : public testing::Test {
  protected:
   ChromeOriginTrialPolicyTest()
       : manager_(base::WrapUnique(new ChromeOriginTrialPolicy())),
         default_key_(manager_->GetPublicKey().as_string()),
         test_key_(std::string(reinterpret_cast<const char*>(kTestPublicKey),
                               arraysize(kTestPublicKey))) {}
   ChromeOriginTrialPolicy* manager() { return manager_.get(); }
   base::StringPiece default_key() { return default_key_; }
   base::StringPiece test_key() { return test_key_; }
@@ skipping 32 matching lines @@
 TEST_F(ChromeOriginTrialPolicyTest, OverrideKeyTooShort) {
   EXPECT_FALSE(
       manager()->SetPublicKeyFromASCIIString(kTooShortPublicKeyString));
   EXPECT_EQ(default_key(), manager()->GetPublicKey());
 }
 
 TEST_F(ChromeOriginTrialPolicyTest, OverrideKeyTooLong) {
   EXPECT_FALSE(manager()->SetPublicKeyFromASCIIString(kTooLongPublicKeyString));
   EXPECT_EQ(default_key(), manager()->GetPublicKey());
 }
+
+TEST_F(ChromeOriginTrialPolicyTest, NoDisabledFeatures) {
+  EXPECT_FALSE(manager()->IsFeatureDisabled("A"));
+  EXPECT_FALSE(manager()->IsFeatureDisabled("B"));
+  EXPECT_FALSE(manager()->IsFeatureDisabled("C"));
+}
+
+TEST_F(ChromeOriginTrialPolicyTest, DisableOneFeature) {
+  EXPECT_TRUE(manager()->SetDisabledFeatures(kOneDisabledFeature));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("A"));
+  EXPECT_FALSE(manager()->IsFeatureDisabled("B"));
+}
+
+TEST_F(ChromeOriginTrialPolicyTest, DisableTwoFeatures) {
+  EXPECT_TRUE(manager()->SetDisabledFeatures(kTwoDisabledFeatures));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("A"));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("B"));
+  EXPECT_FALSE(manager()->IsFeatureDisabled("C"));
+}
+
+TEST_F(ChromeOriginTrialPolicyTest, DisableThreeFeatures) {
+  EXPECT_TRUE(manager()->SetDisabledFeatures(kThreeDisabledFeatures));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("A"));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("B"));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("C"));
+}
+
+TEST_F(ChromeOriginTrialPolicyTest, DisableFeatureWithSpace) {
+  EXPECT_TRUE(manager()->SetDisabledFeatures(kSpacesInDisabledFeatures));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("A"));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("B C"));
+  EXPECT_FALSE(manager()->IsFeatureDisabled("B"));
+  EXPECT_FALSE(manager()->IsFeatureDisabled("C"));
+}