| 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/bind.h" | 5 #include "base/bind.h" |
| 6 #include "base/message_loop.h" | 6 #include "base/message_loop.h" |
| 7 #include "base/threading/thread.h" | 7 #include "base/threading/thread.h" |
| 8 #include "base/utf_string_conversions.h" | 8 #include "base/utf_string_conversions.h" |
| 9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
| 10 #include "webkit/database/database_connections.h" | 10 #include "webkit/database/database_connections.h" |
| 11 | 11 |
| 12 namespace webkit_database { | 12 namespace webkit_database { |
| 13 | 13 |
| 14 namespace { | 14 namespace { |
| 15 | 15 |
| 16 void RemoveConnectionTask( | 16 void RemoveConnectionTask( |
| 17 const string16& origin_id, const string16& database_name, | 17 const base::string16& origin_id, const base::string16& database_name, |
| 18 scoped_refptr<DatabaseConnectionsWrapper> obj, | 18 scoped_refptr<DatabaseConnectionsWrapper> obj, |
| 19 bool* did_task_execute) { | 19 bool* did_task_execute) { |
| 20 *did_task_execute = true; | 20 *did_task_execute = true; |
| 21 obj->RemoveOpenConnection(origin_id, database_name); | 21 obj->RemoveOpenConnection(origin_id, database_name); |
| 22 } | 22 } |
| 23 | 23 |
| 24 void ScheduleRemoveConnectionTask( | 24 void ScheduleRemoveConnectionTask( |
| 25 base::Thread* thread, const string16& origin_id, | 25 base::Thread* thread, const base::string16& origin_id, |
| 26 const string16& database_name, | 26 const base::string16& database_name, |
| 27 scoped_refptr<DatabaseConnectionsWrapper> obj, | 27 scoped_refptr<DatabaseConnectionsWrapper> obj, |
| 28 bool* did_task_execute) { | 28 bool* did_task_execute) { |
| 29 thread->message_loop()->PostTask( | 29 thread->message_loop()->PostTask( |
| 30 FROM_HERE, | 30 FROM_HERE, |
| 31 base::Bind(&RemoveConnectionTask, origin_id, database_name, obj, | 31 base::Bind(&RemoveConnectionTask, origin_id, database_name, obj, |
| 32 did_task_execute)); | 32 did_task_execute)); |
| 33 } | 33 } |
| 34 | 34 |
| 35 } // anonymous namespace | 35 } // anonymous namespace |
| 36 | 36 |
| 37 TEST(DatabaseConnectionsTest, DatabaseConnectionsTest) { | 37 TEST(DatabaseConnectionsTest, DatabaseConnectionsTest) { |
| 38 const string16 kOriginId(ASCIIToUTF16("origin_id")); | 38 const base::string16 kOriginId(ASCIIToUTF16("origin_id")); |
| 39 const string16 kName(ASCIIToUTF16("database_name")); | 39 const base::string16 kName(ASCIIToUTF16("database_name")); |
| 40 const string16 kName2(ASCIIToUTF16("database_name2")); | 40 const base::string16 kName2(ASCIIToUTF16("database_name2")); |
| 41 const int64 kSize = 1000; | 41 const int64 kSize = 1000; |
| 42 | 42 |
| 43 DatabaseConnections connections; | 43 DatabaseConnections connections; |
| 44 | 44 |
| 45 EXPECT_TRUE(connections.IsEmpty()); | 45 EXPECT_TRUE(connections.IsEmpty()); |
| 46 EXPECT_FALSE(connections.IsDatabaseOpened(kOriginId, kName)); | 46 EXPECT_FALSE(connections.IsDatabaseOpened(kOriginId, kName)); |
| 47 EXPECT_FALSE(connections.IsOriginUsed(kOriginId)); | 47 EXPECT_FALSE(connections.IsOriginUsed(kOriginId)); |
| 48 | 48 |
| 49 connections.AddConnection(kOriginId, kName); | 49 connections.AddConnection(kOriginId, kName); |
| 50 EXPECT_FALSE(connections.IsEmpty()); | 50 EXPECT_FALSE(connections.IsEmpty()); |
| (...skipping 16 matching lines...) Expand all Loading... |
| 67 EXPECT_FALSE(connections.IsEmpty()); | 67 EXPECT_FALSE(connections.IsEmpty()); |
| 68 EXPECT_TRUE(connections.IsDatabaseOpened(kOriginId, kName)); | 68 EXPECT_TRUE(connections.IsDatabaseOpened(kOriginId, kName)); |
| 69 EXPECT_TRUE(connections.IsOriginUsed(kOriginId)); | 69 EXPECT_TRUE(connections.IsOriginUsed(kOriginId)); |
| 70 connections.AddConnection(kOriginId, kName2); | 70 connections.AddConnection(kOriginId, kName2); |
| 71 EXPECT_TRUE(connections.IsDatabaseOpened(kOriginId, kName2)); | 71 EXPECT_TRUE(connections.IsDatabaseOpened(kOriginId, kName2)); |
| 72 | 72 |
| 73 DatabaseConnections another; | 73 DatabaseConnections another; |
| 74 another.AddConnection(kOriginId, kName); | 74 another.AddConnection(kOriginId, kName); |
| 75 another.AddConnection(kOriginId, kName2); | 75 another.AddConnection(kOriginId, kName2); |
| 76 | 76 |
| 77 std::vector<std::pair<string16, string16> > closed_dbs; | 77 std::vector<std::pair<base::string16, base::string16> > closed_dbs; |
| 78 connections.RemoveConnections(another, &closed_dbs); | 78 connections.RemoveConnections(another, &closed_dbs); |
| 79 EXPECT_EQ(1u, closed_dbs.size()); | 79 EXPECT_EQ(1u, closed_dbs.size()); |
| 80 EXPECT_EQ(kOriginId, closed_dbs[0].first); | 80 EXPECT_EQ(kOriginId, closed_dbs[0].first); |
| 81 EXPECT_EQ(kName2, closed_dbs[0].second); | 81 EXPECT_EQ(kName2, closed_dbs[0].second); |
| 82 EXPECT_FALSE(connections.IsDatabaseOpened(kOriginId, kName2)); | 82 EXPECT_FALSE(connections.IsDatabaseOpened(kOriginId, kName2)); |
| 83 EXPECT_TRUE(connections.IsDatabaseOpened(kOriginId, kName)); | 83 EXPECT_TRUE(connections.IsDatabaseOpened(kOriginId, kName)); |
| 84 EXPECT_EQ(kSize, connections.GetOpenDatabaseSize(kOriginId, kName)); | 84 EXPECT_EQ(kSize, connections.GetOpenDatabaseSize(kOriginId, kName)); |
| 85 another.RemoveAllConnections(); | 85 another.RemoveAllConnections(); |
| 86 connections.RemoveAllConnections(); | 86 connections.RemoveAllConnections(); |
| 87 EXPECT_TRUE(connections.IsEmpty()); | 87 EXPECT_TRUE(connections.IsEmpty()); |
| 88 | 88 |
| 89 // Ensure the return value properly indicates the initial | 89 // Ensure the return value properly indicates the initial |
| 90 // addition and final removal. | 90 // addition and final removal. |
| 91 EXPECT_TRUE(connections.AddConnection(kOriginId, kName)); | 91 EXPECT_TRUE(connections.AddConnection(kOriginId, kName)); |
| 92 EXPECT_FALSE(connections.AddConnection(kOriginId, kName)); | 92 EXPECT_FALSE(connections.AddConnection(kOriginId, kName)); |
| 93 EXPECT_FALSE(connections.AddConnection(kOriginId, kName)); | 93 EXPECT_FALSE(connections.AddConnection(kOriginId, kName)); |
| 94 EXPECT_FALSE(connections.RemoveConnection(kOriginId, kName)); | 94 EXPECT_FALSE(connections.RemoveConnection(kOriginId, kName)); |
| 95 EXPECT_FALSE(connections.RemoveConnection(kOriginId, kName)); | 95 EXPECT_FALSE(connections.RemoveConnection(kOriginId, kName)); |
| 96 EXPECT_TRUE(connections.RemoveConnection(kOriginId, kName)); | 96 EXPECT_TRUE(connections.RemoveConnection(kOriginId, kName)); |
| 97 } | 97 } |
| 98 | 98 |
| 99 TEST(DatabaseConnectionsTest, DatabaseConnectionsWrapperTest) { | 99 TEST(DatabaseConnectionsTest, DatabaseConnectionsWrapperTest) { |
| 100 const string16 kOriginId(ASCIIToUTF16("origin_id")); | 100 const base::string16 kOriginId(ASCIIToUTF16("origin_id")); |
| 101 const string16 kName(ASCIIToUTF16("database_name")); | 101 const base::string16 kName(ASCIIToUTF16("database_name")); |
| 102 | 102 |
| 103 MessageLoop message_loop; | 103 MessageLoop message_loop; |
| 104 scoped_refptr<DatabaseConnectionsWrapper> obj( | 104 scoped_refptr<DatabaseConnectionsWrapper> obj( |
| 105 new DatabaseConnectionsWrapper); | 105 new DatabaseConnectionsWrapper); |
| 106 EXPECT_FALSE(obj->HasOpenConnections()); | 106 EXPECT_FALSE(obj->HasOpenConnections()); |
| 107 obj->AddOpenConnection(kOriginId, kName); | 107 obj->AddOpenConnection(kOriginId, kName); |
| 108 EXPECT_TRUE(obj->HasOpenConnections()); | 108 EXPECT_TRUE(obj->HasOpenConnections()); |
| 109 obj->AddOpenConnection(kOriginId, kName); | 109 obj->AddOpenConnection(kOriginId, kName); |
| 110 EXPECT_TRUE(obj->HasOpenConnections()); | 110 EXPECT_TRUE(obj->HasOpenConnections()); |
| 111 obj->RemoveOpenConnection(kOriginId, kName); | 111 obj->RemoveOpenConnection(kOriginId, kName); |
| (...skipping 23 matching lines...) Expand all Loading... |
| 135 MessageLoop::current()->PostTask( | 135 MessageLoop::current()->PostTask( |
| 136 FROM_HERE, | 136 FROM_HERE, |
| 137 base::Bind(&ScheduleRemoveConnectionTask, &thread, kOriginId, kName, obj, | 137 base::Bind(&ScheduleRemoveConnectionTask, &thread, kOriginId, kName, obj, |
| 138 &did_task_execute)); | 138 &did_task_execute)); |
| 139 obj->WaitForAllDatabasesToClose(); // should return after the task executes | 139 obj->WaitForAllDatabasesToClose(); // should return after the task executes |
| 140 EXPECT_TRUE(did_task_execute); | 140 EXPECT_TRUE(did_task_execute); |
| 141 EXPECT_FALSE(obj->HasOpenConnections()); | 141 EXPECT_FALSE(obj->HasOpenConnections()); |
| 142 } | 142 } |
| 143 | 143 |
| 144 } // namespace webkit_database | 144 } // namespace webkit_database |
| OLD | NEW |