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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
// Copyright (c) 2012 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 <string>
#include "base/file_path.h"
#include "base/file_util.h"
#include "base/files/scoped_temp_dir.h"
#include "chrome/browser/extensions/activity_database.h"
#include "chrome/browser/extensions/api_actions.h"
#include "chrome/browser/extensions/blocked_actions.h"
#include "chrome/browser/extensions/url_actions.h"
#include "sql/statement.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace extensions {
// Check that the database is initialized properly.
TEST(ActivityDatabaseTest, Init) {
base::ScopedTempDir temp_dir;
base::FilePath db_file;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
db_file = temp_dir.path().AppendASCII("ActivityInit.db");
file_util::Delete(db_file, false);
ActivityDatabase* activity_db = new ActivityDatabase();
activity_db->AddRef();
activity_db->Init(db_file);
ASSERT_TRUE(activity_db->initialized());
activity_db->Release();
sql::Connection db;
ASSERT_TRUE(db.Open(db_file));
ASSERT_TRUE(db.DoesTableExist(UrlAction::kTableName));
ASSERT_TRUE(db.DoesTableExist(APIAction::kTableName));
ASSERT_TRUE(db.DoesTableExist(BlockedAction::kTableName));
db.Close();
}
// Check that actions are recorded in the db.
TEST(ActivityDatabaseTest, RecordAction) {
base::ScopedTempDir temp_dir;
base::FilePath db_file;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
db_file = temp_dir.path().AppendASCII("ActivityRecord.db");
file_util::Delete(db_file, false);
ActivityDatabase* activity_db = new ActivityDatabase();
activity_db->AddRef();
activity_db->Init(db_file);
ASSERT_TRUE(activity_db->initialized());
scoped_refptr<APIAction> action = new APIAction(
"punky",
base::Time::Now(),
APIAction::CALL,
APIAction::READ,
APIAction::BOOKMARK,
"brewster",
"");
activity_db->RecordAction(action);
activity_db->Close();
activity_db->Release();
sql::Connection db;
ASSERT_TRUE(db.Open(db_file));
ASSERT_TRUE(db.DoesTableExist(APIAction::kTableName));
std::string sql_str = "SELECT * FROM " +
std::string(APIAction::kTableName);
sql::Statement statement(db.GetUniqueStatement(sql_str.c_str()));
ASSERT_TRUE(statement.Step());
ASSERT_EQ("punky", statement.ColumnString(0));
ASSERT_EQ("CALL", statement.ColumnString(2));
ASSERT_EQ("READ", statement.ColumnString(3));
ASSERT_EQ("BOOKMARK", statement.ColumnString(4));
ASSERT_EQ("brewster", statement.ColumnString(5));
}
// Check that nothing explodes if the DB isn't initialized.
TEST(ActivityDatabaseTest, InitFailure) {
base::ScopedTempDir temp_dir;
base::FilePath db_file;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
db_file = temp_dir.path().AppendASCII("ActivityRecord.db");
file_util::Delete(db_file, false);
ActivityDatabase* activity_db = new ActivityDatabase();
activity_db->AddRef();
scoped_refptr<APIAction> action = new APIAction(
"punky",
base::Time::Now(),
APIAction::CALL,
APIAction::READ,
APIAction::BOOKMARK,
"brewster",
"");
activity_db->RecordAction(action);
activity_db->Close();
activity_db->Release();
}
} // namespace
|
