| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/file_util.h" | 5 #include "base/file_util.h" |
| 6 #include "base/files/scoped_temp_dir.h" |
| 6 #include "base/path_service.h" | 7 #include "base/path_service.h" |
| 7 #include "base/scoped_temp_dir.h" | |
| 8 #include "base/string_number_conversions.h" | 8 #include "base/string_number_conversions.h" |
| 9 #include "base/time.h" | 9 #include "base/time.h" |
| 10 #include "chrome/browser/webdata/token_service_table.h" | 10 #include "chrome/browser/webdata/token_service_table.h" |
| 11 #include "chrome/browser/webdata/web_database.h" | 11 #include "chrome/browser/webdata/web_database.h" |
| 12 #include "chrome/common/chrome_paths.h" | 12 #include "chrome/common/chrome_paths.h" |
| 13 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
| 14 | 14 |
| 15 using base::Time; | 15 using base::Time; |
| 16 | 16 |
| 17 class TokenServiceTableTest : public testing::Test { | 17 class TokenServiceTableTest : public testing::Test { |
| 18 public: | 18 public: |
| 19 TokenServiceTableTest() {} | 19 TokenServiceTableTest() {} |
| 20 virtual ~TokenServiceTableTest() {} | 20 virtual ~TokenServiceTableTest() {} |
| 21 | 21 |
| 22 protected: | 22 protected: |
| 23 virtual void SetUp() { | 23 virtual void SetUp() { |
| 24 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 24 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 25 file_ = temp_dir_.path().AppendASCII("TestWebDatabase"); | 25 file_ = temp_dir_.path().AppendASCII("TestWebDatabase"); |
| 26 } | 26 } |
| 27 | 27 |
| 28 FilePath file_; | 28 FilePath file_; |
| 29 ScopedTempDir temp_dir_; | 29 base::ScopedTempDir temp_dir_; |
| 30 | 30 |
| 31 private: | 31 private: |
| 32 DISALLOW_COPY_AND_ASSIGN(TokenServiceTableTest); | 32 DISALLOW_COPY_AND_ASSIGN(TokenServiceTableTest); |
| 33 }; | 33 }; |
| 34 | 34 |
| 35 TEST_F(TokenServiceTableTest, TokenServiceGetAllRemoveAll) { | 35 TEST_F(TokenServiceTableTest, TokenServiceGetAllRemoveAll) { |
| 36 WebDatabase db; | 36 WebDatabase db; |
| 37 ASSERT_EQ(sql::INIT_OK, db.Init(file_)); | 37 ASSERT_EQ(sql::INIT_OK, db.Init(file_)); |
| 38 | 38 |
| 39 std::map<std::string, std::string> out_map; | 39 std::map<std::string, std::string> out_map; |
| (...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 86 EXPECT_TRUE(db.GetTokenServiceTable()->SetTokenForService(service, "")); | 86 EXPECT_TRUE(db.GetTokenServiceTable()->SetTokenForService(service, "")); |
| 87 EXPECT_TRUE(db.GetTokenServiceTable()->GetAllTokens(&out_map)); | 87 EXPECT_TRUE(db.GetTokenServiceTable()->GetAllTokens(&out_map)); |
| 88 EXPECT_EQ(out_map.find(service)->second, ""); | 88 EXPECT_EQ(out_map.find(service)->second, ""); |
| 89 out_map.clear(); | 89 out_map.clear(); |
| 90 | 90 |
| 91 // try mutating it | 91 // try mutating it |
| 92 EXPECT_TRUE(db.GetTokenServiceTable()->SetTokenForService(service, "ham")); | 92 EXPECT_TRUE(db.GetTokenServiceTable()->SetTokenForService(service, "ham")); |
| 93 EXPECT_TRUE(db.GetTokenServiceTable()->GetAllTokens(&out_map)); | 93 EXPECT_TRUE(db.GetTokenServiceTable()->GetAllTokens(&out_map)); |
| 94 EXPECT_EQ(out_map.find(service)->second, "ham"); | 94 EXPECT_EQ(out_map.find(service)->second, "ham"); |
| 95 } | 95 } |
| OLD | NEW |