1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/message_loop.h"
#include "base/task.h"
#include "base/threading/thread.h"
#include "base/utf_string_conversions.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "webkit/database/database_connections.h"
namespace webkit_database {
namespace {
void RemoveConnectionTask(
const string16& origin_id, const string16& database_name,
scoped_refptr<DatabaseConnectionsWrapper> obj,
bool* did_task_execute) {
*did_task_execute = true;
obj->RemoveOpenConnection(origin_id, database_name);
}
void ScheduleRemoveConnectionTask(
base::Thread* thread, const string16& origin_id,
const string16& database_name,
scoped_refptr<DatabaseConnectionsWrapper> obj,
bool* did_task_execute) {
thread->message_loop()->PostTask(FROM_HERE,
NewRunnableFunction(RemoveConnectionTask,
origin_id, database_name, obj, did_task_execute));
}
} // anonymous namespace
TEST(DatabaseConnectionsTest, DatabaseConnectionsWrapperTest) {
string16 kOriginId(ASCIIToUTF16("origin_id"));
string16 kName(ASCIIToUTF16("database_name"));
scoped_refptr<DatabaseConnectionsWrapper> obj(
new DatabaseConnectionsWrapper);
EXPECT_FALSE(obj->HasOpenConnections());
obj->AddOpenConnection(kOriginId, kName);
EXPECT_TRUE(obj->HasOpenConnections());
obj->AddOpenConnection(kOriginId, kName);
EXPECT_TRUE(obj->HasOpenConnections());
obj->RemoveOpenConnection(kOriginId, kName);
EXPECT_TRUE(obj->HasOpenConnections());
obj->RemoveOpenConnection(kOriginId, kName);
EXPECT_FALSE(obj->HasOpenConnections());
obj->WaitForAllDatabasesToClose(); // should return immediately
// Test WaitForAllDatabasesToClose with the last connection
// being removed on the current thread.
obj->AddOpenConnection(kOriginId, kName);
bool did_task_execute = false;
MessageLoop::current()->PostTask(FROM_HERE,
NewRunnableFunction(&RemoveConnectionTask,
kOriginId, kName, obj, &did_task_execute));
obj->WaitForAllDatabasesToClose(); // should return after the task executes
EXPECT_TRUE(did_task_execute);
EXPECT_FALSE(obj->HasOpenConnections());
// Test WaitForAllDatabasesToClose with the last connection
// being removed on another thread.
obj->AddOpenConnection(kOriginId, kName);
base::Thread thread("WrapperTestThread");
thread.Start();
did_task_execute = false;
MessageLoop::current()->PostTask(FROM_HERE,
NewRunnableFunction(ScheduleRemoveConnectionTask,
&thread, kOriginId, kName,
obj, &did_task_execute));
obj->WaitForAllDatabasesToClose(); // should return after the task executes
EXPECT_TRUE(did_task_execute);
EXPECT_FALSE(obj->HasOpenConnections());
}
} // namespace webkit_database
|
