Index: chrome/browser/webdata/token_service_table_unittest.cc |
diff --git a/chrome/browser/webdata/token_service_table_unittest.cc b/chrome/browser/webdata/token_service_table_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..0f1398d295a20d6eb9d9723e718ce8c278138a87 |
--- /dev/null |
+++ b/chrome/browser/webdata/token_service_table_unittest.cc |
@@ -0,0 +1,117 @@ |
+// Copyright (c) 2012 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/files/scoped_temp_dir.h" |
+#include "base/path_service.h" |
+#include "base/strings/string_number_conversions.h" |
+#include "base/time/time.h" |
+#include "chrome/browser/webdata/token_service_table.h" |
+#include "components/webdata/common/web_database.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+using base::Time; |
+ |
+class TokenServiceTableTest : public testing::Test { |
+ public: |
+ TokenServiceTableTest() {} |
+ virtual ~TokenServiceTableTest() {} |
+ |
+ protected: |
+ virtual void SetUp() { |
+ ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
+ file_ = temp_dir_.path().AppendASCII("TestWebDatabase"); |
+ |
+ table_.reset(new TokenServiceTable); |
+ db_.reset(new WebDatabase); |
+ db_->AddTable(table_.get()); |
+ ASSERT_EQ(sql::INIT_OK, db_->Init(file_)); |
+ } |
+ |
+ base::FilePath file_; |
+ base::ScopedTempDir temp_dir_; |
+ scoped_ptr<TokenServiceTable> table_; |
+ scoped_ptr<WebDatabase> db_; |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(TokenServiceTableTest); |
+}; |
+ |
+// Flaky on mac_rel. See http://crbug.com/228943 |
+#if defined(OS_MACOSX) |
+#define MAYBE_TokenServiceGetAllRemoveAll DISABLED_TokenServiceGetAllRemoveAll |
+#define MAYBE_TokenServiceGetSet DISABLED_TokenServiceGetSet |
+#define MAYBE_TokenServiceRemove DISABLED_TokenServiceRemove |
+#else |
+#define MAYBE_TokenServiceGetAllRemoveAll TokenServiceGetAllRemoveAll |
+#define MAYBE_TokenServiceGetSet TokenServiceGetSet |
+#define MAYBE_TokenServiceRemove TokenServiceRemove |
+#endif |
+ |
+TEST_F(TokenServiceTableTest, MAYBE_TokenServiceGetAllRemoveAll) { |
+ std::map<std::string, std::string> out_map; |
+ std::string service; |
+ std::string service2; |
+ service = "testservice"; |
+ service2 = "othertestservice"; |
+ |
+ EXPECT_TRUE(table_->GetAllTokens(&out_map)); |
+ EXPECT_TRUE(out_map.empty()); |
+ |
+ // Check that get all tokens works |
+ EXPECT_TRUE(table_->SetTokenForService(service, "pepperoni")); |
+ EXPECT_TRUE(table_->SetTokenForService(service2, "steak")); |
+ EXPECT_TRUE(table_->GetAllTokens(&out_map)); |
+ EXPECT_EQ("pepperoni", out_map.find(service)->second); |
+ EXPECT_EQ("steak", out_map.find(service2)->second); |
+ out_map.clear(); |
+ |
+ // Purge |
+ EXPECT_TRUE(table_->RemoveAllTokens()); |
+ EXPECT_TRUE(table_->GetAllTokens(&out_map)); |
+ EXPECT_TRUE(out_map.empty()); |
+ |
+ // Check that you can still add it back in |
+ EXPECT_TRUE(table_->SetTokenForService(service, "cheese")); |
+ EXPECT_TRUE(table_->GetAllTokens(&out_map)); |
+ EXPECT_EQ("cheese", out_map.find(service)->second); |
+} |
+ |
+TEST_F(TokenServiceTableTest, MAYBE_TokenServiceGetSet) { |
+ std::map<std::string, std::string> out_map; |
+ std::string service; |
+ service = "testservice"; |
+ |
+ EXPECT_TRUE(table_->GetAllTokens(&out_map)); |
+ EXPECT_TRUE(out_map.empty()); |
+ |
+ EXPECT_TRUE(table_->SetTokenForService(service, "pepperoni")); |
+ EXPECT_TRUE(table_->GetAllTokens(&out_map)); |
+ EXPECT_EQ("pepperoni", out_map.find(service)->second); |
+ out_map.clear(); |
+ |
+ // try blanking it - won't remove it from the db though! |
+ EXPECT_TRUE(table_->SetTokenForService(service, std::string())); |
+ EXPECT_TRUE(table_->GetAllTokens(&out_map)); |
+ EXPECT_EQ("", out_map.find(service)->second); |
+ out_map.clear(); |
+ |
+ // try mutating it |
+ EXPECT_TRUE(table_->SetTokenForService(service, "ham")); |
+ EXPECT_TRUE(table_->GetAllTokens(&out_map)); |
+ EXPECT_EQ("ham", out_map.find(service)->second); |
+} |
+ |
+TEST_F(TokenServiceTableTest, MAYBE_TokenServiceRemove) { |
+ std::map<std::string, std::string> out_map; |
+ std::string service; |
+ std::string service2; |
+ service = "testservice"; |
+ service2 = "othertestservice"; |
+ |
+ EXPECT_TRUE(table_->SetTokenForService(service, "pepperoni")); |
+ EXPECT_TRUE(table_->SetTokenForService(service2, "steak")); |
+ EXPECT_TRUE(table_->RemoveTokenForService(service)); |
+ EXPECT_TRUE(table_->GetAllTokens(&out_map)); |
+ EXPECT_EQ(0u, out_map.count(service)); |
+ EXPECT_EQ("steak", out_map.find(service2)->second); |
+} |