OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/path_service.h" | 6 #include "base/path_service.h" |
| 7 #include "base/scoped_temp_dir.h" |
7 #include "base/string_number_conversions.h" | 8 #include "base/string_number_conversions.h" |
8 #include "base/time.h" | 9 #include "base/time.h" |
9 #include "chrome/browser/webdata/token_service_table.h" | 10 #include "chrome/browser/webdata/token_service_table.h" |
10 #include "chrome/browser/webdata/web_database.h" | 11 #include "chrome/browser/webdata/web_database.h" |
11 #include "chrome/common/chrome_paths.h" | 12 #include "chrome/common/chrome_paths.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
13 | 14 |
14 using base::Time; | 15 using base::Time; |
15 | 16 |
16 class TokenServiceTableTest : public testing::Test { | 17 class TokenServiceTableTest : public testing::Test { |
17 public: | 18 public: |
18 TokenServiceTableTest() {} | 19 TokenServiceTableTest() {} |
19 virtual ~TokenServiceTableTest() {} | 20 virtual ~TokenServiceTableTest() {} |
20 | 21 |
21 protected: | 22 protected: |
22 virtual void SetUp() { | 23 virtual void SetUp() { |
23 PathService::Get(chrome::DIR_TEST_DATA, &file_); | 24 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
24 const std::string test_db = "TestWebDatabase" + | 25 file_ = temp_dir_.path().AppendASCII("TestWebDatabase"); |
25 base::Int64ToString(Time::Now().ToTimeT()) + | |
26 ".db"; | |
27 file_ = file_.AppendASCII(test_db); | |
28 file_util::Delete(file_, false); | |
29 } | |
30 | |
31 virtual void TearDown() { | |
32 file_util::Delete(file_, false); | |
33 } | 26 } |
34 | 27 |
35 FilePath file_; | 28 FilePath file_; |
| 29 ScopedTempDir temp_dir_; |
36 | 30 |
37 private: | 31 private: |
38 DISALLOW_COPY_AND_ASSIGN(TokenServiceTableTest); | 32 DISALLOW_COPY_AND_ASSIGN(TokenServiceTableTest); |
39 }; | 33 }; |
40 | 34 |
41 TEST_F(TokenServiceTableTest, TokenServiceGetAllRemoveAll) { | 35 TEST_F(TokenServiceTableTest, TokenServiceGetAllRemoveAll) { |
42 WebDatabase db; | 36 WebDatabase db; |
43 ASSERT_EQ(sql::INIT_OK, db.Init(file_)); | 37 ASSERT_EQ(sql::INIT_OK, db.Init(file_)); |
44 | 38 |
45 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... |
92 EXPECT_TRUE(db.GetTokenServiceTable()->SetTokenForService(service, "")); | 86 EXPECT_TRUE(db.GetTokenServiceTable()->SetTokenForService(service, "")); |
93 EXPECT_TRUE(db.GetTokenServiceTable()->GetAllTokens(&out_map)); | 87 EXPECT_TRUE(db.GetTokenServiceTable()->GetAllTokens(&out_map)); |
94 EXPECT_EQ(out_map.find(service)->second, ""); | 88 EXPECT_EQ(out_map.find(service)->second, ""); |
95 out_map.clear(); | 89 out_map.clear(); |
96 | 90 |
97 // try mutating it | 91 // try mutating it |
98 EXPECT_TRUE(db.GetTokenServiceTable()->SetTokenForService(service, "ham")); | 92 EXPECT_TRUE(db.GetTokenServiceTable()->SetTokenForService(service, "ham")); |
99 EXPECT_TRUE(db.GetTokenServiceTable()->GetAllTokens(&out_map)); | 93 EXPECT_TRUE(db.GetTokenServiceTable()->GetAllTokens(&out_map)); |
100 EXPECT_EQ(out_map.find(service)->second, "ham"); | 94 EXPECT_EQ(out_map.find(service)->second, "ham"); |
101 } | 95 } |
OLD | NEW |