Home | History | Annotate | Download | only in token_cache 
      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:
     19   TokenCacheTest() : cache_(&profile_) {}
     20   virtual ~TokenCacheTest() { cache_.Shutdown(); }
     21 
     22   size_t CacheSize() {
     23     return cache_.token_cache_.size();
     24   }
     25 
     26   bool HasMatch(const std::string& token_name) {
     27     return cache_.RetrieveToken(token_name) != std::string();
     28   }
     29 
     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 
     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 
     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 
     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 
     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 
     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