1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
// Copyright (c) 2013 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 <stddef.h>
#include "base/compiler_specific.h"
#include "base/time/time.h"
#include "chrome/browser/extensions/token_cache/token_cache_service.h"
#include "chrome/test/base/testing_profile.h"
#include "content/public/test/test_browser_thread_bundle.h"
#include "testing/gtest/include/gtest/gtest.h"
using base::Time;
using base::TimeDelta;
namespace extensions {
class TokenCacheTest : public testing::Test {
public:
TokenCacheTest() : cache_(&profile_) {}
~TokenCacheTest() override { cache_.Shutdown(); }
size_t CacheSize() {
return cache_.token_cache_.size();
}
bool HasMatch(const std::string& token_name) {
return cache_.RetrieveToken(token_name) != std::string();
}
void InsertExpiredToken(const std::string& token_name,
const std::string& token_value) {
EXPECT_TRUE(!HasMatch(token_name));
// Compute a time value for yesterday
Time now = Time::Now();
TimeDelta one_day = one_day.FromDays(1);
Time yesterday = now - one_day;
TokenCacheService::TokenCacheData token_data;
token_data.token = token_value;
token_data.expiration_time = yesterday;
cache_.token_cache_[token_name] = token_data;
}
protected:
content::TestBrowserThreadBundle thread_bundle_;
TestingProfile profile_;
TokenCacheService cache_;
};
TEST_F(TokenCacheTest, SaveTokenTest) {
TimeDelta zero;
cache_.StoreToken("foo", "bar", zero);
EXPECT_EQ(1U, CacheSize());
EXPECT_TRUE(HasMatch("foo"));
}
TEST_F(TokenCacheTest, RetrieveTokenTest) {
TimeDelta zero;
cache_.StoreToken("Mozart", "Eine Kleine Nacht Musik", zero);
cache_.StoreToken("Bach", "Brandenburg Concerto #3", zero);
cache_.StoreToken("Beethoven", "Emperor Piano Concerto #5", zero);
cache_.StoreToken("Handel", "Water Music", zero);
cache_.StoreToken("Chopin", "Heroic", zero);
std::string found_token = cache_.RetrieveToken("Chopin");
EXPECT_EQ("Heroic", found_token);
}
TEST_F(TokenCacheTest, ReplaceTokenTest) {
TimeDelta zero;
cache_.StoreToken("Chopin", "Heroic", zero);
std::string found_token = cache_.RetrieveToken("Chopin");
EXPECT_EQ("Heroic", found_token);
cache_.StoreToken("Chopin", "Military", zero);
found_token = cache_.RetrieveToken("Chopin");
EXPECT_EQ("Military", found_token);
EXPECT_EQ(1U, CacheSize());
}
TEST_F(TokenCacheTest, SignoutTest) {
TimeDelta zero;
cache_.StoreToken("foo", "bar", zero);
EXPECT_EQ(1U, CacheSize());
EXPECT_TRUE(HasMatch("foo"));
cache_.GoogleSignedOut("foo", "foo");
EXPECT_EQ(0U, CacheSize());
EXPECT_FALSE(HasMatch("foo"));
}
TEST_F(TokenCacheTest, TokenExpireTest) {
// Use the fact that we are friends to insert an expired token.
InsertExpiredToken("foo", "bar");
EXPECT_EQ(1U, CacheSize());
// If we attempt to find the token, the attempt should fail.
EXPECT_FALSE(HasMatch("foo"));
EXPECT_EQ(0U, CacheSize());
}
} // namespace extensions
|
