[Tracing] V8 Tracing Controller

test/cctest/libplatform/test-tracing.cc

+// Copyright 2016 the V8 project 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 "include/libplatform/v8-tracing.h"
+#include "test/cctest/cctest.h"
+
+namespace v8 {
+namespace platform {
+namespace tracing {
+
+const char* test_str =
+    "{\"record_mode\":\"record-until-full\",\"enable_sampling\":1,\"enable_"
+    "systrace\":0,\"enable_argument_filter\":1,\"included_categories\":[\"v8."
+    "cpu_profile\",\"v8.cpu_profile.hires\"],\"excluded_categories\":[\"v8."
+    "runtime\"]}";
+
+TEST(TraceConfigConstructor) {
+  LocalContext env;
+  TraceConfig* trace_config =
+      TraceConfig::CreateTraceConfigFromJSON(env->GetIsolate(), test_str);
+  CHECK_EQ(trace_config->IsSamplingEnabled(), true);
+  CHECK_EQ(trace_config->IsSystraceEnabled(), false);
+  CHECK_EQ(trace_config->IsArgumentFilterEnabled(), true);
+  CHECK_EQ(trace_config->IsCategoryGroupEnabled("v8"), false);
+  CHECK_EQ(trace_config->IsCategoryGroupEnabled("v8.cpu_profile"), true);
+  CHECK_EQ(trace_config->IsCategoryGroupEnabled("v8.cpu_profile.hires"), true);
+  CHECK_EQ(trace_config->IsCategoryGroupEnabled("v8.runtime"), false);
+  delete trace_config;
+}
+
+TEST(TestTraceObject) {
+  TraceObject trace_object;
+  trace_object.Initialize('X', "Test.Trace", "v8-cat", 42, 123, 0, 0);
+  CHECK_EQ('X', trace_object.phase());
+  CHECK_EQ("Test.Trace", trace_object.name());
+  CHECK_EQ("v8-cat", trace_object.category_group());
+  CHECK_EQ(0, trace_object.duration());
+  CHECK_EQ(0, trace_object.cpu_duration());
+}
+
+class MockTraceWriter : public TraceWriter {
+ public:
+  void AppendTraceEvent(TraceObject* trace_event) override {
+    events_.push_back(trace_event->name());
+  }
+
+  void Flush() override {}
+
+  std::vector<std::string> events() { return events_; }
+
+ private:
+  std::vector<std::string> events_;
+};
+
+TEST(TestTraceBufferRingBuffer) {
+  MockTraceWriter* writer = new MockTraceWriter();
+  TraceBuffer* ring_buffer =
+      TraceBuffer::CreateTraceBufferRingBuffer(2, writer);
+
+  // We should be able to add kChunkSize * 2 + 1 trace events.
+  std::vector<uint64_t> handles(TraceBufferChunk::kChunkSize * 2 + 1);
+  for (size_t i = 0; i < handles.size(); ++i) {
+    TraceObject* trace_object = ring_buffer->AddTraceEvent(&handles[i]);
+    CHECK_NOT_NULL(trace_object);
+    std::string name = "Test.Trace" + std::to_string(i);
+    trace_object->Initialize('X', name, "v8-cat", 42, 123, 0, 0);
+    trace_object = ring_buffer->GetEventByHandle(handles[i]);
+    CHECK_NOT_NULL(trace_object);
+    CHECK_EQ('X', trace_object->phase());
+    CHECK_EQ(name, trace_object->name());
+    CHECK_EQ("v8-cat", trace_object->category_group());
+  }
+
+  // We should only be able to retrieve the last kChunkSize + 1.
+  for (size_t i = 0; i < TraceBufferChunk::kChunkSize; ++i) {
+    CHECK_NULL(ring_buffer->GetEventByHandle(handles[i]));
+  }
+
+  for (size_t i = TraceBufferChunk::kChunkSize; i < handles.size(); ++i) {
+    TraceObject* trace_object = ring_buffer->GetEventByHandle(handles[i]);
+    CHECK_NOT_NULL(trace_object);
+    // The object properties should be correct.
+    CHECK_EQ('X', trace_object->phase());
+    CHECK_EQ("Test.Trace" + std::to_string(i), trace_object->name());
+    CHECK_EQ("v8-cat", trace_object->category_group());
+  }
+
+  // Check Flush(), that the writer wrote the last kChunkSize  1 event names.
+  ring_buffer->Flush();
+  auto events = writer->events();
+  CHECK_EQ(TraceBufferChunk::kChunkSize + 1, events.size());
+  for (size_t i = TraceBufferChunk::kChunkSize; i < handles.size(); ++i) {
+    CHECK_EQ("Test.Trace" + std::to_string(i),
+             events[i - TraceBufferChunk::kChunkSize]);
+  }
+
+  delete ring_buffer;
+}
+
+TEST(TestJSONTraceWriter) {
+  std::ostringstream stream;
+  TraceWriter* writer = TraceWriter::CreateJSONTraceWriter(stream);
+  TraceObject trace_object;
+  trace_object.InitializeForTesting('X', "Test0", "v8-cat", 42, 123, 0, 0, 11,
+                                    22, 100, 50, 33, 44);
+  writer->AppendTraceEvent(&trace_object);
+  trace_object.InitializeForTesting('Y', "Test1", "v8-cat", 43, 456, 0, 0, 55,
+                                    66, 110, 55, 77, 88);
+  writer->AppendTraceEvent(&trace_object);
+  writer->Flush();
+  delete writer;
+
+  std::string trace_str = stream.str();
+  std::string expected_trace_str =
+      "{\"traceEvents\":[{\"pid\":11,\"tid\":22,\"ts\":100,\"tts\":50,"
+      "\"ph\":\"X\",\"cat\":\"v8-cat\",\"name\":\"Test0\",\"args\":{},"
+      "\"dur\":33,\"tdur\":44},{\"pid\":55,\"tid\":66,\"ts\":110,\"tts\":55,"
+      "\"ph\":\"Y\",\"cat\":\"v8-cat\",\"name\":\"Test1\",\"args\":{},\"dur\":"
+      "77,\"tdur\":88}]}";
+  CHECK_EQ(expected_trace_str, trace_str);
+}
+
+}  // namespace tracing
+}  // namespace platform
+}  // namespace v8