OLD | NEW |
| (Empty) |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/common/origin_trials/origin_trial_key_manager.h" | |
6 | |
7 #include <memory> | |
8 | |
9 #include "base/macros.h" | |
10 #include "base/memory/ptr_util.h" | |
11 #include "base/strings/string_piece.h" | |
12 #include "testing/gtest/include/gtest/gtest.h" | |
13 | |
14 const uint8_t kTestPublicKey[] = { | |
15 0x75, 0x10, 0xac, 0xf9, 0x3a, 0x1c, 0xb8, 0xa9, 0x28, 0x70, 0xd2, | |
16 0x9a, 0xd0, 0x0b, 0x59, 0xe1, 0xac, 0x2b, 0xb7, 0xd5, 0xca, 0x1f, | |
17 0x64, 0x90, 0x08, 0x8e, 0xa8, 0xe0, 0x56, 0x3a, 0x04, 0xd0, | |
18 }; | |
19 | |
20 // Base64 encoding of the above sample public key | |
21 const char kTestPublicKeyString[] = | |
22 "dRCs+TocuKkocNKa0AtZ4awrt9XKH2SQCI6o4FY6BNA="; | |
23 const char kBadEncodingPublicKeyString[] = "Not even base64!"; | |
24 // Base64-encoded, 31 bytes long | |
25 const char kTooShortPublicKeyString[] = | |
26 "dRCs+TocuKkocNKa0AtZ4awrt9XKH2SQCI6o4FY6BN=="; | |
27 // Base64-encoded, 33 bytes long | |
28 const char kTooLongPublicKeyString[] = | |
29 "dRCs+TocuKkocNKa0AtZ4awrt9XKH2SQCI6o4FY6BNAA"; | |
30 | |
31 class OriginTrialKeyManagerTest : public testing::Test { | |
32 protected: | |
33 OriginTrialKeyManagerTest() | |
34 : manager_(base::WrapUnique(new OriginTrialKeyManager())), | |
35 default_key_(manager_->GetPublicKey().as_string()), | |
36 test_key_(std::string(reinterpret_cast<const char*>(kTestPublicKey), | |
37 arraysize(kTestPublicKey))) {} | |
38 OriginTrialKeyManager* manager() { return manager_.get(); } | |
39 base::StringPiece default_key() { return default_key_; } | |
40 base::StringPiece test_key() { return test_key_; } | |
41 | |
42 private: | |
43 std::unique_ptr<OriginTrialKeyManager> manager_; | |
44 std::string default_key_; | |
45 std::string test_key_; | |
46 }; | |
47 | |
48 TEST_F(OriginTrialKeyManagerTest, DefaultConstructor) { | |
49 // We don't specify here what the key should be, but make sure that it is | |
50 // returned, is valid, and is consistent. | |
51 base::StringPiece key = manager()->GetPublicKey(); | |
52 EXPECT_EQ(32UL, key.size()); | |
53 EXPECT_EQ(default_key(), key); | |
54 } | |
55 | |
56 TEST_F(OriginTrialKeyManagerTest, DefaultKeyIsConsistent) { | |
57 OriginTrialKeyManager manager2; | |
58 EXPECT_EQ(manager()->GetPublicKey(), manager2.GetPublicKey()); | |
59 } | |
60 | |
61 TEST_F(OriginTrialKeyManagerTest, OverridePublicKey) { | |
62 EXPECT_TRUE(manager()->SetPublicKeyFromASCIIString(kTestPublicKeyString)); | |
63 EXPECT_NE(default_key(), manager()->GetPublicKey()); | |
64 EXPECT_EQ(test_key(), manager()->GetPublicKey()); | |
65 } | |
66 | |
67 TEST_F(OriginTrialKeyManagerTest, OverrideKeyNotBase64) { | |
68 EXPECT_FALSE( | |
69 manager()->SetPublicKeyFromASCIIString(kBadEncodingPublicKeyString)); | |
70 EXPECT_EQ(default_key(), manager()->GetPublicKey()); | |
71 } | |
72 | |
73 TEST_F(OriginTrialKeyManagerTest, OverrideKeyTooShort) { | |
74 EXPECT_FALSE( | |
75 manager()->SetPublicKeyFromASCIIString(kTooShortPublicKeyString)); | |
76 EXPECT_EQ(default_key(), manager()->GetPublicKey()); | |
77 } | |
78 | |
79 TEST_F(OriginTrialKeyManagerTest, OverrideKeyTooLong) { | |
80 EXPECT_FALSE(manager()->SetPublicKeyFromASCIIString(kTooLongPublicKeyString)); | |
81 EXPECT_EQ(default_key(), manager()->GetPublicKey()); | |
82 } | |
OLD | NEW |