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/string_util.h" | 5 #include "base/string_util.h" |
6 #include "base/utf_string_conversions.h" | 6 #include "base/utf_string_conversions.h" |
7 #include "sql/connection.h" | 7 #include "sql/connection.h" |
8 #include "sql/statement.h" | 8 #include "sql/statement.h" |
9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
10 #include "webkit/database/quota_table.h" | 10 #include "webkit/database/quota_table.h" |
11 | 11 |
12 namespace { | 12 namespace { |
13 | 13 |
14 class TestErrorDelegate : public sql::ErrorDelegate { | 14 class TestErrorDelegate : public sql::ErrorDelegate { |
15 public: | 15 public: |
| 16 TestErrorDelegate() {} |
| 17 virtual ~TestErrorDelegate() {} |
| 18 |
16 virtual int OnError(int error, | 19 virtual int OnError(int error, |
17 sql::Connection* connection, | 20 sql::Connection* connection, |
18 sql::Statement* stmt) OVERRIDE { | 21 sql::Statement* stmt) OVERRIDE { |
19 return error; | 22 return error; |
20 } | 23 } |
21 | 24 |
22 protected: | 25 private: |
23 virtual ~TestErrorDelegate() {} | 26 DISALLOW_COPY_AND_ASSIGN(TestErrorDelegate); |
24 }; | 27 }; |
25 | 28 |
26 } // namespace | 29 } // namespace |
27 | 30 |
28 namespace webkit_database { | 31 namespace webkit_database { |
29 | 32 |
30 static bool QuotaTableIsEmpty(sql::Connection* db) { | 33 static bool QuotaTableIsEmpty(sql::Connection* db) { |
31 sql::Statement statement(db->GetCachedStatement( | 34 sql::Statement statement(db->GetCachedStatement( |
32 SQL_FROM_HERE, "SELECT COUNT(*) FROM Quota")); | 35 SQL_FROM_HERE, "SELECT COUNT(*) FROM Quota")); |
33 return (statement.is_valid() && statement.Step() && !statement.ColumnInt(0)); | 36 return (statement.is_valid() && statement.Step() && !statement.ColumnInt(0)); |
34 } | 37 } |
35 | 38 |
36 TEST(QuotaTableTest, TestIt) { | 39 TEST(QuotaTableTest, TestIt) { |
37 // Initialize the 'Quota' table. | 40 // Initialize the 'Quota' table. |
38 sql::Connection db; | 41 sql::Connection db; |
39 | 42 |
40 // Set an error delegate that will make all operations return false on error. | 43 // Set an error delegate that will make all operations return false on error. |
41 scoped_refptr<TestErrorDelegate> error_delegate(new TestErrorDelegate()); | 44 db.set_error_delegate(new TestErrorDelegate()); |
42 db.set_error_delegate(error_delegate); | |
43 | 45 |
44 // Initialize the temp dir and the 'Databases' table. | 46 // Initialize the temp dir and the 'Databases' table. |
45 EXPECT_TRUE(db.OpenInMemory()); | 47 EXPECT_TRUE(db.OpenInMemory()); |
46 QuotaTable quota_table(&db); | 48 QuotaTable quota_table(&db); |
47 EXPECT_TRUE(quota_table.Init()); | 49 EXPECT_TRUE(quota_table.Init()); |
48 | 50 |
49 // The 'Quota' table should be empty. | 51 // The 'Quota' table should be empty. |
50 EXPECT_TRUE(QuotaTableIsEmpty(&db)); | 52 EXPECT_TRUE(QuotaTableIsEmpty(&db)); |
51 | 53 |
52 // Set and check the quota for a new origin. | 54 // Set and check the quota for a new origin. |
53 string16 origin = ASCIIToUTF16("origin"); | 55 string16 origin = ASCIIToUTF16("origin"); |
54 EXPECT_TRUE(quota_table.SetOriginQuota(origin, 1000)); | 56 EXPECT_TRUE(quota_table.SetOriginQuota(origin, 1000)); |
55 EXPECT_EQ(1000, quota_table.GetOriginQuota(origin)); | 57 EXPECT_EQ(1000, quota_table.GetOriginQuota(origin)); |
56 | 58 |
57 // Reset and check the quota for the same origin. | 59 // Reset and check the quota for the same origin. |
58 EXPECT_TRUE(quota_table.SetOriginQuota(origin, 2000)); | 60 EXPECT_TRUE(quota_table.SetOriginQuota(origin, 2000)); |
59 EXPECT_EQ(2000, quota_table.GetOriginQuota(origin)); | 61 EXPECT_EQ(2000, quota_table.GetOriginQuota(origin)); |
60 | 62 |
61 // Clear the quota for an origin | 63 // Clear the quota for an origin |
62 EXPECT_TRUE(quota_table.ClearOriginQuota(origin)); | 64 EXPECT_TRUE(quota_table.ClearOriginQuota(origin)); |
63 EXPECT_TRUE(quota_table.GetOriginQuota(origin) < 0); | 65 EXPECT_TRUE(quota_table.GetOriginQuota(origin) < 0); |
64 | 66 |
65 // Check that there's no quota set for unknown origins. | 67 // Check that there's no quota set for unknown origins. |
66 EXPECT_TRUE(quota_table.GetOriginQuota(ASCIIToUTF16("unknown_origin")) < 0); | 68 EXPECT_TRUE(quota_table.GetOriginQuota(ASCIIToUTF16("unknown_origin")) < 0); |
67 } | 69 } |
68 | 70 |
69 } // namespace webkit_database | 71 } // namespace webkit_database |
OLD | NEW |