Add disabled origin trial feature list

chrome/renderer/origin_trials/origin_trial_key_manager_unittest.cc

 // Copyright (c) 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/renderer/origin_trials/origin_trial_key_manager.h"
 
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/string_piece.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 const uint8_t kTestPublicKey[] = {
     0x75, 0x10, 0xac, 0xf9, 0x3a, 0x1c, 0xb8, 0xa9, 0x28, 0x70, 0xd2,
     0x9a, 0xd0, 0x0b, 0x59, 0xe1, 0xac, 0x2b, 0xb7, 0xd5, 0xca, 0x1f,
     0x64, 0x90, 0x08, 0x8e, 0xa8, 0xe0, 0x56, 0x3a, 0x04, 0xd0,
 };
 
 // Base64 encoding of the above sample public key
 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 OriginTrialKeyManagerTest : public testing::Test {
  protected:
   OriginTrialKeyManagerTest()
       : manager_(make_scoped_ptr(new OriginTrialKeyManager())),
         default_key_(manager_->GetPublicKey().as_string()),
         test_key_(std::string(reinterpret_cast<const char*>(kTestPublicKey),
                               arraysize(kTestPublicKey))) {}
   OriginTrialKeyManager* manager() { return manager_.get(); }
   base::StringPiece default_key() { return default_key_; }
   base::StringPiece test_key() { return test_key_; }
@@ skipping 32 matching lines @@
 TEST_F(OriginTrialKeyManagerTest, OverrideKeyTooShort) {
   EXPECT_FALSE(
       manager()->SetPublicKeyFromASCIIString(kTooShortPublicKeyString));
   EXPECT_EQ(default_key(), manager()->GetPublicKey());
 }
 
 TEST_F(OriginTrialKeyManagerTest, OverrideKeyTooLong) {
   EXPECT_FALSE(manager()->SetPublicKeyFromASCIIString(kTooLongPublicKeyString));
   EXPECT_EQ(default_key(), manager()->GetPublicKey());
 }
+
+TEST_F(OriginTrialKeyManagerTest, NoDisabledFeatures) {
+  EXPECT_FALSE(manager()->IsFeatureDisabled("A"));
+  EXPECT_FALSE(manager()->IsFeatureDisabled("B"));
+  EXPECT_FALSE(manager()->IsFeatureDisabled("C"));
+}
+
+TEST_F(OriginTrialKeyManagerTest, DisableOneFeature) {
+  EXPECT_TRUE(manager()->SetDisabledFeatures(kOneDisabledFeature));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("A"));
+  EXPECT_FALSE(manager()->IsFeatureDisabled("B"));
+}
+
+TEST_F(OriginTrialKeyManagerTest, DisableTwoFeatures) {
+  EXPECT_TRUE(manager()->SetDisabledFeatures(kTwoDisabledFeatures));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("A"));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("B"));
+  EXPECT_FALSE(manager()->IsFeatureDisabled("C"));
+}
+
+TEST_F(OriginTrialKeyManagerTest, DisableThreeFeatures) {
+  EXPECT_TRUE(manager()->SetDisabledFeatures(kThreeDisabledFeatures));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("A"));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("B"));
+  EXPECT_TRUE(manager()->IsFeatureDisabled("C"));
+}
+
+TEST_F(OriginTrialKeyManagerTest, 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"));
+}