Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(35)

Side by Side Diff: chrome/common/origin_trials/origin_trial_key_manager_unittest.cc

Issue 1737693002: Allow command-line arguments to override EF public key (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Rebase Created 4 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(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 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698