This source file includes following definitions.
- RemoveConnectionTask
- ScheduleRemoveConnectionTask
- TEST
- TEST
#include "base/bind.h"
#include "base/message_loop/message_loop.h"
#include "base/strings/utf_string_conversions.h"
#include "base/threading/thread.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "webkit/common/database/database_connections.h"
using base::ASCIIToUTF16;
namespace webkit_database {
namespace {
void RemoveConnectionTask(
const std::string& origin_id, const base::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 std::string& origin_id,
const base::string16& database_name,
scoped_refptr<DatabaseConnectionsWrapper> obj,
bool* did_task_execute) {
thread->message_loop()->PostTask(
FROM_HERE,
base::Bind(&RemoveConnectionTask, origin_id, database_name, obj,
did_task_execute));
}
}
TEST(DatabaseConnectionsTest, DatabaseConnectionsTest) {
const std::string kOriginId("origin_id");
const base::string16 kName(ASCIIToUTF16("database_name"));
const base::string16 kName2(ASCIIToUTF16("database_name2"));
const int64 kSize = 1000;
DatabaseConnections connections;
EXPECT_TRUE(connections.IsEmpty());
EXPECT_FALSE(connections.IsDatabaseOpened(kOriginId, kName));
EXPECT_FALSE(connections.IsOriginUsed(kOriginId));
connections.AddConnection(kOriginId, kName);
EXPECT_FALSE(connections.IsEmpty());
EXPECT_TRUE(connections.IsDatabaseOpened(kOriginId, kName));
EXPECT_TRUE(connections.IsOriginUsed(kOriginId));
EXPECT_EQ(0, connections.GetOpenDatabaseSize(kOriginId, kName));
connections.SetOpenDatabaseSize(kOriginId, kName, kSize);
EXPECT_EQ(kSize, connections.GetOpenDatabaseSize(kOriginId, kName));
connections.RemoveConnection(kOriginId, kName);
EXPECT_TRUE(connections.IsEmpty());
EXPECT_FALSE(connections.IsDatabaseOpened(kOriginId, kName));
EXPECT_FALSE(connections.IsOriginUsed(kOriginId));
connections.AddConnection(kOriginId, kName);
connections.SetOpenDatabaseSize(kOriginId, kName, kSize);
EXPECT_EQ(kSize, connections.GetOpenDatabaseSize(kOriginId, kName));
connections.AddConnection(kOriginId, kName);
EXPECT_EQ(kSize, connections.GetOpenDatabaseSize(kOriginId, kName));
EXPECT_FALSE(connections.IsEmpty());
EXPECT_TRUE(connections.IsDatabaseOpened(kOriginId, kName));
EXPECT_TRUE(connections.IsOriginUsed(kOriginId));
connections.AddConnection(kOriginId, kName2);
EXPECT_TRUE(connections.IsDatabaseOpened(kOriginId, kName2));
DatabaseConnections another;
another.AddConnection(kOriginId, kName);
another.AddConnection(kOriginId, kName2);
std::vector<std::pair<std::string, base::string16> > closed_dbs;
connections.RemoveConnections(another, &closed_dbs);
EXPECT_EQ(1u, closed_dbs.size());
EXPECT_EQ(kOriginId, closed_dbs[0].first);
EXPECT_EQ(kName2, closed_dbs[0].second);
EXPECT_FALSE(connections.IsDatabaseOpened(kOriginId, kName2));
EXPECT_TRUE(connections.IsDatabaseOpened(kOriginId, kName));
EXPECT_EQ(kSize, connections.GetOpenDatabaseSize(kOriginId, kName));
another.RemoveAllConnections();
connections.RemoveAllConnections();
EXPECT_TRUE(connections.IsEmpty());
EXPECT_TRUE(connections.AddConnection(kOriginId, kName));
EXPECT_FALSE(connections.AddConnection(kOriginId, kName));
EXPECT_FALSE(connections.AddConnection(kOriginId, kName));
EXPECT_FALSE(connections.RemoveConnection(kOriginId, kName));
EXPECT_FALSE(connections.RemoveConnection(kOriginId, kName));
EXPECT_TRUE(connections.RemoveConnection(kOriginId, kName));
}
TEST(DatabaseConnectionsTest, DatabaseConnectionsWrapperTest) {
const std::string kOriginId("origin_id");
const base::string16 kName(ASCIIToUTF16("database_name"));
base::MessageLoop message_loop;
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();
obj->AddOpenConnection(kOriginId, kName);
bool did_task_execute = false;
base::MessageLoop::current()->PostTask(
FROM_HERE,
base::Bind(&RemoveConnectionTask, kOriginId, kName, obj,
&did_task_execute));
obj->WaitForAllDatabasesToClose();
EXPECT_TRUE(did_task_execute);
EXPECT_FALSE(obj->HasOpenConnections());
obj->AddOpenConnection(kOriginId, kName);
base::Thread thread("WrapperTestThread");
thread.Start();
did_task_execute = false;
base::MessageLoop::current()->PostTask(
FROM_HERE,
base::Bind(&ScheduleRemoveConnectionTask, &thread, kOriginId, kName, obj,
&did_task_execute));
obj->WaitForAllDatabasesToClose();
EXPECT_TRUE(did_task_execute);
EXPECT_FALSE(obj->HasOpenConnections());
}
}