This source file includes following definitions.
- Close
- AddInvalidMetric
- AddInvalidEvent
- GetNumberOfMetricEntries
- GetNumberOfEventEntries
- GetNumberOfEntries
- GetTime
- SetUp
- PopulateDB
- InitEvents
- SetUp
- PopulateDB
- TEST
- TEST
- TEST
- TEST
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
#include <algorithm>
#include <string>
#include <vector>
#include "base/files/file_path.h"
#include "base/files/scoped_temp_dir.h"
#include "base/memory/scoped_ptr.h"
#include "base/time/time.h"
#include "chrome/browser/performance_monitor/database.h"
#include "chrome/browser/performance_monitor/key_builder.h"
#include "chrome/browser/performance_monitor/metric.h"
#include "chrome/browser/performance_monitor/performance_monitor_util.h"
#include "extensions/common/extension.h"
#include "extensions/common/manifest.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/leveldatabase/src/include/leveldb/db.h"
#include "third_party/leveldatabase/src/include/leveldb/iterator.h"
using extensions::Extension;
using extensions::Manifest;
namespace performance_monitor {
class DatabaseTestHelper {
public:
explicit DatabaseTestHelper(Database* database) : database_(database) { }
~DatabaseTestHelper() { }
bool Close() { return database_->Close(); }
bool AddInvalidMetric(std::string activity, Metric metric) {
std::string metric_key =
database_->key_builder_->CreateMetricKey(database_->clock_->GetTime(),
metric.type,
activity);
leveldb::Status status =
database_->metric_db_->Put(database_->write_options_,
metric_key,
metric.ValueAsString());
return status.ok();
}
bool AddInvalidEvent(base::Time time, EventType type) {
std::string key = database_->key_builder_->CreateEventKey(time, type);
leveldb::Status status =
database_->event_db_->Put(database_->write_options_, key, "fake_event");
return status.ok();
}
size_t GetNumberOfMetricEntries() {
return GetNumberOfEntries(database_->metric_db_.get());
}
size_t GetNumberOfEventEntries() {
return GetNumberOfEntries(database_->event_db_.get());
}
private:
size_t GetNumberOfEntries(leveldb::DB* level_db) {
size_t number_of_entries = 0;
scoped_ptr<leveldb::Iterator> iter(
level_db->NewIterator(database_->read_options_));
for (iter->SeekToFirst(); iter->Valid(); iter->Next())
++number_of_entries;
return number_of_entries;
}
Database* database_;
};
class TestingClock : public Database::Clock {
public:
TestingClock()
: counter_(0) {
}
explicit TestingClock(const TestingClock& other)
: counter_(other.counter_) {
}
virtual ~TestingClock() {}
virtual base::Time GetTime() OVERRIDE {
return base::Time::FromInternalValue(++counter_);
}
private:
int64 counter_;
};
class PerformanceMonitorDatabaseEventTest : public ::testing::Test {
protected:
PerformanceMonitorDatabaseEventTest() {
clock_ = new TestingClock();
CHECK(temp_dir_.CreateUniqueTempDir());
db_ = Database::Create(temp_dir_.path());
CHECK(db_.get());
db_->set_clock(scoped_ptr<Database::Clock>(clock_));
}
virtual void SetUp() {
ASSERT_TRUE(db_.get());
PopulateDB();
}
void PopulateDB() {
InitEvents();
db_->AddEvent(*install_event_1_.get());
db_->AddEvent(*install_event_2_.get());
db_->AddEvent(*uninstall_event_1_.get());
db_->AddEvent(*uninstall_event_2_.get());
}
scoped_ptr<Database> db_;
Database::Clock* clock_;
base::ScopedTempDir temp_dir_;
scoped_ptr<Event> install_event_1_;
scoped_ptr<Event> install_event_2_;
scoped_ptr<Event> uninstall_event_1_;
scoped_ptr<Event> uninstall_event_2_;
private:
void InitEvents() {
install_event_1_ = util::CreateExtensionEvent(
EVENT_EXTENSION_INSTALL, clock_->GetTime(), "a", "extension 1",
"http://foo.com", static_cast<int>(Manifest::UNPACKED), "0.1",
"Test Test");
install_event_2_ = util::CreateExtensionEvent(
EVENT_EXTENSION_INSTALL, clock_->GetTime(), "b", "extension 2",
"http://bar.com", static_cast<int>(Manifest::UNPACKED), "0.1",
"Test Test");
uninstall_event_1_ = util::CreateExtensionEvent(
EVENT_EXTENSION_UNINSTALL, clock_->GetTime(), "a", "extension 1",
"http://foo.com", static_cast<int>(Manifest::UNPACKED), "0.1",
"Test Test");
uninstall_event_2_ = util::CreateExtensionEvent(
EVENT_EXTENSION_UNINSTALL, clock_->GetTime(), "b", "extension 2",
"http://bar.com", static_cast<int>(Manifest::UNPACKED), "0.1",
"Test Test");
}
};
class PerformanceMonitorDatabaseMetricTest : public ::testing::Test {
protected:
PerformanceMonitorDatabaseMetricTest() {
clock_ = new TestingClock();
CHECK(temp_dir_.CreateUniqueTempDir());
db_ = Database::Create(temp_dir_.path());
CHECK(db_.get());
db_->set_clock(scoped_ptr<Database::Clock>(clock_));
activity_ = std::string("A");
}
virtual void SetUp() {
ASSERT_TRUE(db_.get());
PopulateDB();
}
void PopulateDB() {
db_->AddMetric(kProcessChromeAggregate,
Metric(METRIC_CPU_USAGE, clock_->GetTime(), 50.5));
db_->AddMetric(activity_,
Metric(METRIC_CPU_USAGE, clock_->GetTime(), 13.1));
db_->AddMetric(kProcessChromeAggregate,
Metric(METRIC_PRIVATE_MEMORY_USAGE,
clock_->GetTime(),
1000000.0));
db_->AddMetric(activity_,
Metric(METRIC_PRIVATE_MEMORY_USAGE,
clock_->GetTime(),
3000000.0));
}
scoped_ptr<Database> db_;
Database::Clock* clock_;
base::ScopedTempDir temp_dir_;
std::string activity_;
};
TEST(PerformanceMonitorDatabaseSetupTest, OpenClose) {
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
scoped_ptr<Database> db = Database::Create(temp_dir.path());
ASSERT_TRUE(db.get());
DatabaseTestHelper helper(db.get());
ASSERT_TRUE(helper.Close());
}
TEST(PerformanceMonitorDatabaseSetupTest, ActiveInterval) {
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
TestingClock* clock_1 = new TestingClock();
base::Time start_time = clock_1->GetTime();
scoped_ptr<Database> db_1 = Database::Create(temp_dir.path());
db_1->set_clock(scoped_ptr<Database::Clock>(clock_1));
DatabaseTestHelper helper1(db_1.get());
db_1->AddStateValue("test", "test");
ASSERT_TRUE(db_1.get());
ASSERT_TRUE(helper1.Close());
TestingClock* clock_2 = new TestingClock(*clock_1);
base::Time mid_time = clock_2->GetTime();
scoped_ptr<Database> db_2 = Database::Create(temp_dir.path());
db_2->set_clock(scoped_ptr<Database::Clock>(clock_2));
DatabaseTestHelper helper2(db_2.get());
db_2->AddStateValue("test", "test");
ASSERT_TRUE(db_2.get());
ASSERT_TRUE(helper2.Close());
TestingClock* clock_3 = new TestingClock(*clock_2);
base::Time end_time = clock_3->GetTime();
scoped_ptr<Database> db_3 = Database::Create(temp_dir.path());
db_3->set_clock(scoped_ptr<Database::Clock>(clock_3));
db_3->AddStateValue("test", "test");
ASSERT_TRUE(db_3.get());
std::vector<TimeRange> active_interval = db_3->GetActiveIntervals(start_time,
end_time);
ASSERT_EQ(active_interval.size(), 2u);
ASSERT_TRUE(active_interval[0].start > start_time &&
active_interval[0].end < mid_time);
ASSERT_TRUE(active_interval[1].start > mid_time &&
active_interval[1].end < end_time);
}
TEST(PerformanceMonitorDatabaseSetupTest,
ActiveIntervalRetrievalDuringActiveInterval) {
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
TestingClock* clock = new TestingClock();
scoped_ptr<Database> db = Database::Create(temp_dir.path());
db->set_clock(scoped_ptr<Database::Clock>(clock));
db->AddStateValue("test", "test");
base::Time start_time = clock->GetTime();
db->AddStateValue("test", "test");
base::Time end_time = clock->GetTime();
db->AddStateValue("test", "test");
std::vector<TimeRange> active_interval = db->GetActiveIntervals(start_time,
end_time);
ASSERT_EQ(1u, active_interval.size());
EXPECT_LT(active_interval[0].start, start_time);
EXPECT_GT(active_interval[0].end, start_time);
}
TEST(PerformanceMonitorDatabaseSetupTest,
ActiveIntervalRetrievalAfterActiveInterval) {
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
TestingClock* clock = new TestingClock();
scoped_ptr<Database> db = Database::Create(temp_dir.path());
db->set_clock(scoped_ptr<Database::Clock>(clock));
db->AddStateValue("test", "test");
base::Time start_time = clock->GetTime();
base::Time end_time = clock->GetTime();
std::vector<TimeRange> active_interval = db->GetActiveIntervals(start_time,
end_time);
EXPECT_TRUE(active_interval.empty());
}
TEST_F(PerformanceMonitorDatabaseEventTest, GetAllEvents) {
Database::EventVector events = db_->GetEvents();
ASSERT_EQ(4u, events.size());
EXPECT_TRUE(events[0]->data()->Equals(install_event_1_->data()));
EXPECT_TRUE(events[1]->data()->Equals(install_event_2_->data()));
EXPECT_TRUE(events[2]->data()->Equals(uninstall_event_1_->data()));
EXPECT_TRUE(events[3]->data()->Equals(uninstall_event_2_->data()));
}
TEST_F(PerformanceMonitorDatabaseMetricTest, GetMaxMetric) {
Metric stat;
EXPECT_EQ(0.0, db_->GetMaxStatsForActivityAndMetric(activity_,
METRIC_PAGE_LOAD_TIME));
EXPECT_EQ(1000000,
db_->GetMaxStatsForActivityAndMetric(METRIC_PRIVATE_MEMORY_USAGE));
db_->AddMetric(kProcessChromeAggregate,
Metric(METRIC_PRIVATE_MEMORY_USAGE, clock_->GetTime(), 99.0));
EXPECT_EQ(1000000,
db_->GetMaxStatsForActivityAndMetric(METRIC_PRIVATE_MEMORY_USAGE));
db_->AddMetric(kProcessChromeAggregate,
Metric(METRIC_PRIVATE_MEMORY_USAGE,
clock_->GetTime(),
6000000.0));
EXPECT_EQ(6000000,
db_->GetMaxStatsForActivityAndMetric(METRIC_PRIVATE_MEMORY_USAGE));
}
TEST_F(PerformanceMonitorDatabaseEventTest, GetAllEventTypes) {
Database::EventTypeSet types = db_->GetEventTypes();
ASSERT_EQ(2u, types.size());
ASSERT_EQ(1u, types.count(EVENT_EXTENSION_INSTALL));
ASSERT_EQ(1u, types.count(EVENT_EXTENSION_UNINSTALL));
}
TEST_F(PerformanceMonitorDatabaseEventTest, GetEventInTimeRange) {
base::Time start_time = clock_->GetTime();
scoped_ptr<Event> crash_event = util::CreateRendererFailureEvent(
clock_->GetTime(), EVENT_RENDERER_CRASH, "http://www.google.com/");
db_->AddEvent(*crash_event.get());
Database::EventVector events = db_->GetEvents(start_time, clock_->GetTime());
ASSERT_EQ(1u, events.size());
EXPECT_TRUE(events[0]->data()->Equals(crash_event->data()));
}
TEST_F(PerformanceMonitorDatabaseEventTest, GetInstallEvents) {
Database::EventVector events = db_->GetEvents(EVENT_EXTENSION_INSTALL);
ASSERT_EQ(2u, events.size());
EXPECT_TRUE(events[0]->data()->Equals(install_event_1_->data()));
EXPECT_TRUE(events[1]->data()->Equals(install_event_2_->data()));
}
TEST_F(PerformanceMonitorDatabaseEventTest, InvalidEvents) {
DatabaseTestHelper helper(db_.get());
size_t original_number_of_entries = helper.GetNumberOfEventEntries();
ASSERT_TRUE(helper.AddInvalidEvent(
clock_->GetTime(), EVENT_EXTENSION_INSTALL));
ASSERT_EQ(original_number_of_entries + 1u, helper.GetNumberOfEventEntries());
Database::EventVector events = db_->GetEvents();
ASSERT_EQ(original_number_of_entries, events.size());
ASSERT_EQ(original_number_of_entries, helper.GetNumberOfEventEntries());
}
TEST_F(PerformanceMonitorDatabaseEventTest, GetUnusedEventType) {
Database::EventVector events = db_->GetEvents(EVENT_EXTENSION_DISABLE);
ASSERT_TRUE(events.empty());
events = db_->GetEvents(EVENT_EXTENSION_DISABLE, clock_->GetTime(),
clock_->GetTime());
ASSERT_TRUE(events.empty());
}
TEST_F(PerformanceMonitorDatabaseEventTest, GetEventsTimeRange) {
base::Time start_time = clock_->GetTime();
scoped_ptr<Event> new_install_event =
util::CreateExtensionEvent(EVENT_EXTENSION_INSTALL, clock_->GetTime(),
"c", "test extension", "http://foo.com",
static_cast<int>(Manifest::UNPACKED), "0.1",
"Test Test");
scoped_ptr<Event> new_uninstall_event =
util::CreateExtensionEvent(EVENT_EXTENSION_UNINSTALL, clock_->GetTime(),
"c", "test extension", "http://foo.com",
static_cast<int>(Manifest::UNPACKED), "0.1",
"Test Test");
base::Time end_time = clock_->GetTime();
db_->AddEvent(*new_install_event.get());
db_->AddEvent(*new_uninstall_event.get());
Database::EventVector events = db_->GetEvents(start_time, end_time);
ASSERT_EQ(2u, events.size());
EXPECT_TRUE(events[0]->data()->Equals(new_install_event->data()));
EXPECT_TRUE(events[1]->data()->Equals(new_uninstall_event->data()));
events = db_->GetEvents(
EVENT_EXTENSION_INSTALL, start_time, end_time);
ASSERT_EQ(1u, events.size());
EXPECT_TRUE(events[0]->data()->Equals(new_install_event->data()));
}
TEST_F(PerformanceMonitorDatabaseMetricTest, GetActiveMetrics) {
Database::MetricTypeSet active_metrics =
db_->GetActiveMetrics(base::Time(), clock_->GetTime());
Database::MetricTypeSet expected_metrics;
expected_metrics.insert(METRIC_CPU_USAGE);
expected_metrics.insert(METRIC_PRIVATE_MEMORY_USAGE);
EXPECT_EQ(expected_metrics, active_metrics);
}
TEST_F(PerformanceMonitorDatabaseMetricTest, GetRecentMetric) {
Metric stat;
ASSERT_TRUE(db_->GetRecentStatsForActivityAndMetric(activity_,
METRIC_PRIVATE_MEMORY_USAGE, &stat));
EXPECT_EQ(3000000, stat.value);
ASSERT_TRUE(db_->GetRecentStatsForActivityAndMetric(METRIC_CPU_USAGE, &stat));
EXPECT_EQ(50.5, stat.value);
base::ScopedTempDir dir;
CHECK(dir.CreateUniqueTempDir());
scoped_ptr<Database> db = Database::Create(dir.path());
CHECK(db.get());
db->set_clock(scoped_ptr<Database::Clock>(new TestingClock()));
EXPECT_FALSE(db->GetRecentStatsForActivityAndMetric(METRIC_CPU_USAGE, &stat));
}
TEST_F(PerformanceMonitorDatabaseMetricTest, GetState) {
std::string key("version");
std::string value("1.0.0.0.1");
db_->AddStateValue(key, value);
EXPECT_EQ(db_->GetStateValue(key), value);
}
TEST_F(PerformanceMonitorDatabaseMetricTest, GetStateOverride) {
std::string key("version");
std::string value_1("1.0.0.0.0");
std::string value_2("1.0.0.0.1");
db_->AddStateValue(key, value_1);
db_->AddStateValue(key, value_2);
EXPECT_EQ(db_->GetStateValue(key), value_2);
}
TEST_F(PerformanceMonitorDatabaseMetricTest, GetStatsForActivityAndMetric) {
Database::MetricVector stats = *db_->GetStatsForActivityAndMetric(
activity_, METRIC_CPU_USAGE, base::Time(), clock_->GetTime());
ASSERT_EQ(1u, stats.size());
EXPECT_EQ(13.1, stats[0].value);
base::Time before = clock_->GetTime();
db_->AddMetric(activity_,
Metric(METRIC_CPU_USAGE, clock_->GetTime(), 18.0));
stats = *db_->GetStatsForActivityAndMetric(activity_, METRIC_CPU_USAGE,
before, clock_->GetTime());
ASSERT_EQ(1u, stats.size());
EXPECT_EQ(18, stats[0].value);
stats = *db_->GetStatsForActivityAndMetric(activity_, METRIC_CPU_USAGE);
ASSERT_EQ(2u, stats.size());
EXPECT_EQ(13.1, stats[0].value);
EXPECT_EQ(18, stats[1].value);
stats = *db_->GetStatsForActivityAndMetric(METRIC_PRIVATE_MEMORY_USAGE);
ASSERT_EQ(1u, stats.size());
EXPECT_EQ(1000000, stats[0].value);
stats = *db_->GetStatsForActivityAndMetric(activity_, METRIC_CPU_USAGE,
clock_->GetTime(),
clock_->GetTime());
EXPECT_TRUE(stats.empty());
}
TEST_F(PerformanceMonitorDatabaseMetricTest, GetStatsForMetricByActivity) {
Database::MetricVectorMap stats_map = db_->GetStatsForMetricByActivity(
METRIC_CPU_USAGE, base::Time(), clock_->GetTime());
ASSERT_EQ(2u, stats_map.size());
linked_ptr<Database::MetricVector> stats = stats_map[activity_];
ASSERT_EQ(1u, stats->size());
EXPECT_EQ(13.1, stats->at(0).value);
stats = stats_map[kProcessChromeAggregate];
ASSERT_EQ(1u, stats->size());
EXPECT_EQ(50.5, stats->at(0).value);
stats_map = db_->GetStatsForMetricByActivity(
METRIC_CPU_USAGE, clock_->GetTime(), clock_->GetTime());
EXPECT_EQ(0u, stats_map.size());
base::Time before = clock_->GetTime();
db_->AddMetric(activity_,
Metric(METRIC_CPU_USAGE, clock_->GetTime(), 18.0));
stats_map = db_->GetStatsForMetricByActivity(METRIC_CPU_USAGE, before,
clock_->GetTime());
ASSERT_EQ(1u, stats_map.size());
stats = stats_map[activity_];
ASSERT_EQ(1u, stats->size());
EXPECT_EQ(18, stats->at(0).value);
}
TEST_F(PerformanceMonitorDatabaseMetricTest, InvalidMetrics) {
DatabaseTestHelper helper(db_.get());
Metric invalid_metric(METRIC_CPU_USAGE, clock_->GetTime(), -5.0);
ASSERT_FALSE(invalid_metric.IsValid());
size_t original_number_of_entries = helper.GetNumberOfMetricEntries();
Database::MetricVector stats = *db_->GetStatsForActivityAndMetric(
activity_, METRIC_CPU_USAGE, base::Time(), clock_->GetTime());
size_t original_number_of_cpu_entries = stats.size();
EXPECT_FALSE(db_->AddMetric(invalid_metric));
stats = *db_->GetStatsForActivityAndMetric(
activity_, METRIC_CPU_USAGE, base::Time(), clock_->GetTime());
ASSERT_EQ(original_number_of_cpu_entries, stats.size());
ASSERT_TRUE(helper.AddInvalidMetric(activity_, invalid_metric));
ASSERT_EQ(original_number_of_entries + 1u, helper.GetNumberOfMetricEntries());
stats = *db_->GetStatsForActivityAndMetric(
activity_, METRIC_CPU_USAGE, base::Time(), clock_->GetTime());
ASSERT_EQ(original_number_of_cpu_entries, stats.size());
ASSERT_EQ(original_number_of_entries, helper.GetNumberOfMetricEntries());
}
TEST_F(PerformanceMonitorDatabaseMetricTest, GetFullRange) {
db_->AddMetric(kProcessChromeAggregate,
Metric(METRIC_CPU_USAGE, clock_->GetTime(), 3.4));
db_->AddMetric(kProcessChromeAggregate,
Metric(METRIC_CPU_USAGE, clock_->GetTime(), 21.0));
Database::MetricVector stats =
*db_->GetStatsForActivityAndMetric(METRIC_CPU_USAGE);
ASSERT_EQ(3u, stats.size());
ASSERT_EQ(50.5, stats[0].value);
ASSERT_EQ(3.4, stats[1].value);
ASSERT_EQ(21, stats[2].value);
}
TEST_F(PerformanceMonitorDatabaseMetricTest, GetRange) {
base::Time start = clock_->GetTime();
db_->AddMetric(kProcessChromeAggregate,
Metric(METRIC_CPU_USAGE, clock_->GetTime(), 3.0));
db_->AddMetric(kProcessChromeAggregate,
Metric(METRIC_CPU_USAGE, clock_->GetTime(), 9.0));
base::Time end = clock_->GetTime();
db_->AddMetric(kProcessChromeAggregate,
Metric(METRIC_CPU_USAGE, clock_->GetTime(), 21.0));
Database::MetricVector stats =
*db_->GetStatsForActivityAndMetric(METRIC_CPU_USAGE, start, end);
ASSERT_EQ(2u, stats.size());
ASSERT_EQ(3, stats[0].value);
ASSERT_EQ(9, stats[1].value);
}
}