chrome/browser/extensions/token_cache/token_cache_service_unittest.cc

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