| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "chrome/test/chromedriver/performance_logger.h" | 5 #include "chrome/test/chromedriver/performance_logger.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 | 8 |
| 9 #include <memory> | 9 #include <memory> |
| 10 #include <utility> | 10 #include <utility> |
| 11 #include <vector> |
| 11 | 12 |
| 12 #include "base/compiler_specific.h" | 13 #include "base/compiler_specific.h" |
| 13 #include "base/format_macros.h" | 14 #include "base/format_macros.h" |
| 14 #include "base/json/json_reader.h" | 15 #include "base/json/json_reader.h" |
| 15 #include "base/memory/scoped_vector.h" | 16 #include "base/memory/ptr_util.h" |
| 16 #include "base/time/time.h" | 17 #include "base/time/time.h" |
| 17 #include "base/values.h" | 18 #include "base/values.h" |
| 18 #include "chrome/test/chromedriver/chrome/devtools_client_impl.h" | 19 #include "chrome/test/chromedriver/chrome/devtools_client_impl.h" |
| 19 #include "chrome/test/chromedriver/chrome/log.h" | 20 #include "chrome/test/chromedriver/chrome/log.h" |
| 20 #include "chrome/test/chromedriver/chrome/status.h" | 21 #include "chrome/test/chromedriver/chrome/status.h" |
| 21 #include "chrome/test/chromedriver/chrome/stub_devtools_client.h" | 22 #include "chrome/test/chromedriver/chrome/stub_devtools_client.h" |
| 22 #include "chrome/test/chromedriver/session.h" | 23 #include "chrome/test/chromedriver/session.h" |
| 23 #include "testing/gtest/include/gtest/gtest.h" | 24 #include "testing/gtest/include/gtest/gtest.h" |
| 24 | 25 |
| 25 namespace { | 26 namespace { |
| (...skipping 11 matching lines...) Expand all Loading... |
| 37 }; | 38 }; |
| 38 | 39 |
| 39 class FakeDevToolsClient : public StubDevToolsClient { | 40 class FakeDevToolsClient : public StubDevToolsClient { |
| 40 public: | 41 public: |
| 41 explicit FakeDevToolsClient(const std::string& id) | 42 explicit FakeDevToolsClient(const std::string& id) |
| 42 : id_(id), listener_(nullptr), command_index_(0) {} | 43 : id_(id), listener_(nullptr), command_index_(0) {} |
| 43 ~FakeDevToolsClient() override {} | 44 ~FakeDevToolsClient() override {} |
| 44 | 45 |
| 45 bool PopSentCommand(DevToolsCommand** out_command) { | 46 bool PopSentCommand(DevToolsCommand** out_command) { |
| 46 if (sent_commands_.size() > command_index_) { | 47 if (sent_commands_.size() > command_index_) { |
| 47 *out_command = sent_commands_.get().at(command_index_++); | 48 *out_command = sent_commands_[command_index_++].get(); |
| 48 return true; | 49 return true; |
| 49 } | 50 } |
| 50 return false; | 51 return false; |
| 51 } | 52 } |
| 52 | 53 |
| 53 Status TriggerEvent(const std::string& method) { | 54 Status TriggerEvent(const std::string& method) { |
| 54 base::DictionaryValue empty_params; | 55 base::DictionaryValue empty_params; |
| 55 return listener_->OnEvent(this, method, empty_params); | 56 return listener_->OnEvent(this, method, empty_params); |
| 56 } | 57 } |
| 57 | 58 |
| 58 Status TriggerEvent(const std::string& method, | 59 Status TriggerEvent(const std::string& method, |
| 59 const base::DictionaryValue& params) { | 60 const base::DictionaryValue& params) { |
| 60 return listener_->OnEvent(this, method, params); | 61 return listener_->OnEvent(this, method, params); |
| 61 } | 62 } |
| 62 | 63 |
| 63 // Overridden from DevToolsClient: | 64 // Overridden from DevToolsClient: |
| 64 Status ConnectIfNecessary() override { return listener_->OnConnected(this); } | 65 Status ConnectIfNecessary() override { return listener_->OnConnected(this); } |
| 65 | 66 |
| 66 Status SendCommandAndGetResult( | 67 Status SendCommandAndGetResult( |
| 67 const std::string& method, | 68 const std::string& method, |
| 68 const base::DictionaryValue& params, | 69 const base::DictionaryValue& params, |
| 69 std::unique_ptr<base::DictionaryValue>* result) override { | 70 std::unique_ptr<base::DictionaryValue>* result) override { |
| 70 sent_commands_.push_back(new DevToolsCommand(method, | 71 sent_commands_.push_back( |
| 71 params.DeepCopy())); | 72 base::MakeUnique<DevToolsCommand>(method, params.DeepCopy())); |
| 72 return Status(kOk); | 73 return Status(kOk); |
| 73 } | 74 } |
| 74 | 75 |
| 75 void AddListener(DevToolsEventListener* listener) override { | 76 void AddListener(DevToolsEventListener* listener) override { |
| 76 CHECK(!listener_); | 77 CHECK(!listener_); |
| 77 listener_ = listener; | 78 listener_ = listener; |
| 78 } | 79 } |
| 79 | 80 |
| 80 const std::string& GetId() override { return id_; } | 81 const std::string& GetId() override { return id_; } |
| 81 | 82 |
| 82 private: | 83 private: |
| 83 const std::string id_; // WebView id. | 84 const std::string id_; // WebView id. |
| 84 ScopedVector<DevToolsCommand> sent_commands_; // Commands that were sent. | 85 std::vector<std::unique_ptr<DevToolsCommand>> |
| 86 sent_commands_; // Commands that were sent. |
| 85 DevToolsEventListener* listener_; // The fake allows only one event listener. | 87 DevToolsEventListener* listener_; // The fake allows only one event listener. |
| 86 size_t command_index_; | 88 size_t command_index_; |
| 87 }; | 89 }; |
| 88 | 90 |
| 89 struct LogEntry { | 91 struct LogEntry { |
| 90 const base::Time timestamp; | 92 const base::Time timestamp; |
| 91 const Log::Level level; | 93 const Log::Level level; |
| 92 const std::string source; | 94 const std::string source; |
| 93 const std::string message; | 95 const std::string message; |
| 94 | 96 |
| 95 LogEntry(const base::Time& timestamp, | 97 LogEntry(const base::Time& timestamp, |
| 96 Log::Level level, | 98 Log::Level level, |
| 97 const std::string& source, | 99 const std::string& source, |
| 98 const std::string& message) | 100 const std::string& message) |
| 99 : timestamp(timestamp), level(level), source(source), message(message) {} | 101 : timestamp(timestamp), level(level), source(source), message(message) {} |
| 100 }; | 102 }; |
| 101 | 103 |
| 102 class FakeLog : public Log { | 104 class FakeLog : public Log { |
| 103 public: | 105 public: |
| 104 void AddEntryTimestamped(const base::Time& timestamp, | 106 void AddEntryTimestamped(const base::Time& timestamp, |
| 105 Level level, | 107 Level level, |
| 106 const std::string& source, | 108 const std::string& source, |
| 107 const std::string& message) override; | 109 const std::string& message) override; |
| 108 | 110 |
| 109 bool Emptied() const override; | 111 bool Emptied() const override; |
| 110 | 112 |
| 111 const ScopedVector<LogEntry>& GetEntries() { | 113 const std::vector<std::unique_ptr<LogEntry>>& GetEntries() { |
| 112 return entries_; | 114 return entries_; |
| 113 } | 115 } |
| 114 | 116 |
| 115 private: | 117 private: |
| 116 ScopedVector<LogEntry> entries_; | 118 std::vector<std::unique_ptr<LogEntry>> entries_; |
| 117 }; | 119 }; |
| 118 | 120 |
| 119 void FakeLog::AddEntryTimestamped(const base::Time& timestamp, | 121 void FakeLog::AddEntryTimestamped(const base::Time& timestamp, |
| 120 Level level, | 122 Level level, |
| 121 const std::string& source, | 123 const std::string& source, |
| 122 const std::string& message) { | 124 const std::string& message) { |
| 123 entries_.push_back(new LogEntry(timestamp, level, source, message)); | 125 entries_.push_back( |
| 126 base::MakeUnique<LogEntry>(timestamp, level, source, message)); |
| 124 } | 127 } |
| 125 | 128 |
| 126 bool FakeLog::Emptied() const { | 129 bool FakeLog::Emptied() const { |
| 127 return true; | 130 return true; |
| 128 } | 131 } |
| 129 | 132 |
| 130 std::unique_ptr<base::DictionaryValue> ParseDictionary( | 133 std::unique_ptr<base::DictionaryValue> ParseDictionary( |
| 131 const std::string& json) { | 134 const std::string& json) { |
| 132 std::string error; | 135 std::string error; |
| 133 std::unique_ptr<base::Value> value = base::JSONReader::ReadAndReturnError( | 136 std::unique_ptr<base::Value> value = base::JSONReader::ReadAndReturnError( |
| (...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 196 | 199 |
| 197 client.AddListener(&logger); | 200 client.AddListener(&logger); |
| 198 logger.OnConnected(&client); | 201 logger.OnConnected(&client); |
| 199 ExpectEnableDomains(&client); | 202 ExpectEnableDomains(&client); |
| 200 ASSERT_EQ(kOk, client.TriggerEvent("Network.gaga").code()); | 203 ASSERT_EQ(kOk, client.TriggerEvent("Network.gaga").code()); |
| 201 ASSERT_EQ(kOk, client.TriggerEvent("Page.ulala").code()); | 204 ASSERT_EQ(kOk, client.TriggerEvent("Page.ulala").code()); |
| 202 // Ignore -- different domain. | 205 // Ignore -- different domain. |
| 203 ASSERT_EQ(kOk, client.TriggerEvent("Console.bad").code()); | 206 ASSERT_EQ(kOk, client.TriggerEvent("Console.bad").code()); |
| 204 | 207 |
| 205 ASSERT_EQ(2u, log.GetEntries().size()); | 208 ASSERT_EQ(2u, log.GetEntries().size()); |
| 206 ValidateLogEntry(log.GetEntries()[0], "webview-1", "Network.gaga"); | 209 ValidateLogEntry(log.GetEntries()[0].get(), "webview-1", "Network.gaga"); |
| 207 ValidateLogEntry(log.GetEntries()[1], "webview-1", "Page.ulala"); | 210 ValidateLogEntry(log.GetEntries()[1].get(), "webview-1", "Page.ulala"); |
| 208 } | 211 } |
| 209 | 212 |
| 210 TEST(PerformanceLogger, TwoWebViews) { | 213 TEST(PerformanceLogger, TwoWebViews) { |
| 211 FakeDevToolsClient client1("webview-1"); | 214 FakeDevToolsClient client1("webview-1"); |
| 212 FakeDevToolsClient client2("webview-2"); | 215 FakeDevToolsClient client2("webview-2"); |
| 213 FakeLog log; | 216 FakeLog log; |
| 214 Session session("test"); | 217 Session session("test"); |
| 215 PerformanceLogger logger(&log, &session); | 218 PerformanceLogger logger(&log, &session); |
| 216 | 219 |
| 217 client1.AddListener(&logger); | 220 client1.AddListener(&logger); |
| 218 client2.AddListener(&logger); | 221 client2.AddListener(&logger); |
| 219 logger.OnConnected(&client1); | 222 logger.OnConnected(&client1); |
| 220 logger.OnConnected(&client2); | 223 logger.OnConnected(&client2); |
| 221 ExpectEnableDomains(&client1); | 224 ExpectEnableDomains(&client1); |
| 222 ExpectEnableDomains(&client2); | 225 ExpectEnableDomains(&client2); |
| 223 // OnConnected sends the enable command only to that client, not others. | 226 // OnConnected sends the enable command only to that client, not others. |
| 224 client1.ConnectIfNecessary(); | 227 client1.ConnectIfNecessary(); |
| 225 ExpectEnableDomains(&client1); | 228 ExpectEnableDomains(&client1); |
| 226 DevToolsCommand* cmd; | 229 DevToolsCommand* cmd; |
| 227 ASSERT_FALSE(client2.PopSentCommand(&cmd)); | 230 ASSERT_FALSE(client2.PopSentCommand(&cmd)); |
| 228 | 231 |
| 229 ASSERT_EQ(kOk, client1.TriggerEvent("Page.gaga1").code()); | 232 ASSERT_EQ(kOk, client1.TriggerEvent("Page.gaga1").code()); |
| 230 ASSERT_EQ(kOk, client2.TriggerEvent("Network.gaga2").code()); | 233 ASSERT_EQ(kOk, client2.TriggerEvent("Network.gaga2").code()); |
| 231 | 234 |
| 232 ASSERT_EQ(2u, log.GetEntries().size()); | 235 ASSERT_EQ(2u, log.GetEntries().size()); |
| 233 ValidateLogEntry(log.GetEntries()[0], "webview-1", "Page.gaga1"); | 236 ValidateLogEntry(log.GetEntries()[0].get(), "webview-1", "Page.gaga1"); |
| 234 ValidateLogEntry(log.GetEntries()[1], "webview-2", "Network.gaga2"); | 237 ValidateLogEntry(log.GetEntries()[1].get(), "webview-2", "Network.gaga2"); |
| 235 } | 238 } |
| 236 | 239 |
| 237 TEST(PerformanceLogger, PerfLoggingPrefs) { | 240 TEST(PerformanceLogger, PerfLoggingPrefs) { |
| 238 FakeDevToolsClient client("webview-1"); | 241 FakeDevToolsClient client("webview-1"); |
| 239 FakeLog log; | 242 FakeLog log; |
| 240 Session session("test"); | 243 Session session("test"); |
| 241 PerfLoggingPrefs prefs; | 244 PerfLoggingPrefs prefs; |
| 242 EXPECT_EQ(PerfLoggingPrefs::InspectorDomainStatus::kDefaultEnabled, | 245 EXPECT_EQ(PerfLoggingPrefs::InspectorDomainStatus::kDefaultEnabled, |
| 243 prefs.network); | 246 prefs.network); |
| 244 prefs.network = PerfLoggingPrefs::InspectorDomainStatus::kExplicitlyDisabled; | 247 prefs.network = PerfLoggingPrefs::InspectorDomainStatus::kExplicitlyDisabled; |
| (...skipping 88 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 333 std::unique_ptr<base::DictionaryValue> event1(new base::DictionaryValue()); | 336 std::unique_ptr<base::DictionaryValue> event1(new base::DictionaryValue()); |
| 334 event1->SetString("cat", "foo"); | 337 event1->SetString("cat", "foo"); |
| 335 trace_events->Append(event1->CreateDeepCopy()); | 338 trace_events->Append(event1->CreateDeepCopy()); |
| 336 std::unique_ptr<base::DictionaryValue> event2(new base::DictionaryValue()); | 339 std::unique_ptr<base::DictionaryValue> event2(new base::DictionaryValue()); |
| 337 event2->SetString("cat", "bar"); | 340 event2->SetString("cat", "bar"); |
| 338 trace_events->Append(event2->CreateDeepCopy()); | 341 trace_events->Append(event2->CreateDeepCopy()); |
| 339 params.Set("value", trace_events); | 342 params.Set("value", trace_events); |
| 340 ASSERT_EQ(kOk, client.TriggerEvent("Tracing.dataCollected", params).code()); | 343 ASSERT_EQ(kOk, client.TriggerEvent("Tracing.dataCollected", params).code()); |
| 341 | 344 |
| 342 ASSERT_EQ(2u, log.GetEntries().size()); | 345 ASSERT_EQ(2u, log.GetEntries().size()); |
| 343 ValidateLogEntry(log.GetEntries()[0], | 346 ValidateLogEntry(log.GetEntries()[0].get(), |
| 344 DevToolsClientImpl::kBrowserwideDevToolsClientId, | 347 DevToolsClientImpl::kBrowserwideDevToolsClientId, |
| 345 "Tracing.dataCollected", *event1); | 348 "Tracing.dataCollected", *event1); |
| 346 ValidateLogEntry(log.GetEntries()[1], | 349 ValidateLogEntry(log.GetEntries()[1].get(), |
| 347 DevToolsClientImpl::kBrowserwideDevToolsClientId, | 350 DevToolsClientImpl::kBrowserwideDevToolsClientId, |
| 348 "Tracing.dataCollected", *event2); | 351 "Tracing.dataCollected", *event2); |
| 349 } | 352 } |
| 350 | 353 |
| 351 TEST(PerformanceLogger, ShouldRequestTraceEvents) { | 354 TEST(PerformanceLogger, ShouldRequestTraceEvents) { |
| 352 FakeBrowserwideClient client; | 355 FakeBrowserwideClient client; |
| 353 FakeLog log; | 356 FakeLog log; |
| 354 Session session("test"); | 357 Session session("test"); |
| 355 PerfLoggingPrefs prefs; | 358 PerfLoggingPrefs prefs; |
| 356 prefs.trace_categories = "benchmark,blink.console"; | 359 prefs.trace_categories = "benchmark,blink.console"; |
| (...skipping 20 matching lines...) Expand all Loading... |
| 377 prefs.trace_categories = "benchmark,blink.console"; | 380 prefs.trace_categories = "benchmark,blink.console"; |
| 378 PerformanceLogger logger(&log, &session, prefs); | 381 PerformanceLogger logger(&log, &session, prefs); |
| 379 | 382 |
| 380 client.AddListener(&logger); | 383 client.AddListener(&logger); |
| 381 logger.OnConnected(&client); | 384 logger.OnConnected(&client); |
| 382 base::DictionaryValue params; | 385 base::DictionaryValue params; |
| 383 params.SetDouble("percentFull", 1.0); | 386 params.SetDouble("percentFull", 1.0); |
| 384 ASSERT_EQ(kOk, client.TriggerEvent("Tracing.bufferUsage", params).code()); | 387 ASSERT_EQ(kOk, client.TriggerEvent("Tracing.bufferUsage", params).code()); |
| 385 | 388 |
| 386 ASSERT_EQ(1u, log.GetEntries().size()); | 389 ASSERT_EQ(1u, log.GetEntries().size()); |
| 387 LogEntry* entry = log.GetEntries()[0]; | 390 LogEntry* entry = log.GetEntries()[0].get(); |
| 388 EXPECT_EQ(Log::kWarning, entry->level); | 391 EXPECT_EQ(Log::kWarning, entry->level); |
| 389 EXPECT_LT(0, entry->timestamp.ToTimeT()); | 392 EXPECT_LT(0, entry->timestamp.ToTimeT()); |
| 390 std::unique_ptr<base::DictionaryValue> message( | 393 std::unique_ptr<base::DictionaryValue> message( |
| 391 ParseDictionary(entry->message)); | 394 ParseDictionary(entry->message)); |
| 392 std::string webview; | 395 std::string webview; |
| 393 EXPECT_TRUE(message->GetString("webview", &webview)); | 396 EXPECT_TRUE(message->GetString("webview", &webview)); |
| 394 EXPECT_EQ(DevToolsClientImpl::kBrowserwideDevToolsClientId, webview); | 397 EXPECT_EQ(DevToolsClientImpl::kBrowserwideDevToolsClientId, webview); |
| 395 std::string method; | 398 std::string method; |
| 396 EXPECT_TRUE(message->GetString("message.method", &method)); | 399 EXPECT_TRUE(message->GetString("message.method", &method)); |
| 397 EXPECT_EQ("Tracing.bufferUsage", method); | 400 EXPECT_EQ("Tracing.bufferUsage", method); |
| 398 base::DictionaryValue* actual_params; | 401 base::DictionaryValue* actual_params; |
| 399 EXPECT_TRUE(message->GetDictionary("message.params", &actual_params)); | 402 EXPECT_TRUE(message->GetDictionary("message.params", &actual_params)); |
| 400 EXPECT_TRUE(actual_params->HasKey("error")); | 403 EXPECT_TRUE(actual_params->HasKey("error")); |
| 401 } | 404 } |
| OLD | NEW |