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