| Index: chrome/browser/token_cache/token_cache_unittest.cc
|
| diff --git a/chrome/browser/token_cache/token_cache_unittest.cc b/chrome/browser/token_cache/token_cache_unittest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..345cc3c3b99b0716e83a4e0b4787d61990b30341
|
| --- /dev/null
|
| +++ b/chrome/browser/token_cache/token_cache_unittest.cc
|
| @@ -0,0 +1,136 @@
|
| +// 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 "base/basictypes.h"
|
| +#include "base/compiler_specific.h"
|
| +#include "base/time.h"
|
| +#include "chrome/browser/token_cache/token_cache.h"
|
| +#include "chrome/common/chrome_notification_types.h"
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
| +
|
| +using base::Time;
|
| +using base::TimeDelta;
|
| +
|
| +class TokenCacheTest : public testing::Test {
|
| + public:
|
| + TokenCacheTest() {}
|
| + virtual ~TokenCacheTest() {}
|
| +
|
| + // Methods from testing::Test.
|
| + virtual void SetUp() OVERRIDE {}
|
| + virtual void TearDown() OVERRIDE {
|
| + cache_.token_cache_.clear();
|
| + }
|
| +
|
| + size_t CacheSize() {
|
| + return cache_.token_cache_.size();
|
| + }
|
| +
|
| + bool HasMatch(std::string token_name) {
|
| + return cache_.FindMatch(token_name) != cache_.token_cache_.end();
|
| + }
|
| +
|
| + void InsertExpiredToken(std::string token_name, 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;
|
| +
|
| + TokenCacheData token_data;
|
| + token_data.token_name = token_name;
|
| + token_data.token = token_value;
|
| + token_data.expiration_time = yesterday;
|
| +
|
| + cache_.token_cache_.push_back(token_data);
|
| + }
|
| +
|
| + protected:
|
| + TokenCacheService cache_;
|
| +
|
| +};
|
| +
|
| +TEST_F(TokenCacheTest, SaveTokenTest) {
|
| +
|
| + cache_.StoreToken("foo", "bar", 0);
|
| +
|
| + EXPECT_EQ(CacheSize(), (size_t) 1);
|
| + EXPECT_TRUE(HasMatch("foo"));
|
| +}
|
| +
|
| +TEST_F(TokenCacheTest, RetrieveTokenTest) {
|
| + // TODO: does it make the test nicer to make these std::string and not rely on
|
| + // autoconverting?
|
| + cache_.StoreToken("Mozart", "Eine Kleine Nacht Musik", 0);
|
| + cache_.StoreToken("Bach", "Brandenburg Concerto #3", 0);
|
| + cache_.StoreToken("Beethoven", "Emperor Piano Concerto #5", 0);
|
| + cache_.StoreToken("Handel", "Water Music", 0);
|
| + cache_.StoreToken("Chopin", "Heroic", 0);
|
| +
|
| + std::string found_token = cache_.RetrieveToken("Chopin");
|
| + EXPECT_EQ("Heroic", found_token);
|
| +}
|
| +
|
| +TEST_F(TokenCacheTest, ReplaceTokenTest) {
|
| + cache_.StoreToken("Chopin", "Heroic", 0);
|
| +
|
| + std::string found_token = cache_.RetrieveToken("Chopin");
|
| + EXPECT_EQ("Heroic", found_token);
|
| +
|
| + cache_.StoreToken("Chopin", "Military", 0);
|
| +
|
| + found_token = cache_.RetrieveToken("Chopin");
|
| + EXPECT_EQ("Military", found_token);
|
| + EXPECT_EQ(CacheSize(), (size_t) 1);
|
| +}
|
| +
|
| +TEST_F(TokenCacheTest, SignoutTest) {
|
| + cache_.StoreToken("foo", "bar", 0);
|
| + content::Source<TokenCacheService> stub_source(&cache_);
|
| + content::NotificationDetails stub_details;
|
| +
|
| + EXPECT_EQ(CacheSize(), (size_t) 1);
|
| + EXPECT_TRUE(HasMatch("foo"));
|
| +
|
| + cache_.Observe(chrome::NOTIFICATION_GOOGLE_SIGNED_OUT,
|
| + stub_source, stub_details);
|
| +
|
| + EXPECT_EQ(CacheSize(), (size_t) 0);
|
| + EXPECT_FALSE(HasMatch("foo"));
|
| +}
|
| +
|
| +TEST_F(TokenCacheTest, ShutdownTest) {
|
| + cache_.StoreToken("foo", "bar", 0);
|
| +
|
| + EXPECT_EQ(CacheSize(), (size_t) 1);
|
| + EXPECT_TRUE(HasMatch("foo"));
|
| +
|
| + cache_.Shutdown();
|
| +
|
| + EXPECT_EQ(CacheSize(), (size_t) 0);
|
| + EXPECT_FALSE(HasMatch("foo"));
|
| +}
|
| +
|
| +TEST_F(TokenCacheTest, ExpiredToken) {
|
| +
|
| + cache_.StoreToken("foo", "bar", -500);
|
| +
|
| + // Since the token is expired, we expect the insert to fail.
|
| + EXPECT_EQ(CacheSize(), (size_t) 0);
|
| + 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(CacheSize(), (size_t) 1);
|
| +
|
| + // If we attempt to find the token, the attempt should fail.
|
| + EXPECT_FALSE(HasMatch("foo"));
|
| + EXPECT_EQ(CacheSize(), (size_t) 0);
|
| +
|
| +}
|
|
|