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 |