base/trace_event/trace_event_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <math.h>
   6 #include <cstdlib>
   7
   8 #include "base/bind.h"
   9 #include "base/command_line.h"
  10 #include "base/json/json_reader.h"
  11 #include "base/json/json_writer.h"
  12 #include "base/location.h"
  13 #include "base/memory/ref_counted_memory.h"
  14 #include "base/memory/scoped_ptr.h"
  15 #include "base/memory/singleton.h"
  16 #include "base/process/process_handle.h"
  17 #include "base/single_thread_task_runner.h"
  18 #include "base/strings/stringprintf.h"
  19 #include "base/synchronization/waitable_event.h"
  20 #include "base/threading/platform_thread.h"
  21 #include "base/threading/thread.h"
  22 #include "base/time/time.h"
  23 #include "base/trace_event/trace_event.h"
  24 #include "base/trace_event/trace_event_synthetic_delay.h"
  25 #include "base/values.h"
  26 #include "testing/gmock/include/gmock/gmock.h"
  27 #include "testing/gtest/include/gtest/gtest.h"
  28
  29 namespace base {
  30 namespace trace_event {
  31
  32 namespace {
  33
  34 enum CompareOp {
  35   IS_EQUAL,
  36   IS_NOT_EQUAL,
  37 };
  38
  39 struct JsonKeyValue {
  40   const char* key;
  41   const char* value;
  42   CompareOp op;
  43 };
  44
  45 const int kThreadId = 42;
  46 const int kAsyncId = 5;
  47 const char kAsyncIdStr[] = "0x5";
  48 const int kAsyncId2 = 6;
  49 const char kAsyncId2Str[] = "0x6";
  50
  51 const  char kRecordAllCategoryFilter[] = "*";
  52
  53 class TraceEventTestFixture : public testing::Test {
  54  public:
  55   void OnTraceDataCollected(
  56       WaitableEvent* flush_complete_event,
  57       const scoped_refptr<base::RefCountedString>& events_str,
  58       bool has_more_events);
  59   void OnWatchEventMatched() {
  60     ++event_watch_notification_;
  61   }
  62   DictionaryValue* FindMatchingTraceEntry(const JsonKeyValue* key_values);
  63   DictionaryValue* FindNamePhase(const char* name, const char* phase);
  64   DictionaryValue* FindNamePhaseKeyValue(const char* name,
  65                                          const char* phase,
  66                                          const char* key,
  67                                          const char* value);
  68   void DropTracedMetadataRecords();
  69   bool FindMatchingValue(const char* key,
  70                          const char* value);
  71   bool FindNonMatchingValue(const char* key,
  72                             const char* value);
  73   void Clear() {
  74     trace_parsed_.Clear();
  75     json_output_.json_output.clear();
  76   }
  77
  78   void BeginTrace() {
  79     BeginSpecificTrace("*");
  80   }
  81
  82   void BeginSpecificTrace(const std::string& filter) {
  83     event_watch_notification_ = 0;
  84     TraceLog::GetInstance()->SetEnabled(TraceConfig(filter, ""),
  85                                         TraceLog::RECORDING_MODE);
  86   }
  87
  88   void EndTraceAndFlush() {
  89     WaitableEvent flush_complete_event(false, false);
  90     EndTraceAndFlushAsync(&flush_complete_event);
  91     flush_complete_event.Wait();
  92   }
  93
  94   // Used when testing thread-local buffers which requires the thread initiating
  95   // flush to have a message loop.
  96   void EndTraceAndFlushInThreadWithMessageLoop() {
  97     WaitableEvent flush_complete_event(false, false);
  98     Thread flush_thread("flush");
  99     flush_thread.Start();
 100     flush_thread.task_runner()->PostTask(
 101         FROM_HERE, base::Bind(&TraceEventTestFixture::EndTraceAndFlushAsync,
 102                               base::Unretained(this), &flush_complete_event));
 103     flush_complete_event.Wait();
 104   }
 105
 106   void EndTraceAndFlushAsync(WaitableEvent* flush_complete_event) {
 107     TraceLog::GetInstance()->SetDisabled();
 108     TraceLog::GetInstance()->Flush(
 109         base::Bind(&TraceEventTestFixture::OnTraceDataCollected,
 110                    base::Unretained(static_cast<TraceEventTestFixture*>(this)),
 111                    base::Unretained(flush_complete_event)));
 112   }
 113
 114   void FlushMonitoring() {
 115     WaitableEvent flush_complete_event(false, false);
 116     FlushMonitoring(&flush_complete_event);
 117     flush_complete_event.Wait();
 118   }
 119
 120   void FlushMonitoring(WaitableEvent* flush_complete_event) {
 121     TraceLog::GetInstance()->FlushButLeaveBufferIntact(
 122         base::Bind(&TraceEventTestFixture::OnTraceDataCollected,
 123                    base::Unretained(static_cast<TraceEventTestFixture*>(this)),
 124                    base::Unretained(flush_complete_event)));
 125   }
 126
 127   void SetUp() override {
 128     const char* name = PlatformThread::GetName();
 129     old_thread_name_ = name ? strdup(name) : NULL;
 130
 131     TraceLog::DeleteForTesting();
 132     TraceLog* tracelog = TraceLog::GetInstance();
 133     ASSERT_TRUE(tracelog);
 134     ASSERT_FALSE(tracelog->IsEnabled());
 135     trace_buffer_.SetOutputCallback(json_output_.GetCallback());
 136     event_watch_notification_ = 0;
 137   }
 138   void TearDown() override {
 139     if (TraceLog::GetInstance())
 140       EXPECT_FALSE(TraceLog::GetInstance()->IsEnabled());
 141     PlatformThread::SetName(old_thread_name_ ? old_thread_name_ : "");
 142     free(old_thread_name_);
 143     old_thread_name_ = NULL;
 144     // We want our singleton torn down after each test.
 145     TraceLog::DeleteForTesting();
 146   }
 147
 148   char* old_thread_name_;
 149   ListValue trace_parsed_;
 150   TraceResultBuffer trace_buffer_;
 151   TraceResultBuffer::SimpleOutput json_output_;
 152   int event_watch_notification_;
 153
 154  private:
 155   // We want our singleton torn down after each test.
 156   ShadowingAtExitManager at_exit_manager_;
 157   Lock lock_;
 158 };
 159
 160 void TraceEventTestFixture::OnTraceDataCollected(
 161     WaitableEvent* flush_complete_event,
 162     const scoped_refptr<base::RefCountedString>& events_str,
 163     bool has_more_events) {
 164   AutoLock lock(lock_);
 165   json_output_.json_output.clear();
 166   trace_buffer_.Start();
 167   trace_buffer_.AddFragment(events_str->data());
 168   trace_buffer_.Finish();
 169
 170   scoped_ptr<Value> root;
 171   root.reset(base::JSONReader::DeprecatedRead(
 172       json_output_.json_output, JSON_PARSE_RFC | JSON_DETACHABLE_CHILDREN));
 173
 174   if (!root.get()) {
 175     LOG(ERROR) << json_output_.json_output;
 176   }
 177
 178   ListValue* root_list = NULL;
 179   ASSERT_TRUE(root.get());
 180   ASSERT_TRUE(root->GetAsList(&root_list));
 181
 182   // Move items into our aggregate collection
 183   while (root_list->GetSize()) {
 184     scoped_ptr<Value> item;
 185     root_list->Remove(0, &item);
 186     trace_parsed_.Append(item.release());
 187   }
 188
 189   if (!has_more_events)
 190     flush_complete_event->Signal();
 191 }
 192
 193 static bool CompareJsonValues(const std::string& lhs,
 194                               const std::string& rhs,
 195                               CompareOp op) {
 196   switch (op) {
 197     case IS_EQUAL:
 198       return lhs == rhs;
 199     case IS_NOT_EQUAL:
 200       return lhs != rhs;
 201     default:
 202       CHECK(0);
 203   }
 204   return false;
 205 }
 206
 207 static bool IsKeyValueInDict(const JsonKeyValue* key_value,
 208                              DictionaryValue* dict) {
 209   Value* value = NULL;
 210   std::string value_str;
 211   if (dict->Get(key_value->key, &value) &&
 212       value->GetAsString(&value_str) &&
 213       CompareJsonValues(value_str, key_value->value, key_value->op))
 214     return true;
 215
 216   // Recurse to test arguments
 217   DictionaryValue* args_dict = NULL;
 218   dict->GetDictionary("args", &args_dict);
 219   if (args_dict)
 220     return IsKeyValueInDict(key_value, args_dict);
 221
 222   return false;
 223 }
 224
 225 static bool IsAllKeyValueInDict(const JsonKeyValue* key_values,
 226                                 DictionaryValue* dict) {
 227   // Scan all key_values, they must all be present and equal.
 228   while (key_values && key_values->key) {
 229     if (!IsKeyValueInDict(key_values, dict))
 230       return false;
 231     ++key_values;
 232   }
 233   return true;
 234 }
 235
 236 DictionaryValue* TraceEventTestFixture::FindMatchingTraceEntry(
 237     const JsonKeyValue* key_values) {
 238   // Scan all items
 239   size_t trace_parsed_count = trace_parsed_.GetSize();
 240   for (size_t i = 0; i < trace_parsed_count; i++) {
 241     Value* value = NULL;
 242     trace_parsed_.Get(i, &value);
 243     if (!value || value->GetType() != Value::TYPE_DICTIONARY)
 244       continue;
 245     DictionaryValue* dict = static_cast<DictionaryValue*>(value);
 246
 247     if (IsAllKeyValueInDict(key_values, dict))
 248       return dict;
 249   }
 250   return NULL;
 251 }
 252
 253 void TraceEventTestFixture::DropTracedMetadataRecords() {
 254   scoped_ptr<ListValue> old_trace_parsed(trace_parsed_.DeepCopy());
 255   size_t old_trace_parsed_size = old_trace_parsed->GetSize();
 256   trace_parsed_.Clear();
 257
 258   for (size_t i = 0; i < old_trace_parsed_size; i++) {
 259     Value* value = NULL;
 260     old_trace_parsed->Get(i, &value);
 261     if (!value || value->GetType() != Value::TYPE_DICTIONARY) {
 262       trace_parsed_.Append(value->DeepCopy());
 263       continue;
 264     }
 265     DictionaryValue* dict = static_cast<DictionaryValue*>(value);
 266     std::string tmp;
 267     if (dict->GetString("ph", &tmp) && tmp == "M")
 268       continue;
 269
 270     trace_parsed_.Append(value->DeepCopy());
 271   }
 272 }
 273
 274 DictionaryValue* TraceEventTestFixture::FindNamePhase(const char* name,
 275                                                       const char* phase) {
 276   JsonKeyValue key_values[] = {
 277     {"name", name, IS_EQUAL},
 278     {"ph", phase, IS_EQUAL},
 279     {0, 0, IS_EQUAL}
 280   };
 281   return FindMatchingTraceEntry(key_values);
 282 }
 283
 284 DictionaryValue* TraceEventTestFixture::FindNamePhaseKeyValue(
 285     const char* name,
 286     const char* phase,
 287     const char* key,
 288     const char* value) {
 289   JsonKeyValue key_values[] = {
 290     {"name", name, IS_EQUAL},
 291     {"ph", phase, IS_EQUAL},
 292     {key, value, IS_EQUAL},
 293     {0, 0, IS_EQUAL}
 294   };
 295   return FindMatchingTraceEntry(key_values);
 296 }
 297
 298 bool TraceEventTestFixture::FindMatchingValue(const char* key,
 299                                               const char* value) {
 300   JsonKeyValue key_values[] = {
 301     {key, value, IS_EQUAL},
 302     {0, 0, IS_EQUAL}
 303   };
 304   return FindMatchingTraceEntry(key_values);
 305 }
 306
 307 bool TraceEventTestFixture::FindNonMatchingValue(const char* key,
 308                                                  const char* value) {
 309   JsonKeyValue key_values[] = {
 310     {key, value, IS_NOT_EQUAL},
 311     {0, 0, IS_EQUAL}
 312   };
 313   return FindMatchingTraceEntry(key_values);
 314 }
 315
 316 bool IsStringInDict(const char* string_to_match, const DictionaryValue* dict) {
 317   for (DictionaryValue::Iterator it(*dict); !it.IsAtEnd(); it.Advance()) {
 318     if (it.key().find(string_to_match) != std::string::npos)
 319       return true;
 320
 321     std::string value_str;
 322     it.value().GetAsString(&value_str);
 323     if (value_str.find(string_to_match) != std::string::npos)
 324       return true;
 325   }
 326
 327   // Recurse to test arguments
 328   const DictionaryValue* args_dict = NULL;
 329   dict->GetDictionary("args", &args_dict);
 330   if (args_dict)
 331     return IsStringInDict(string_to_match, args_dict);
 332
 333   return false;
 334 }
 335
 336 const DictionaryValue* FindTraceEntry(
 337     const ListValue& trace_parsed,
 338     const char* string_to_match,
 339     const DictionaryValue* match_after_this_item = NULL) {
 340   // Scan all items
 341   size_t trace_parsed_count = trace_parsed.GetSize();
 342   for (size_t i = 0; i < trace_parsed_count; i++) {
 343     const Value* value = NULL;
 344     trace_parsed.Get(i, &value);
 345     if (match_after_this_item) {
 346       if (value == match_after_this_item)
 347          match_after_this_item = NULL;
 348       continue;
 349     }
 350     if (!value || value->GetType() != Value::TYPE_DICTIONARY)
 351       continue;
 352     const DictionaryValue* dict = static_cast<const DictionaryValue*>(value);
 353
 354     if (IsStringInDict(string_to_match, dict))
 355       return dict;
 356   }
 357   return NULL;
 358 }
 359
 360 std::vector<const DictionaryValue*> FindTraceEntries(
 361     const ListValue& trace_parsed,
 362     const char* string_to_match) {
 363   std::vector<const DictionaryValue*> hits;
 364   size_t trace_parsed_count = trace_parsed.GetSize();
 365   for (size_t i = 0; i < trace_parsed_count; i++) {
 366     const Value* value = NULL;
 367     trace_parsed.Get(i, &value);
 368     if (!value || value->GetType() != Value::TYPE_DICTIONARY)
 369       continue;
 370     const DictionaryValue* dict = static_cast<const DictionaryValue*>(value);
 371
 372     if (IsStringInDict(string_to_match, dict))
 373       hits.push_back(dict);
 374   }
 375   return hits;
 376 }
 377
 378 const char kControlCharacters[] = "\001\002\003\n\r";
 379
 380 void TraceWithAllMacroVariants(WaitableEvent* task_complete_event) {
 381   {
 382     TRACE_EVENT_BEGIN_ETW("TRACE_EVENT_BEGIN_ETW call", 0x1122, "extrastring1");
 383     TRACE_EVENT_END_ETW("TRACE_EVENT_END_ETW call", 0x3344, "extrastring2");
 384     TRACE_EVENT_INSTANT_ETW("TRACE_EVENT_INSTANT_ETW call",
 385                             0x5566, "extrastring3");
 386
 387     TRACE_EVENT0("all", "TRACE_EVENT0 call");
 388     TRACE_EVENT1("all", "TRACE_EVENT1 call", "name1", "value1");
 389     TRACE_EVENT2("all", "TRACE_EVENT2 call",
 390                  "name1", "\"value1\"",
 391                  "name2", "value\\2");
 392
 393     TRACE_EVENT_INSTANT0("all", "TRACE_EVENT_INSTANT0 call",
 394                          TRACE_EVENT_SCOPE_GLOBAL);
 395     TRACE_EVENT_INSTANT1("all", "TRACE_EVENT_INSTANT1 call",
 396                          TRACE_EVENT_SCOPE_PROCESS, "name1", "value1");
 397     TRACE_EVENT_INSTANT2("all", "TRACE_EVENT_INSTANT2 call",
 398                          TRACE_EVENT_SCOPE_THREAD,
 399                          "name1", "value1",
 400                          "name2", "value2");
 401
 402     TRACE_EVENT_BEGIN0("all", "TRACE_EVENT_BEGIN0 call");
 403     TRACE_EVENT_BEGIN1("all", "TRACE_EVENT_BEGIN1 call", "name1", "value1");
 404     TRACE_EVENT_BEGIN2("all", "TRACE_EVENT_BEGIN2 call",
 405                        "name1", "value1",
 406                        "name2", "value2");
 407
 408     TRACE_EVENT_END0("all", "TRACE_EVENT_END0 call");
 409     TRACE_EVENT_END1("all", "TRACE_EVENT_END1 call", "name1", "value1");
 410     TRACE_EVENT_END2("all", "TRACE_EVENT_END2 call",
 411                      "name1", "value1",
 412                      "name2", "value2");
 413
 414     TRACE_EVENT_ASYNC_BEGIN0("all", "TRACE_EVENT_ASYNC_BEGIN0 call", kAsyncId);
 415     TRACE_EVENT_ASYNC_BEGIN1("all", "TRACE_EVENT_ASYNC_BEGIN1 call", kAsyncId,
 416                              "name1", "value1");
 417     TRACE_EVENT_ASYNC_BEGIN2("all", "TRACE_EVENT_ASYNC_BEGIN2 call", kAsyncId,
 418                              "name1", "value1",
 419                              "name2", "value2");
 420
 421     TRACE_EVENT_ASYNC_STEP_INTO0("all", "TRACE_EVENT_ASYNC_STEP_INTO0 call",
 422                                  kAsyncId, "step_begin1");
 423     TRACE_EVENT_ASYNC_STEP_INTO1("all", "TRACE_EVENT_ASYNC_STEP_INTO1 call",
 424                                  kAsyncId, "step_begin2", "name1", "value1");
 425
 426     TRACE_EVENT_ASYNC_END0("all", "TRACE_EVENT_ASYNC_END0 call", kAsyncId);
 427     TRACE_EVENT_ASYNC_END1("all", "TRACE_EVENT_ASYNC_END1 call", kAsyncId,
 428                            "name1", "value1");
 429     TRACE_EVENT_ASYNC_END2("all", "TRACE_EVENT_ASYNC_END2 call", kAsyncId,
 430                            "name1", "value1",
 431                            "name2", "value2");
 432
 433     TRACE_EVENT_BEGIN_ETW("TRACE_EVENT_BEGIN_ETW0 call", kAsyncId, NULL);
 434     TRACE_EVENT_BEGIN_ETW("TRACE_EVENT_BEGIN_ETW1 call", kAsyncId, "value");
 435     TRACE_EVENT_END_ETW("TRACE_EVENT_END_ETW0 call", kAsyncId, NULL);
 436     TRACE_EVENT_END_ETW("TRACE_EVENT_END_ETW1 call", kAsyncId, "value");
 437     TRACE_EVENT_INSTANT_ETW("TRACE_EVENT_INSTANT_ETW0 call", kAsyncId, NULL);
 438     TRACE_EVENT_INSTANT_ETW("TRACE_EVENT_INSTANT_ETW1 call", kAsyncId, "value");
 439
 440     TRACE_COUNTER1("all", "TRACE_COUNTER1 call", 31415);
 441     TRACE_COUNTER2("all", "TRACE_COUNTER2 call",
 442                    "a", 30000,
 443                    "b", 1415);
 444
 445     TRACE_COUNTER_ID1("all", "TRACE_COUNTER_ID1 call", 0x319009, 31415);
 446     TRACE_COUNTER_ID2("all", "TRACE_COUNTER_ID2 call", 0x319009,
 447                       "a", 30000, "b", 1415);
 448
 449     TRACE_EVENT_COPY_BEGIN_WITH_ID_TID_AND_TIMESTAMP0("all",
 450         "TRACE_EVENT_COPY_BEGIN_WITH_ID_TID_AND_TIMESTAMP0 call",
 451         kAsyncId, kThreadId, 12345);
 452     TRACE_EVENT_COPY_END_WITH_ID_TID_AND_TIMESTAMP0("all",
 453         "TRACE_EVENT_COPY_END_WITH_ID_TID_AND_TIMESTAMP0 call",
 454         kAsyncId, kThreadId, 23456);
 455
 456     TRACE_EVENT_BEGIN_WITH_ID_TID_AND_TIMESTAMP0("all",
 457         "TRACE_EVENT_BEGIN_WITH_ID_TID_AND_TIMESTAMP0 call",
 458         kAsyncId2, kThreadId, 34567);
 459     TRACE_EVENT_ASYNC_STEP_PAST0("all", "TRACE_EVENT_ASYNC_STEP_PAST0 call",
 460                                  kAsyncId2, "step_end1");
 461     TRACE_EVENT_ASYNC_STEP_PAST1("all", "TRACE_EVENT_ASYNC_STEP_PAST1 call",
 462                                  kAsyncId2, "step_end2", "name1", "value1");
 463
 464     TRACE_EVENT_END_WITH_ID_TID_AND_TIMESTAMP0("all",
 465         "TRACE_EVENT_END_WITH_ID_TID_AND_TIMESTAMP0 call",
 466         kAsyncId2, kThreadId, 45678);
 467
 468     TRACE_EVENT_OBJECT_CREATED_WITH_ID("all", "tracked object 1", 0x42);
 469     TRACE_EVENT_OBJECT_SNAPSHOT_WITH_ID(
 470         "all", "tracked object 1", 0x42, "hello");
 471     TRACE_EVENT_OBJECT_DELETED_WITH_ID("all", "tracked object 1", 0x42);
 472
 473     TraceScopedTrackableObject<int> trackable("all", "tracked object 2",
 474                                               0x2128506);
 475     trackable.snapshot("world");
 476
 477     TRACE_EVENT1(kControlCharacters, kControlCharacters,
 478                  kControlCharacters, kControlCharacters);
 479   }  // Scope close causes TRACE_EVENT0 etc to send their END events.
 480
 481   if (task_complete_event)
 482     task_complete_event->Signal();
 483 }
 484
 485 void ValidateAllTraceMacrosCreatedData(const ListValue& trace_parsed) {
 486   const DictionaryValue* item = NULL;
 487
 488 #define EXPECT_FIND_(string) \
 489     item = FindTraceEntry(trace_parsed, string); \
 490     EXPECT_TRUE(item);
 491 #define EXPECT_NOT_FIND_(string) \
 492     item = FindTraceEntry(trace_parsed, string); \
 493     EXPECT_FALSE(item);
 494 #define EXPECT_SUB_FIND_(string) \
 495     if (item) \
 496       EXPECT_TRUE(IsStringInDict(string, item));
 497
 498   EXPECT_FIND_("ETW Trace Event");
 499   EXPECT_FIND_("all");
 500   EXPECT_FIND_("TRACE_EVENT_BEGIN_ETW call");
 501   {
 502     std::string str_val;
 503     EXPECT_TRUE(item && item->GetString("args.id", &str_val));
 504     EXPECT_STREQ("0x1122", str_val.c_str());
 505   }
 506   EXPECT_SUB_FIND_("extrastring1");
 507   EXPECT_FIND_("TRACE_EVENT_END_ETW call");
 508   EXPECT_FIND_("TRACE_EVENT_INSTANT_ETW call");
 509   EXPECT_FIND_("TRACE_EVENT0 call");
 510   {
 511     std::string ph;
 512     std::string ph_end;
 513     EXPECT_TRUE((item = FindTraceEntry(trace_parsed, "TRACE_EVENT0 call")));
 514     EXPECT_TRUE((item && item->GetString("ph", &ph)));
 515     EXPECT_EQ("X", ph);
 516     item = FindTraceEntry(trace_parsed, "TRACE_EVENT0 call", item);
 517     EXPECT_FALSE(item);
 518   }
 519   EXPECT_FIND_("TRACE_EVENT1 call");
 520   EXPECT_SUB_FIND_("name1");
 521   EXPECT_SUB_FIND_("value1");
 522   EXPECT_FIND_("TRACE_EVENT2 call");
 523   EXPECT_SUB_FIND_("name1");
 524   EXPECT_SUB_FIND_("\"value1\"");
 525   EXPECT_SUB_FIND_("name2");
 526   EXPECT_SUB_FIND_("value\\2");
 527
 528   EXPECT_FIND_("TRACE_EVENT_INSTANT0 call");
 529   {
 530     std::string scope;
 531     EXPECT_TRUE((item && item->GetString("s", &scope)));
 532     EXPECT_EQ("g", scope);
 533   }
 534   EXPECT_FIND_("TRACE_EVENT_INSTANT1 call");
 535   {
 536     std::string scope;
 537     EXPECT_TRUE((item && item->GetString("s", &scope)));
 538     EXPECT_EQ("p", scope);
 539   }
 540   EXPECT_SUB_FIND_("name1");
 541   EXPECT_SUB_FIND_("value1");
 542   EXPECT_FIND_("TRACE_EVENT_INSTANT2 call");
 543   {
 544     std::string scope;
 545     EXPECT_TRUE((item && item->GetString("s", &scope)));
 546     EXPECT_EQ("t", scope);
 547   }
 548   EXPECT_SUB_FIND_("name1");
 549   EXPECT_SUB_FIND_("value1");
 550   EXPECT_SUB_FIND_("name2");
 551   EXPECT_SUB_FIND_("value2");
 552
 553   EXPECT_FIND_("TRACE_EVENT_BEGIN0 call");
 554   EXPECT_FIND_("TRACE_EVENT_BEGIN1 call");
 555   EXPECT_SUB_FIND_("name1");
 556   EXPECT_SUB_FIND_("value1");
 557   EXPECT_FIND_("TRACE_EVENT_BEGIN2 call");
 558   EXPECT_SUB_FIND_("name1");
 559   EXPECT_SUB_FIND_("value1");
 560   EXPECT_SUB_FIND_("name2");
 561   EXPECT_SUB_FIND_("value2");
 562
 563   EXPECT_FIND_("TRACE_EVENT_END0 call");
 564   EXPECT_FIND_("TRACE_EVENT_END1 call");
 565   EXPECT_SUB_FIND_("name1");
 566   EXPECT_SUB_FIND_("value1");
 567   EXPECT_FIND_("TRACE_EVENT_END2 call");
 568   EXPECT_SUB_FIND_("name1");
 569   EXPECT_SUB_FIND_("value1");
 570   EXPECT_SUB_FIND_("name2");
 571   EXPECT_SUB_FIND_("value2");
 572
 573   EXPECT_FIND_("TRACE_EVENT_ASYNC_BEGIN0 call");
 574   EXPECT_SUB_FIND_("id");
 575   EXPECT_SUB_FIND_(kAsyncIdStr);
 576   EXPECT_FIND_("TRACE_EVENT_ASYNC_BEGIN1 call");
 577   EXPECT_SUB_FIND_("id");
 578   EXPECT_SUB_FIND_(kAsyncIdStr);
 579   EXPECT_SUB_FIND_("name1");
 580   EXPECT_SUB_FIND_("value1");
 581   EXPECT_FIND_("TRACE_EVENT_ASYNC_BEGIN2 call");
 582   EXPECT_SUB_FIND_("id");
 583   EXPECT_SUB_FIND_(kAsyncIdStr);
 584   EXPECT_SUB_FIND_("name1");
 585   EXPECT_SUB_FIND_("value1");
 586   EXPECT_SUB_FIND_("name2");
 587   EXPECT_SUB_FIND_("value2");
 588
 589   EXPECT_FIND_("TRACE_EVENT_ASYNC_STEP_INTO0 call");
 590   EXPECT_SUB_FIND_("id");
 591   EXPECT_SUB_FIND_(kAsyncIdStr);
 592   EXPECT_SUB_FIND_("step_begin1");
 593   EXPECT_FIND_("TRACE_EVENT_ASYNC_STEP_INTO1 call");
 594   EXPECT_SUB_FIND_("id");
 595   EXPECT_SUB_FIND_(kAsyncIdStr);
 596   EXPECT_SUB_FIND_("step_begin2");
 597   EXPECT_SUB_FIND_("name1");
 598   EXPECT_SUB_FIND_("value1");
 599
 600   EXPECT_FIND_("TRACE_EVENT_ASYNC_END0 call");
 601   EXPECT_SUB_FIND_("id");
 602   EXPECT_SUB_FIND_(kAsyncIdStr);
 603   EXPECT_FIND_("TRACE_EVENT_ASYNC_END1 call");
 604   EXPECT_SUB_FIND_("id");
 605   EXPECT_SUB_FIND_(kAsyncIdStr);
 606   EXPECT_SUB_FIND_("name1");
 607   EXPECT_SUB_FIND_("value1");
 608   EXPECT_FIND_("TRACE_EVENT_ASYNC_END2 call");
 609   EXPECT_SUB_FIND_("id");
 610   EXPECT_SUB_FIND_(kAsyncIdStr);
 611   EXPECT_SUB_FIND_("name1");
 612   EXPECT_SUB_FIND_("value1");
 613   EXPECT_SUB_FIND_("name2");
 614   EXPECT_SUB_FIND_("value2");
 615
 616   EXPECT_FIND_("TRACE_EVENT_BEGIN_ETW0 call");
 617   EXPECT_SUB_FIND_("id");
 618   EXPECT_SUB_FIND_(kAsyncIdStr);
 619   EXPECT_SUB_FIND_("extra");
 620   EXPECT_SUB_FIND_("NULL");
 621   EXPECT_FIND_("TRACE_EVENT_BEGIN_ETW1 call");
 622   EXPECT_SUB_FIND_("id");
 623   EXPECT_SUB_FIND_(kAsyncIdStr);
 624   EXPECT_SUB_FIND_("extra");
 625   EXPECT_SUB_FIND_("value");
 626   EXPECT_FIND_("TRACE_EVENT_END_ETW0 call");
 627   EXPECT_SUB_FIND_("id");
 628   EXPECT_SUB_FIND_(kAsyncIdStr);
 629   EXPECT_SUB_FIND_("extra");
 630   EXPECT_SUB_FIND_("NULL");
 631   EXPECT_FIND_("TRACE_EVENT_END_ETW1 call");
 632   EXPECT_SUB_FIND_("id");
 633   EXPECT_SUB_FIND_(kAsyncIdStr);
 634   EXPECT_SUB_FIND_("extra");
 635   EXPECT_SUB_FIND_("value");
 636   EXPECT_FIND_("TRACE_EVENT_INSTANT_ETW0 call");
 637   EXPECT_SUB_FIND_("id");
 638   EXPECT_SUB_FIND_(kAsyncIdStr);
 639   EXPECT_SUB_FIND_("extra");
 640   EXPECT_SUB_FIND_("NULL");
 641   EXPECT_FIND_("TRACE_EVENT_INSTANT_ETW1 call");
 642   EXPECT_SUB_FIND_("id");
 643   EXPECT_SUB_FIND_(kAsyncIdStr);
 644   EXPECT_SUB_FIND_("extra");
 645   EXPECT_SUB_FIND_("value");
 646
 647   EXPECT_FIND_("TRACE_COUNTER1 call");
 648   {
 649     std::string ph;
 650     EXPECT_TRUE((item && item->GetString("ph", &ph)));
 651     EXPECT_EQ("C", ph);
 652
 653     int value;
 654     EXPECT_TRUE((item && item->GetInteger("args.value", &value)));
 655     EXPECT_EQ(31415, value);
 656   }
 657
 658   EXPECT_FIND_("TRACE_COUNTER2 call");
 659   {
 660     std::string ph;
 661     EXPECT_TRUE((item && item->GetString("ph", &ph)));
 662     EXPECT_EQ("C", ph);
 663
 664     int value;
 665     EXPECT_TRUE((item && item->GetInteger("args.a", &value)));
 666     EXPECT_EQ(30000, value);
 667
 668     EXPECT_TRUE((item && item->GetInteger("args.b", &value)));
 669     EXPECT_EQ(1415, value);
 670   }
 671
 672   EXPECT_FIND_("TRACE_COUNTER_ID1 call");
 673   {
 674     std::string id;
 675     EXPECT_TRUE((item && item->GetString("id", &id)));
 676     EXPECT_EQ("0x319009", id);
 677
 678     std::string ph;
 679     EXPECT_TRUE((item && item->GetString("ph", &ph)));
 680     EXPECT_EQ("C", ph);
 681
 682     int value;
 683     EXPECT_TRUE((item && item->GetInteger("args.value", &value)));
 684     EXPECT_EQ(31415, value);
 685   }
 686
 687   EXPECT_FIND_("TRACE_COUNTER_ID2 call");
 688   {
 689     std::string id;
 690     EXPECT_TRUE((item && item->GetString("id", &id)));
 691     EXPECT_EQ("0x319009", id);
 692
 693     std::string ph;
 694     EXPECT_TRUE((item && item->GetString("ph", &ph)));
 695     EXPECT_EQ("C", ph);
 696
 697     int value;
 698     EXPECT_TRUE((item && item->GetInteger("args.a", &value)));
 699     EXPECT_EQ(30000, value);
 700
 701     EXPECT_TRUE((item && item->GetInteger("args.b", &value)));
 702     EXPECT_EQ(1415, value);
 703   }
 704
 705   EXPECT_FIND_("TRACE_EVENT_COPY_BEGIN_WITH_ID_TID_AND_TIMESTAMP0 call");
 706   {
 707     int val;
 708     EXPECT_TRUE((item && item->GetInteger("ts", &val)));
 709     EXPECT_EQ(12345, val);
 710     EXPECT_TRUE((item && item->GetInteger("tid", &val)));
 711     EXPECT_EQ(kThreadId, val);
 712     std::string id;
 713     EXPECT_TRUE((item && item->GetString("id", &id)));
 714     EXPECT_EQ(kAsyncIdStr, id);
 715   }
 716
 717   EXPECT_FIND_("TRACE_EVENT_COPY_END_WITH_ID_TID_AND_TIMESTAMP0 call");
 718   {
 719     int val;
 720     EXPECT_TRUE((item && item->GetInteger("ts", &val)));
 721     EXPECT_EQ(23456, val);
 722     EXPECT_TRUE((item && item->GetInteger("tid", &val)));
 723     EXPECT_EQ(kThreadId, val);
 724     std::string id;
 725     EXPECT_TRUE((item && item->GetString("id", &id)));
 726     EXPECT_EQ(kAsyncIdStr, id);
 727   }
 728
 729   EXPECT_FIND_("TRACE_EVENT_BEGIN_WITH_ID_TID_AND_TIMESTAMP0 call");
 730   {
 731     int val;
 732     EXPECT_TRUE((item && item->GetInteger("ts", &val)));
 733     EXPECT_EQ(34567, val);
 734     EXPECT_TRUE((item && item->GetInteger("tid", &val)));
 735     EXPECT_EQ(kThreadId, val);
 736     std::string id;
 737     EXPECT_TRUE((item && item->GetString("id", &id)));
 738     EXPECT_EQ(kAsyncId2Str, id);
 739   }
 740
 741   EXPECT_FIND_("TRACE_EVENT_ASYNC_STEP_PAST0 call");
 742   {
 743     EXPECT_SUB_FIND_("id");
 744     EXPECT_SUB_FIND_(kAsyncId2Str);
 745     EXPECT_SUB_FIND_("step_end1");
 746     EXPECT_FIND_("TRACE_EVENT_ASYNC_STEP_PAST1 call");
 747     EXPECT_SUB_FIND_("id");
 748     EXPECT_SUB_FIND_(kAsyncId2Str);
 749     EXPECT_SUB_FIND_("step_end2");
 750     EXPECT_SUB_FIND_("name1");
 751     EXPECT_SUB_FIND_("value1");
 752   }
 753
 754   EXPECT_FIND_("TRACE_EVENT_END_WITH_ID_TID_AND_TIMESTAMP0 call");
 755   {
 756     int val;
 757     EXPECT_TRUE((item && item->GetInteger("ts", &val)));
 758     EXPECT_EQ(45678, val);
 759     EXPECT_TRUE((item && item->GetInteger("tid", &val)));
 760     EXPECT_EQ(kThreadId, val);
 761     std::string id;
 762     EXPECT_TRUE((item && item->GetString("id", &id)));
 763     EXPECT_EQ(kAsyncId2Str, id);
 764   }
 765
 766   EXPECT_FIND_("tracked object 1");
 767   {
 768     std::string phase;
 769     std::string id;
 770     std::string snapshot;
 771
 772     EXPECT_TRUE((item && item->GetString("ph", &phase)));
 773     EXPECT_EQ("N", phase);
 774     EXPECT_TRUE((item && item->GetString("id", &id)));
 775     EXPECT_EQ("0x42", id);
 776
 777     item = FindTraceEntry(trace_parsed, "tracked object 1", item);
 778     EXPECT_TRUE(item);
 779     EXPECT_TRUE(item && item->GetString("ph", &phase));
 780     EXPECT_EQ("O", phase);
 781     EXPECT_TRUE(item && item->GetString("id", &id));
 782     EXPECT_EQ("0x42", id);
 783     EXPECT_TRUE(item && item->GetString("args.snapshot", &snapshot));
 784     EXPECT_EQ("hello", snapshot);
 785
 786     item = FindTraceEntry(trace_parsed, "tracked object 1", item);
 787     EXPECT_TRUE(item);
 788     EXPECT_TRUE(item && item->GetString("ph", &phase));
 789     EXPECT_EQ("D", phase);
 790     EXPECT_TRUE(item && item->GetString("id", &id));
 791     EXPECT_EQ("0x42", id);
 792   }
 793
 794   EXPECT_FIND_("tracked object 2");
 795   {
 796     std::string phase;
 797     std::string id;
 798     std::string snapshot;
 799
 800     EXPECT_TRUE(item && item->GetString("ph", &phase));
 801     EXPECT_EQ("N", phase);
 802     EXPECT_TRUE(item && item->GetString("id", &id));
 803     EXPECT_EQ("0x2128506", id);
 804
 805     item = FindTraceEntry(trace_parsed, "tracked object 2", item);
 806     EXPECT_TRUE(item);
 807     EXPECT_TRUE(item && item->GetString("ph", &phase));
 808     EXPECT_EQ("O", phase);
 809     EXPECT_TRUE(item && item->GetString("id", &id));
 810     EXPECT_EQ("0x2128506", id);
 811     EXPECT_TRUE(item && item->GetString("args.snapshot", &snapshot));
 812     EXPECT_EQ("world", snapshot);
 813
 814     item = FindTraceEntry(trace_parsed, "tracked object 2", item);
 815     EXPECT_TRUE(item);
 816     EXPECT_TRUE(item && item->GetString("ph", &phase));
 817     EXPECT_EQ("D", phase);
 818     EXPECT_TRUE(item && item->GetString("id", &id));
 819     EXPECT_EQ("0x2128506", id);
 820   }
 821
 822   EXPECT_FIND_(kControlCharacters);
 823   EXPECT_SUB_FIND_(kControlCharacters);
 824 }
 825
 826 void TraceManyInstantEvents(int thread_id, int num_events,
 827                             WaitableEvent* task_complete_event) {
 828   for (int i = 0; i < num_events; i++) {
 829     TRACE_EVENT_INSTANT2("all", "multi thread event",
 830                          TRACE_EVENT_SCOPE_THREAD,
 831                          "thread", thread_id,
 832                          "event", i);
 833   }
 834
 835   if (task_complete_event)
 836     task_complete_event->Signal();
 837 }
 838
 839 void ValidateInstantEventPresentOnEveryThread(const ListValue& trace_parsed,
 840                                               int num_threads,
 841                                               int num_events) {
 842   std::map<int, std::map<int, bool> > results;
 843
 844   size_t trace_parsed_count = trace_parsed.GetSize();
 845   for (size_t i = 0; i < trace_parsed_count; i++) {
 846     const Value* value = NULL;
 847     trace_parsed.Get(i, &value);
 848     if (!value || value->GetType() != Value::TYPE_DICTIONARY)
 849       continue;
 850     const DictionaryValue* dict = static_cast<const DictionaryValue*>(value);
 851     std::string name;
 852     dict->GetString("name", &name);
 853     if (name != "multi thread event")
 854       continue;
 855
 856     int thread = 0;
 857     int event = 0;
 858     EXPECT_TRUE(dict->GetInteger("args.thread", &thread));
 859     EXPECT_TRUE(dict->GetInteger("args.event", &event));
 860     results[thread][event] = true;
 861   }
 862
 863   EXPECT_FALSE(results[-1][-1]);
 864   for (int thread = 0; thread < num_threads; thread++) {
 865     for (int event = 0; event < num_events; event++) {
 866       EXPECT_TRUE(results[thread][event]);
 867     }
 868   }
 869 }
 870
 871 }  // namespace
 872
 873 // Simple Test for emitting data and validating it was received.
 874 TEST_F(TraceEventTestFixture, DataCaptured) {
 875   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
 876                                       TraceLog::RECORDING_MODE);
 877
 878   TraceWithAllMacroVariants(NULL);
 879
 880   EndTraceAndFlush();
 881
 882   ValidateAllTraceMacrosCreatedData(trace_parsed_);
 883 }
 884
 885 class MockEnabledStateChangedObserver :
 886       public TraceLog::EnabledStateObserver {
 887  public:
 888   MOCK_METHOD0(OnTraceLogEnabled, void());
 889   MOCK_METHOD0(OnTraceLogDisabled, void());
 890 };
 891
 892 TEST_F(TraceEventTestFixture, EnabledObserverFiresOnEnable) {
 893   MockEnabledStateChangedObserver observer;
 894   TraceLog::GetInstance()->AddEnabledStateObserver(&observer);
 895
 896   EXPECT_CALL(observer, OnTraceLogEnabled())
 897       .Times(1);
 898   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
 899                                       TraceLog::RECORDING_MODE);
 900   testing::Mock::VerifyAndClear(&observer);
 901   EXPECT_TRUE(TraceLog::GetInstance()->IsEnabled());
 902
 903   // Cleanup.
 904   TraceLog::GetInstance()->RemoveEnabledStateObserver(&observer);
 905   TraceLog::GetInstance()->SetDisabled();
 906 }
 907
 908 TEST_F(TraceEventTestFixture, EnabledObserverDoesntFireOnSecondEnable) {
 909   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
 910                                       TraceLog::RECORDING_MODE);
 911
 912   testing::StrictMock<MockEnabledStateChangedObserver> observer;
 913   TraceLog::GetInstance()->AddEnabledStateObserver(&observer);
 914
 915   EXPECT_CALL(observer, OnTraceLogEnabled())
 916       .Times(0);
 917   EXPECT_CALL(observer, OnTraceLogDisabled())
 918       .Times(0);
 919   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
 920                                       TraceLog::RECORDING_MODE);
 921   testing::Mock::VerifyAndClear(&observer);
 922   EXPECT_TRUE(TraceLog::GetInstance()->IsEnabled());
 923
 924   // Cleanup.
 925   TraceLog::GetInstance()->RemoveEnabledStateObserver(&observer);
 926   TraceLog::GetInstance()->SetDisabled();
 927   TraceLog::GetInstance()->SetDisabled();
 928 }
 929
 930 TEST_F(TraceEventTestFixture, EnabledObserverFiresOnFirstDisable) {
 931   TraceConfig tc_inc_all("*", "");
 932   TraceLog::GetInstance()->SetEnabled(tc_inc_all, TraceLog::RECORDING_MODE);
 933   TraceLog::GetInstance()->SetEnabled(tc_inc_all, TraceLog::RECORDING_MODE);
 934
 935   testing::StrictMock<MockEnabledStateChangedObserver> observer;
 936   TraceLog::GetInstance()->AddEnabledStateObserver(&observer);
 937
 938   EXPECT_CALL(observer, OnTraceLogEnabled())
 939       .Times(0);
 940   EXPECT_CALL(observer, OnTraceLogDisabled())
 941       .Times(1);
 942   TraceLog::GetInstance()->SetDisabled();
 943   testing::Mock::VerifyAndClear(&observer);
 944
 945   // Cleanup.
 946   TraceLog::GetInstance()->RemoveEnabledStateObserver(&observer);
 947   TraceLog::GetInstance()->SetDisabled();
 948 }
 949
 950 TEST_F(TraceEventTestFixture, EnabledObserverFiresOnDisable) {
 951   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
 952                                       TraceLog::RECORDING_MODE);
 953
 954   MockEnabledStateChangedObserver observer;
 955   TraceLog::GetInstance()->AddEnabledStateObserver(&observer);
 956
 957   EXPECT_CALL(observer, OnTraceLogDisabled())
 958       .Times(1);
 959   TraceLog::GetInstance()->SetDisabled();
 960   testing::Mock::VerifyAndClear(&observer);
 961
 962   // Cleanup.
 963   TraceLog::GetInstance()->RemoveEnabledStateObserver(&observer);
 964 }
 965
 966 // Tests the IsEnabled() state of TraceLog changes before callbacks.
 967 class AfterStateChangeEnabledStateObserver
 968     : public TraceLog::EnabledStateObserver {
 969  public:
 970   AfterStateChangeEnabledStateObserver() {}
 971   virtual ~AfterStateChangeEnabledStateObserver() {}
 972
 973   // TraceLog::EnabledStateObserver overrides:
 974   void OnTraceLogEnabled() override {
 975     EXPECT_TRUE(TraceLog::GetInstance()->IsEnabled());
 976   }
 977
 978   void OnTraceLogDisabled() override {
 979     EXPECT_FALSE(TraceLog::GetInstance()->IsEnabled());
 980   }
 981 };
 982
 983 TEST_F(TraceEventTestFixture, ObserversFireAfterStateChange) {
 984   AfterStateChangeEnabledStateObserver observer;
 985   TraceLog::GetInstance()->AddEnabledStateObserver(&observer);
 986
 987   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
 988                                       TraceLog::RECORDING_MODE);
 989   EXPECT_TRUE(TraceLog::GetInstance()->IsEnabled());
 990
 991   TraceLog::GetInstance()->SetDisabled();
 992   EXPECT_FALSE(TraceLog::GetInstance()->IsEnabled());
 993
 994   TraceLog::GetInstance()->RemoveEnabledStateObserver(&observer);
 995 }
 996
 997 // Tests that a state observer can remove itself during a callback.
 998 class SelfRemovingEnabledStateObserver
 999     : public TraceLog::EnabledStateObserver {
1000  public:
1001   SelfRemovingEnabledStateObserver() {}
1002   virtual ~SelfRemovingEnabledStateObserver() {}
1003
1004   // TraceLog::EnabledStateObserver overrides:
1005   void OnTraceLogEnabled() override {}
1006
1007   void OnTraceLogDisabled() override {
1008     TraceLog::GetInstance()->RemoveEnabledStateObserver(this);
1009   }
1010 };
1011
1012 TEST_F(TraceEventTestFixture, SelfRemovingObserver) {
1013   ASSERT_EQ(0u, TraceLog::GetInstance()->GetObserverCountForTest());
1014
1015   SelfRemovingEnabledStateObserver observer;
1016   TraceLog::GetInstance()->AddEnabledStateObserver(&observer);
1017   EXPECT_EQ(1u, TraceLog::GetInstance()->GetObserverCountForTest());
1018
1019   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
1020                                       TraceLog::RECORDING_MODE);
1021   TraceLog::GetInstance()->SetDisabled();
1022   // The observer removed itself on disable.
1023   EXPECT_EQ(0u, TraceLog::GetInstance()->GetObserverCountForTest());
1024 }
1025
1026 bool IsNewTrace() {
1027   bool is_new_trace;
1028   TRACE_EVENT_IS_NEW_TRACE(&is_new_trace);
1029   return is_new_trace;
1030 }
1031
1032 TEST_F(TraceEventTestFixture, NewTraceRecording) {
1033   ASSERT_FALSE(IsNewTrace());
1034   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
1035                                       TraceLog::RECORDING_MODE);
1036   // First call to IsNewTrace() should succeed. But, the second shouldn't.
1037   ASSERT_TRUE(IsNewTrace());
1038   ASSERT_FALSE(IsNewTrace());
1039   EndTraceAndFlush();
1040
1041   // IsNewTrace() should definitely be false now.
1042   ASSERT_FALSE(IsNewTrace());
1043
1044   // Start another trace. IsNewTrace() should become true again, briefly, as
1045   // before.
1046   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
1047                                       TraceLog::RECORDING_MODE);
1048   ASSERT_TRUE(IsNewTrace());
1049   ASSERT_FALSE(IsNewTrace());
1050
1051   // Cleanup.
1052   EndTraceAndFlush();
1053 }
1054
1055
1056 // Test that categories work.
1057 TEST_F(TraceEventTestFixture, Categories) {
1058   // Test that categories that are used can be retrieved whether trace was
1059   // enabled or disabled when the trace event was encountered.
1060   TRACE_EVENT_INSTANT0("c1", "name", TRACE_EVENT_SCOPE_THREAD);
1061   TRACE_EVENT_INSTANT0("c2", "name", TRACE_EVENT_SCOPE_THREAD);
1062   BeginTrace();
1063   TRACE_EVENT_INSTANT0("c3", "name", TRACE_EVENT_SCOPE_THREAD);
1064   TRACE_EVENT_INSTANT0("c4", "name", TRACE_EVENT_SCOPE_THREAD);
1065   // Category groups containing more than one category.
1066   TRACE_EVENT_INSTANT0("c5,c6", "name", TRACE_EVENT_SCOPE_THREAD);
1067   TRACE_EVENT_INSTANT0("c7,c8", "name", TRACE_EVENT_SCOPE_THREAD);
1068   TRACE_EVENT_INSTANT0(TRACE_DISABLED_BY_DEFAULT("c9"), "name",
1069                        TRACE_EVENT_SCOPE_THREAD);
1070
1071   EndTraceAndFlush();
1072   std::vector<std::string> cat_groups;
1073   TraceLog::GetInstance()->GetKnownCategoryGroups(&cat_groups);
1074   EXPECT_TRUE(std::find(cat_groups.begin(),
1075                         cat_groups.end(), "c1") != cat_groups.end());
1076   EXPECT_TRUE(std::find(cat_groups.begin(),
1077                         cat_groups.end(), "c2") != cat_groups.end());
1078   EXPECT_TRUE(std::find(cat_groups.begin(),
1079                         cat_groups.end(), "c3") != cat_groups.end());
1080   EXPECT_TRUE(std::find(cat_groups.begin(),
1081                         cat_groups.end(), "c4") != cat_groups.end());
1082   EXPECT_TRUE(std::find(cat_groups.begin(),
1083                         cat_groups.end(), "c5,c6") != cat_groups.end());
1084   EXPECT_TRUE(std::find(cat_groups.begin(),
1085                         cat_groups.end(), "c7,c8") != cat_groups.end());
1086   EXPECT_TRUE(std::find(cat_groups.begin(),
1087                         cat_groups.end(),
1088                         "disabled-by-default-c9") != cat_groups.end());
1089   // Make sure metadata isn't returned.
1090   EXPECT_TRUE(std::find(cat_groups.begin(),
1091                         cat_groups.end(), "__metadata") == cat_groups.end());
1092
1093   const std::vector<std::string> empty_categories;
1094   std::vector<std::string> included_categories;
1095   std::vector<std::string> excluded_categories;
1096
1097   // Test that category filtering works.
1098
1099   // Include nonexistent category -> no events
1100   Clear();
1101   included_categories.clear();
1102   TraceLog::GetInstance()->SetEnabled(TraceConfig("not_found823564786", ""),
1103                                       TraceLog::RECORDING_MODE);
1104   TRACE_EVENT_INSTANT0("cat1", "name", TRACE_EVENT_SCOPE_THREAD);
1105   TRACE_EVENT_INSTANT0("cat2", "name", TRACE_EVENT_SCOPE_THREAD);
1106   EndTraceAndFlush();
1107   DropTracedMetadataRecords();
1108   EXPECT_TRUE(trace_parsed_.empty());
1109
1110   // Include existent category -> only events of that category
1111   Clear();
1112   included_categories.clear();
1113   TraceLog::GetInstance()->SetEnabled(TraceConfig("inc", ""),
1114                                       TraceLog::RECORDING_MODE);
1115   TRACE_EVENT_INSTANT0("inc", "name", TRACE_EVENT_SCOPE_THREAD);
1116   TRACE_EVENT_INSTANT0("inc2", "name", TRACE_EVENT_SCOPE_THREAD);
1117   EndTraceAndFlush();
1118   DropTracedMetadataRecords();
1119   EXPECT_TRUE(FindMatchingValue("cat", "inc"));
1120   EXPECT_FALSE(FindNonMatchingValue("cat", "inc"));
1121
1122   // Include existent wildcard -> all categories matching wildcard
1123   Clear();
1124   included_categories.clear();
1125   TraceLog::GetInstance()->SetEnabled(
1126       TraceConfig("inc_wildcard_*,inc_wildchar_?_end", ""),
1127       TraceLog::RECORDING_MODE);
1128   TRACE_EVENT_INSTANT0("inc_wildcard_abc", "included",
1129       TRACE_EVENT_SCOPE_THREAD);
1130   TRACE_EVENT_INSTANT0("inc_wildcard_", "included", TRACE_EVENT_SCOPE_THREAD);
1131   TRACE_EVENT_INSTANT0("inc_wildchar_x_end", "included",
1132       TRACE_EVENT_SCOPE_THREAD);
1133   TRACE_EVENT_INSTANT0("inc_wildchar_bla_end", "not_inc",
1134       TRACE_EVENT_SCOPE_THREAD);
1135   TRACE_EVENT_INSTANT0("cat1", "not_inc", TRACE_EVENT_SCOPE_THREAD);
1136   TRACE_EVENT_INSTANT0("cat2", "not_inc", TRACE_EVENT_SCOPE_THREAD);
1137   TRACE_EVENT_INSTANT0("inc_wildcard_category,other_category", "included",
1138       TRACE_EVENT_SCOPE_THREAD);
1139   TRACE_EVENT_INSTANT0(
1140       "non_included_category,inc_wildcard_category", "included",
1141       TRACE_EVENT_SCOPE_THREAD);
1142   EndTraceAndFlush();
1143   EXPECT_TRUE(FindMatchingValue("cat", "inc_wildcard_abc"));
1144   EXPECT_TRUE(FindMatchingValue("cat", "inc_wildcard_"));
1145   EXPECT_TRUE(FindMatchingValue("cat", "inc_wildchar_x_end"));
1146   EXPECT_FALSE(FindMatchingValue("name", "not_inc"));
1147   EXPECT_TRUE(FindMatchingValue("cat", "inc_wildcard_category,other_category"));
1148   EXPECT_TRUE(FindMatchingValue("cat",
1149                                 "non_included_category,inc_wildcard_category"));
1150
1151   included_categories.clear();
1152
1153   // Exclude nonexistent category -> all events
1154   Clear();
1155   TraceLog::GetInstance()->SetEnabled(TraceConfig("-not_found823564786", ""),
1156                                       TraceLog::RECORDING_MODE);
1157   TRACE_EVENT_INSTANT0("cat1", "name", TRACE_EVENT_SCOPE_THREAD);
1158   TRACE_EVENT_INSTANT0("cat2", "name", TRACE_EVENT_SCOPE_THREAD);
1159   TRACE_EVENT_INSTANT0("category1,category2", "name", TRACE_EVENT_SCOPE_THREAD);
1160   EndTraceAndFlush();
1161   EXPECT_TRUE(FindMatchingValue("cat", "cat1"));
1162   EXPECT_TRUE(FindMatchingValue("cat", "cat2"));
1163   EXPECT_TRUE(FindMatchingValue("cat", "category1,category2"));
1164
1165   // Exclude existent category -> only events of other categories
1166   Clear();
1167   TraceLog::GetInstance()->SetEnabled(TraceConfig("-inc", ""),
1168                                       TraceLog::RECORDING_MODE);
1169   TRACE_EVENT_INSTANT0("inc", "name", TRACE_EVENT_SCOPE_THREAD);
1170   TRACE_EVENT_INSTANT0("inc2", "name", TRACE_EVENT_SCOPE_THREAD);
1171   TRACE_EVENT_INSTANT0("inc2,inc", "name", TRACE_EVENT_SCOPE_THREAD);
1172   TRACE_EVENT_INSTANT0("inc,inc2", "name", TRACE_EVENT_SCOPE_THREAD);
1173   EndTraceAndFlush();
1174   EXPECT_TRUE(FindMatchingValue("cat", "inc2"));
1175   EXPECT_FALSE(FindMatchingValue("cat", "inc"));
1176   EXPECT_TRUE(FindMatchingValue("cat", "inc2,inc"));
1177   EXPECT_TRUE(FindMatchingValue("cat", "inc,inc2"));
1178
1179   // Exclude existent wildcard -> all categories not matching wildcard
1180   Clear();
1181   TraceLog::GetInstance()->SetEnabled(
1182       TraceConfig("-inc_wildcard_*,-inc_wildchar_?_end", ""),
1183       TraceLog::RECORDING_MODE);
1184   TRACE_EVENT_INSTANT0("inc_wildcard_abc", "not_inc",
1185       TRACE_EVENT_SCOPE_THREAD);
1186   TRACE_EVENT_INSTANT0("inc_wildcard_", "not_inc",
1187       TRACE_EVENT_SCOPE_THREAD);
1188   TRACE_EVENT_INSTANT0("inc_wildchar_x_end", "not_inc",
1189       TRACE_EVENT_SCOPE_THREAD);
1190   TRACE_EVENT_INSTANT0("inc_wildchar_bla_end", "included",
1191       TRACE_EVENT_SCOPE_THREAD);
1192   TRACE_EVENT_INSTANT0("cat1", "included", TRACE_EVENT_SCOPE_THREAD);
1193   TRACE_EVENT_INSTANT0("cat2", "included", TRACE_EVENT_SCOPE_THREAD);
1194   EndTraceAndFlush();
1195   EXPECT_TRUE(FindMatchingValue("cat", "inc_wildchar_bla_end"));
1196   EXPECT_TRUE(FindMatchingValue("cat", "cat1"));
1197   EXPECT_TRUE(FindMatchingValue("cat", "cat2"));
1198   EXPECT_FALSE(FindMatchingValue("name", "not_inc"));
1199 }
1200
1201
1202 // Test EVENT_WATCH_NOTIFICATION
1203 TEST_F(TraceEventTestFixture, EventWatchNotification) {
1204   // Basic one occurrence.
1205   BeginTrace();
1206   TraceLog::WatchEventCallback callback =
1207       base::Bind(&TraceEventTestFixture::OnWatchEventMatched,
1208                  base::Unretained(this));
1209   TraceLog::GetInstance()->SetWatchEvent("cat", "event", callback);
1210   TRACE_EVENT_INSTANT0("cat", "event", TRACE_EVENT_SCOPE_THREAD);
1211   EndTraceAndFlush();
1212   EXPECT_EQ(event_watch_notification_, 1);
1213
1214   // Auto-reset after end trace.
1215   BeginTrace();
1216   TraceLog::GetInstance()->SetWatchEvent("cat", "event", callback);
1217   EndTraceAndFlush();
1218   BeginTrace();
1219   TRACE_EVENT_INSTANT0("cat", "event", TRACE_EVENT_SCOPE_THREAD);
1220   EndTraceAndFlush();
1221   EXPECT_EQ(event_watch_notification_, 0);
1222
1223   // Multiple occurrence.
1224   BeginTrace();
1225   int num_occurrences = 5;
1226   TraceLog::GetInstance()->SetWatchEvent("cat", "event", callback);
1227   for (int i = 0; i < num_occurrences; ++i)
1228     TRACE_EVENT_INSTANT0("cat", "event", TRACE_EVENT_SCOPE_THREAD);
1229   EndTraceAndFlush();
1230   EXPECT_EQ(event_watch_notification_, num_occurrences);
1231
1232   // Wrong category.
1233   BeginTrace();
1234   TraceLog::GetInstance()->SetWatchEvent("cat", "event", callback);
1235   TRACE_EVENT_INSTANT0("wrong_cat", "event", TRACE_EVENT_SCOPE_THREAD);
1236   EndTraceAndFlush();
1237   EXPECT_EQ(event_watch_notification_, 0);
1238
1239   // Wrong name.
1240   BeginTrace();
1241   TraceLog::GetInstance()->SetWatchEvent("cat", "event", callback);
1242   TRACE_EVENT_INSTANT0("cat", "wrong_event", TRACE_EVENT_SCOPE_THREAD);
1243   EndTraceAndFlush();
1244   EXPECT_EQ(event_watch_notification_, 0);
1245
1246   // Canceled.
1247   BeginTrace();
1248   TraceLog::GetInstance()->SetWatchEvent("cat", "event", callback);
1249   TraceLog::GetInstance()->CancelWatchEvent();
1250   TRACE_EVENT_INSTANT0("cat", "event", TRACE_EVENT_SCOPE_THREAD);
1251   EndTraceAndFlush();
1252   EXPECT_EQ(event_watch_notification_, 0);
1253 }
1254
1255 // Test ASYNC_BEGIN/END events
1256 TEST_F(TraceEventTestFixture, AsyncBeginEndEvents) {
1257   BeginTrace();
1258
1259   unsigned long long id = 0xfeedbeeffeedbeefull;
1260   TRACE_EVENT_ASYNC_BEGIN0("cat", "name1", id);
1261   TRACE_EVENT_ASYNC_STEP_INTO0("cat", "name1", id, "step1");
1262   TRACE_EVENT_ASYNC_END0("cat", "name1", id);
1263   TRACE_EVENT_BEGIN0("cat", "name2");
1264   TRACE_EVENT_ASYNC_BEGIN0("cat", "name3", 0);
1265   TRACE_EVENT_ASYNC_STEP_PAST0("cat", "name3", 0, "step2");
1266
1267   EndTraceAndFlush();
1268
1269   EXPECT_TRUE(FindNamePhase("name1", "S"));
1270   EXPECT_TRUE(FindNamePhase("name1", "T"));
1271   EXPECT_TRUE(FindNamePhase("name1", "F"));
1272
1273   std::string id_str;
1274   StringAppendF(&id_str, "0x%llx", id);
1275
1276   EXPECT_TRUE(FindNamePhaseKeyValue("name1", "S", "id", id_str.c_str()));
1277   EXPECT_TRUE(FindNamePhaseKeyValue("name1", "T", "id", id_str.c_str()));
1278   EXPECT_TRUE(FindNamePhaseKeyValue("name1", "F", "id", id_str.c_str()));
1279   EXPECT_TRUE(FindNamePhaseKeyValue("name3", "S", "id", "0x0"));
1280   EXPECT_TRUE(FindNamePhaseKeyValue("name3", "p", "id", "0x0"));
1281
1282   // BEGIN events should not have id
1283   EXPECT_FALSE(FindNamePhaseKeyValue("name2", "B", "id", "0"));
1284 }
1285
1286 // Test ASYNC_BEGIN/END events
1287 TEST_F(TraceEventTestFixture, AsyncBeginEndPointerMangling) {
1288   void* ptr = this;
1289
1290   TraceLog::GetInstance()->SetProcessID(100);
1291   BeginTrace();
1292   TRACE_EVENT_ASYNC_BEGIN0("cat", "name1", ptr);
1293   TRACE_EVENT_ASYNC_BEGIN0("cat", "name2", ptr);
1294   EndTraceAndFlush();
1295
1296   TraceLog::GetInstance()->SetProcessID(200);
1297   BeginTrace();
1298   TRACE_EVENT_ASYNC_END0("cat", "name1", ptr);
1299   EndTraceAndFlush();
1300
1301   DictionaryValue* async_begin = FindNamePhase("name1", "S");
1302   DictionaryValue* async_begin2 = FindNamePhase("name2", "S");
1303   DictionaryValue* async_end = FindNamePhase("name1", "F");
1304   EXPECT_TRUE(async_begin);
1305   EXPECT_TRUE(async_begin2);
1306   EXPECT_TRUE(async_end);
1307
1308   Value* value = NULL;
1309   std::string async_begin_id_str;
1310   std::string async_begin2_id_str;
1311   std::string async_end_id_str;
1312   ASSERT_TRUE(async_begin->Get("id", &value));
1313   ASSERT_TRUE(value->GetAsString(&async_begin_id_str));
1314   ASSERT_TRUE(async_begin2->Get("id", &value));
1315   ASSERT_TRUE(value->GetAsString(&async_begin2_id_str));
1316   ASSERT_TRUE(async_end->Get("id", &value));
1317   ASSERT_TRUE(value->GetAsString(&async_end_id_str));
1318
1319   EXPECT_STREQ(async_begin_id_str.c_str(), async_begin2_id_str.c_str());
1320   EXPECT_STRNE(async_begin_id_str.c_str(), async_end_id_str.c_str());
1321 }
1322
1323 // Test that static strings are not copied.
1324 TEST_F(TraceEventTestFixture, StaticStringVsString) {
1325   TraceLog* tracer = TraceLog::GetInstance();
1326   // Make sure old events are flushed:
1327   EXPECT_EQ(0u, tracer->GetStatus().event_count);
1328   const unsigned char* category_group_enabled =
1329       TRACE_EVENT_API_GET_CATEGORY_GROUP_ENABLED("cat");
1330
1331   {
1332     BeginTrace();
1333     // Test that string arguments are copied.
1334     TraceEventHandle handle1 =
1335         trace_event_internal::AddTraceEvent(
1336             TRACE_EVENT_PHASE_INSTANT, category_group_enabled, "name1", 0, 0,
1337             "arg1", std::string("argval"), "arg2", std::string("argval"));
1338     // Test that static TRACE_STR_COPY string arguments are copied.
1339     TraceEventHandle handle2 =
1340         trace_event_internal::AddTraceEvent(
1341             TRACE_EVENT_PHASE_INSTANT, category_group_enabled, "name2", 0, 0,
1342             "arg1", TRACE_STR_COPY("argval"),
1343             "arg2", TRACE_STR_COPY("argval"));
1344     EXPECT_GT(tracer->GetStatus().event_count, 1u);
1345     const TraceEvent* event1 = tracer->GetEventByHandle(handle1);
1346     const TraceEvent* event2 = tracer->GetEventByHandle(handle2);
1347     ASSERT_TRUE(event1);
1348     ASSERT_TRUE(event2);
1349     EXPECT_STREQ("name1", event1->name());
1350     EXPECT_STREQ("name2", event2->name());
1351     EXPECT_TRUE(event1->parameter_copy_storage() != NULL);
1352     EXPECT_TRUE(event2->parameter_copy_storage() != NULL);
1353     EXPECT_GT(event1->parameter_copy_storage()->size(), 0u);
1354     EXPECT_GT(event2->parameter_copy_storage()->size(), 0u);
1355     EndTraceAndFlush();
1356   }
1357
1358   {
1359     BeginTrace();
1360     // Test that static literal string arguments are not copied.
1361     TraceEventHandle handle1 =
1362         trace_event_internal::AddTraceEvent(
1363             TRACE_EVENT_PHASE_INSTANT, category_group_enabled, "name1", 0, 0,
1364             "arg1", "argval", "arg2", "argval");
1365     // Test that static TRACE_STR_COPY NULL string arguments are not copied.
1366     const char* str1 = NULL;
1367     const char* str2 = NULL;
1368     TraceEventHandle handle2 =
1369         trace_event_internal::AddTraceEvent(
1370             TRACE_EVENT_PHASE_INSTANT, category_group_enabled, "name2", 0, 0,
1371             "arg1", TRACE_STR_COPY(str1),
1372             "arg2", TRACE_STR_COPY(str2));
1373     EXPECT_GT(tracer->GetStatus().event_count, 1u);
1374     const TraceEvent* event1 = tracer->GetEventByHandle(handle1);
1375     const TraceEvent* event2 = tracer->GetEventByHandle(handle2);
1376     ASSERT_TRUE(event1);
1377     ASSERT_TRUE(event2);
1378     EXPECT_STREQ("name1", event1->name());
1379     EXPECT_STREQ("name2", event2->name());
1380     EXPECT_TRUE(event1->parameter_copy_storage() == NULL);
1381     EXPECT_TRUE(event2->parameter_copy_storage() == NULL);
1382     EndTraceAndFlush();
1383   }
1384 }
1385
1386 // Test that data sent from other threads is gathered
1387 TEST_F(TraceEventTestFixture, DataCapturedOnThread) {
1388   BeginTrace();
1389
1390   Thread thread("1");
1391   WaitableEvent task_complete_event(false, false);
1392   thread.Start();
1393
1394   thread.task_runner()->PostTask(
1395       FROM_HERE, base::Bind(&TraceWithAllMacroVariants, &task_complete_event));
1396   task_complete_event.Wait();
1397   thread.Stop();
1398
1399   EndTraceAndFlush();
1400   ValidateAllTraceMacrosCreatedData(trace_parsed_);
1401 }
1402
1403 // Test that data sent from multiple threads is gathered
1404 TEST_F(TraceEventTestFixture, DataCapturedManyThreads) {
1405   BeginTrace();
1406
1407   const int num_threads = 4;
1408   const int num_events = 4000;
1409   Thread* threads[num_threads];
1410   WaitableEvent* task_complete_events[num_threads];
1411   for (int i = 0; i < num_threads; i++) {
1412     threads[i] = new Thread(StringPrintf("Thread %d", i));
1413     task_complete_events[i] = new WaitableEvent(false, false);
1414     threads[i]->Start();
1415     threads[i]->task_runner()->PostTask(
1416         FROM_HERE, base::Bind(&TraceManyInstantEvents, i, num_events,
1417                               task_complete_events[i]));
1418   }
1419
1420   for (int i = 0; i < num_threads; i++) {
1421     task_complete_events[i]->Wait();
1422   }
1423
1424   // Let half of the threads end before flush.
1425   for (int i = 0; i < num_threads / 2; i++) {
1426     threads[i]->Stop();
1427     delete threads[i];
1428     delete task_complete_events[i];
1429   }
1430
1431   EndTraceAndFlushInThreadWithMessageLoop();
1432   ValidateInstantEventPresentOnEveryThread(trace_parsed_,
1433                                            num_threads, num_events);
1434
1435   // Let the other half of the threads end after flush.
1436   for (int i = num_threads / 2; i < num_threads; i++) {
1437     threads[i]->Stop();
1438     delete threads[i];
1439     delete task_complete_events[i];
1440   }
1441 }
1442
1443 // Test that thread and process names show up in the trace
1444 TEST_F(TraceEventTestFixture, ThreadNames) {
1445   // Create threads before we enable tracing to make sure
1446   // that tracelog still captures them.
1447   const int kNumThreads = 4;
1448   const int kNumEvents = 10;
1449   Thread* threads[kNumThreads];
1450   PlatformThreadId thread_ids[kNumThreads];
1451   for (int i = 0; i < kNumThreads; i++)
1452     threads[i] = new Thread(StringPrintf("Thread %d", i));
1453
1454   // Enable tracing.
1455   BeginTrace();
1456
1457   // Now run some trace code on these threads.
1458   WaitableEvent* task_complete_events[kNumThreads];
1459   for (int i = 0; i < kNumThreads; i++) {
1460     task_complete_events[i] = new WaitableEvent(false, false);
1461     threads[i]->Start();
1462     thread_ids[i] = threads[i]->thread_id();
1463     threads[i]->task_runner()->PostTask(
1464         FROM_HERE, base::Bind(&TraceManyInstantEvents, i, kNumEvents,
1465                               task_complete_events[i]));
1466   }
1467   for (int i = 0; i < kNumThreads; i++) {
1468     task_complete_events[i]->Wait();
1469   }
1470
1471   // Shut things down.
1472   for (int i = 0; i < kNumThreads; i++) {
1473     threads[i]->Stop();
1474     delete threads[i];
1475     delete task_complete_events[i];
1476   }
1477
1478   EndTraceAndFlush();
1479
1480   std::string tmp;
1481   int tmp_int;
1482   const DictionaryValue* item;
1483
1484   // Make sure we get thread name metadata.
1485   // Note, the test suite may have created a ton of threads.
1486   // So, we'll have thread names for threads we didn't create.
1487   std::vector<const DictionaryValue*> items =
1488       FindTraceEntries(trace_parsed_, "thread_name");
1489   for (int i = 0; i < static_cast<int>(items.size()); i++) {
1490     item = items[i];
1491     ASSERT_TRUE(item);
1492     EXPECT_TRUE(item->GetInteger("tid", &tmp_int));
1493
1494     // See if this thread name is one of the threads we just created
1495     for (int j = 0; j < kNumThreads; j++) {
1496       if (static_cast<int>(thread_ids[j]) != tmp_int)
1497         continue;
1498
1499       std::string expected_name = StringPrintf("Thread %d", j);
1500       EXPECT_TRUE(item->GetString("ph", &tmp) && tmp == "M");
1501       EXPECT_TRUE(item->GetInteger("pid", &tmp_int) &&
1502                   tmp_int == static_cast<int>(base::GetCurrentProcId()));
1503       // If the thread name changes or the tid gets reused, the name will be
1504       // a comma-separated list of thread names, so look for a substring.
1505       EXPECT_TRUE(item->GetString("args.name", &tmp) &&
1506                   tmp.find(expected_name) != std::string::npos);
1507     }
1508   }
1509 }
1510
1511 TEST_F(TraceEventTestFixture, ThreadNameChanges) {
1512   BeginTrace();
1513
1514   PlatformThread::SetName("");
1515   TRACE_EVENT_INSTANT0("drink", "water", TRACE_EVENT_SCOPE_THREAD);
1516
1517   PlatformThread::SetName("cafe");
1518   TRACE_EVENT_INSTANT0("drink", "coffee", TRACE_EVENT_SCOPE_THREAD);
1519
1520   PlatformThread::SetName("shop");
1521   // No event here, so won't appear in combined name.
1522
1523   PlatformThread::SetName("pub");
1524   TRACE_EVENT_INSTANT0("drink", "beer", TRACE_EVENT_SCOPE_THREAD);
1525   TRACE_EVENT_INSTANT0("drink", "wine", TRACE_EVENT_SCOPE_THREAD);
1526
1527   PlatformThread::SetName(" bar");
1528   TRACE_EVENT_INSTANT0("drink", "whisky", TRACE_EVENT_SCOPE_THREAD);
1529
1530   EndTraceAndFlush();
1531
1532   std::vector<const DictionaryValue*> items =
1533       FindTraceEntries(trace_parsed_, "thread_name");
1534   EXPECT_EQ(1u, items.size());
1535   ASSERT_GT(items.size(), 0u);
1536   const DictionaryValue* item = items[0];
1537   ASSERT_TRUE(item);
1538   int tid;
1539   EXPECT_TRUE(item->GetInteger("tid", &tid));
1540   EXPECT_EQ(PlatformThread::CurrentId(), static_cast<PlatformThreadId>(tid));
1541
1542   std::string expected_name = "cafe,pub, bar";
1543   std::string tmp;
1544   EXPECT_TRUE(item->GetString("args.name", &tmp));
1545   EXPECT_EQ(expected_name, tmp);
1546 }
1547
1548 // Test that the disabled trace categories are included/excluded from the
1549 // trace output correctly.
1550 TEST_F(TraceEventTestFixture, DisabledCategories) {
1551   BeginTrace();
1552   TRACE_EVENT_INSTANT0(TRACE_DISABLED_BY_DEFAULT("cc"), "first",
1553                        TRACE_EVENT_SCOPE_THREAD);
1554   TRACE_EVENT_INSTANT0("included", "first", TRACE_EVENT_SCOPE_THREAD);
1555   EndTraceAndFlush();
1556   {
1557     const DictionaryValue* item = NULL;
1558     ListValue& trace_parsed = trace_parsed_;
1559     EXPECT_NOT_FIND_("disabled-by-default-cc");
1560     EXPECT_FIND_("included");
1561   }
1562   Clear();
1563
1564   BeginSpecificTrace("disabled-by-default-cc");
1565   TRACE_EVENT_INSTANT0(TRACE_DISABLED_BY_DEFAULT("cc"), "second",
1566                        TRACE_EVENT_SCOPE_THREAD);
1567   TRACE_EVENT_INSTANT0("other_included", "second", TRACE_EVENT_SCOPE_THREAD);
1568   EndTraceAndFlush();
1569
1570   {
1571     const DictionaryValue* item = NULL;
1572     ListValue& trace_parsed = trace_parsed_;
1573     EXPECT_FIND_("disabled-by-default-cc");
1574     EXPECT_FIND_("other_included");
1575   }
1576
1577   Clear();
1578
1579   BeginSpecificTrace("other_included");
1580   TRACE_EVENT_INSTANT0(TRACE_DISABLED_BY_DEFAULT("cc") ",other_included",
1581                        "first", TRACE_EVENT_SCOPE_THREAD);
1582   TRACE_EVENT_INSTANT0("other_included," TRACE_DISABLED_BY_DEFAULT("cc"),
1583                        "second", TRACE_EVENT_SCOPE_THREAD);
1584   EndTraceAndFlush();
1585
1586   {
1587     const DictionaryValue* item = NULL;
1588     ListValue& trace_parsed = trace_parsed_;
1589     EXPECT_FIND_("disabled-by-default-cc,other_included");
1590     EXPECT_FIND_("other_included,disabled-by-default-cc");
1591   }
1592 }
1593
1594 TEST_F(TraceEventTestFixture, NormallyNoDeepCopy) {
1595   // Test that the TRACE_EVENT macros do not deep-copy their string. If they
1596   // do so it may indicate a performance regression, but more-over it would
1597   // make the DEEP_COPY overloads redundant.
1598   std::string name_string("event name");
1599
1600   BeginTrace();
1601   TRACE_EVENT_INSTANT0("category", name_string.c_str(),
1602                        TRACE_EVENT_SCOPE_THREAD);
1603
1604   // Modify the string in place (a wholesale reassignment may leave the old
1605   // string intact on the heap).
1606   name_string[0] = '@';
1607
1608   EndTraceAndFlush();
1609
1610   EXPECT_FALSE(FindTraceEntry(trace_parsed_, "event name"));
1611   EXPECT_TRUE(FindTraceEntry(trace_parsed_, name_string.c_str()));
1612 }
1613
1614 TEST_F(TraceEventTestFixture, DeepCopy) {
1615   static const char kOriginalName1[] = "name1";
1616   static const char kOriginalName2[] = "name2";
1617   static const char kOriginalName3[] = "name3";
1618   std::string name1(kOriginalName1);
1619   std::string name2(kOriginalName2);
1620   std::string name3(kOriginalName3);
1621   std::string arg1("arg1");
1622   std::string arg2("arg2");
1623   std::string val1("val1");
1624   std::string val2("val2");
1625
1626   BeginTrace();
1627   TRACE_EVENT_COPY_INSTANT0("category", name1.c_str(),
1628                             TRACE_EVENT_SCOPE_THREAD);
1629   TRACE_EVENT_COPY_BEGIN1("category", name2.c_str(),
1630                           arg1.c_str(), 5);
1631   TRACE_EVENT_COPY_END2("category", name3.c_str(),
1632                         arg1.c_str(), val1,
1633                         arg2.c_str(), val2);
1634
1635   // As per NormallyNoDeepCopy, modify the strings in place.
1636   name1[0] = name2[0] = name3[0] = arg1[0] = arg2[0] = val1[0] = val2[0] = '@';
1637
1638   EndTraceAndFlush();
1639
1640   EXPECT_FALSE(FindTraceEntry(trace_parsed_, name1.c_str()));
1641   EXPECT_FALSE(FindTraceEntry(trace_parsed_, name2.c_str()));
1642   EXPECT_FALSE(FindTraceEntry(trace_parsed_, name3.c_str()));
1643
1644   const DictionaryValue* entry1 = FindTraceEntry(trace_parsed_, kOriginalName1);
1645   const DictionaryValue* entry2 = FindTraceEntry(trace_parsed_, kOriginalName2);
1646   const DictionaryValue* entry3 = FindTraceEntry(trace_parsed_, kOriginalName3);
1647   ASSERT_TRUE(entry1);
1648   ASSERT_TRUE(entry2);
1649   ASSERT_TRUE(entry3);
1650
1651   int i;
1652   EXPECT_FALSE(entry2->GetInteger("args.@rg1", &i));
1653   EXPECT_TRUE(entry2->GetInteger("args.arg1", &i));
1654   EXPECT_EQ(5, i);
1655
1656   std::string s;
1657   EXPECT_TRUE(entry3->GetString("args.arg1", &s));
1658   EXPECT_EQ("val1", s);
1659   EXPECT_TRUE(entry3->GetString("args.arg2", &s));
1660   EXPECT_EQ("val2", s);
1661 }
1662
1663 // Test that TraceResultBuffer outputs the correct result whether it is added
1664 // in chunks or added all at once.
1665 TEST_F(TraceEventTestFixture, TraceResultBuffer) {
1666   Clear();
1667
1668   trace_buffer_.Start();
1669   trace_buffer_.AddFragment("bla1");
1670   trace_buffer_.AddFragment("bla2");
1671   trace_buffer_.AddFragment("bla3,bla4");
1672   trace_buffer_.Finish();
1673   EXPECT_STREQ(json_output_.json_output.c_str(), "[bla1,bla2,bla3,bla4]");
1674
1675   Clear();
1676
1677   trace_buffer_.Start();
1678   trace_buffer_.AddFragment("bla1,bla2,bla3,bla4");
1679   trace_buffer_.Finish();
1680   EXPECT_STREQ(json_output_.json_output.c_str(), "[bla1,bla2,bla3,bla4]");
1681 }
1682
1683 // Test that trace_event parameters are not evaluated if the tracing
1684 // system is disabled.
1685 TEST_F(TraceEventTestFixture, TracingIsLazy) {
1686   BeginTrace();
1687
1688   int a = 0;
1689   TRACE_EVENT_INSTANT1("category", "test", TRACE_EVENT_SCOPE_THREAD, "a", a++);
1690   EXPECT_EQ(1, a);
1691
1692   TraceLog::GetInstance()->SetDisabled();
1693
1694   TRACE_EVENT_INSTANT1("category", "test", TRACE_EVENT_SCOPE_THREAD, "a", a++);
1695   EXPECT_EQ(1, a);
1696
1697   EndTraceAndFlush();
1698 }
1699
1700 TEST_F(TraceEventTestFixture, TraceEnableDisable) {
1701   TraceLog* trace_log = TraceLog::GetInstance();
1702   TraceConfig tc_inc_all("*", "");
1703   trace_log->SetEnabled(tc_inc_all, TraceLog::RECORDING_MODE);
1704   EXPECT_TRUE(trace_log->IsEnabled());
1705   trace_log->SetDisabled();
1706   EXPECT_FALSE(trace_log->IsEnabled());
1707
1708   trace_log->SetEnabled(tc_inc_all, TraceLog::RECORDING_MODE);
1709   EXPECT_TRUE(trace_log->IsEnabled());
1710   const std::vector<std::string> empty;
1711   trace_log->SetEnabled(TraceConfig(), TraceLog::RECORDING_MODE);
1712   EXPECT_TRUE(trace_log->IsEnabled());
1713   trace_log->SetDisabled();
1714   EXPECT_FALSE(trace_log->IsEnabled());
1715   trace_log->SetDisabled();
1716   EXPECT_FALSE(trace_log->IsEnabled());
1717 }
1718
1719 TEST_F(TraceEventTestFixture, TraceCategoriesAfterNestedEnable) {
1720   TraceLog* trace_log = TraceLog::GetInstance();
1721   trace_log->SetEnabled(TraceConfig("foo,bar", ""), TraceLog::RECORDING_MODE);
1722   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("foo"));
1723   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("bar"));
1724   EXPECT_FALSE(*trace_log->GetCategoryGroupEnabled("baz"));
1725   trace_log->SetEnabled(TraceConfig("foo2", ""), TraceLog::RECORDING_MODE);
1726   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("foo2"));
1727   EXPECT_FALSE(*trace_log->GetCategoryGroupEnabled("baz"));
1728   // The "" becomes the default catergory set when applied.
1729   trace_log->SetEnabled(TraceConfig(), TraceLog::RECORDING_MODE);
1730   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("foo"));
1731   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("baz"));
1732   EXPECT_STREQ(
1733     "-*Debug,-*Test",
1734     trace_log->GetCurrentTraceConfig().ToCategoryFilterString().c_str());
1735   trace_log->SetDisabled();
1736   trace_log->SetDisabled();
1737   trace_log->SetDisabled();
1738   EXPECT_FALSE(*trace_log->GetCategoryGroupEnabled("foo"));
1739   EXPECT_FALSE(*trace_log->GetCategoryGroupEnabled("baz"));
1740
1741   trace_log->SetEnabled(TraceConfig("-foo,-bar", ""), TraceLog::RECORDING_MODE);
1742   EXPECT_FALSE(*trace_log->GetCategoryGroupEnabled("foo"));
1743   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("baz"));
1744   trace_log->SetEnabled(TraceConfig("moo", ""), TraceLog::RECORDING_MODE);
1745   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("baz"));
1746   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("moo"));
1747   EXPECT_FALSE(*trace_log->GetCategoryGroupEnabled("foo"));
1748   EXPECT_STREQ(
1749     "-foo,-bar",
1750     trace_log->GetCurrentTraceConfig().ToCategoryFilterString().c_str());
1751   trace_log->SetDisabled();
1752   trace_log->SetDisabled();
1753
1754   // Make sure disabled categories aren't cleared if we set in the second.
1755   trace_log->SetEnabled(TraceConfig("disabled-by-default-cc,foo", ""),
1756                         TraceLog::RECORDING_MODE);
1757   EXPECT_FALSE(*trace_log->GetCategoryGroupEnabled("bar"));
1758   trace_log->SetEnabled(TraceConfig("disabled-by-default-gpu", ""),
1759                         TraceLog::RECORDING_MODE);
1760   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("disabled-by-default-cc"));
1761   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("disabled-by-default-gpu"));
1762   EXPECT_TRUE(*trace_log->GetCategoryGroupEnabled("bar"));
1763   EXPECT_STREQ(
1764     "disabled-by-default-cc,disabled-by-default-gpu",
1765     trace_log->GetCurrentTraceConfig().ToCategoryFilterString().c_str());
1766   trace_log->SetDisabled();
1767   trace_log->SetDisabled();
1768 }
1769
1770 TEST_F(TraceEventTestFixture, TraceSampling) {
1771   TraceLog::GetInstance()->SetEnabled(
1772     TraceConfig(kRecordAllCategoryFilter, "record-until-full,enable-sampling"),
1773     TraceLog::RECORDING_MODE);
1774
1775   TRACE_EVENT_SET_SAMPLING_STATE_FOR_BUCKET(1, "cc", "Stuff");
1776   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1777   TRACE_EVENT_SET_SAMPLING_STATE_FOR_BUCKET(1, "cc", "Things");
1778   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1779
1780   EndTraceAndFlush();
1781
1782   // Make sure we hit at least once.
1783   EXPECT_TRUE(FindNamePhase("Stuff", "P"));
1784   EXPECT_TRUE(FindNamePhase("Things", "P"));
1785 }
1786
1787 TEST_F(TraceEventTestFixture, TraceSamplingScope) {
1788   TraceLog::GetInstance()->SetEnabled(
1789     TraceConfig(kRecordAllCategoryFilter, "record-until-full,enable-sampling"),
1790     TraceLog::RECORDING_MODE);
1791
1792   TRACE_EVENT_SCOPED_SAMPLING_STATE("AAA", "name");
1793   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1794   {
1795     EXPECT_STREQ(TRACE_EVENT_GET_SAMPLING_STATE(), "AAA");
1796     TRACE_EVENT_SCOPED_SAMPLING_STATE("BBB", "name");
1797     TraceLog::GetInstance()->WaitSamplingEventForTesting();
1798     EXPECT_STREQ(TRACE_EVENT_GET_SAMPLING_STATE(), "BBB");
1799   }
1800   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1801   {
1802     EXPECT_STREQ(TRACE_EVENT_GET_SAMPLING_STATE(), "AAA");
1803     TRACE_EVENT_SCOPED_SAMPLING_STATE("CCC", "name");
1804     TraceLog::GetInstance()->WaitSamplingEventForTesting();
1805     EXPECT_STREQ(TRACE_EVENT_GET_SAMPLING_STATE(), "CCC");
1806   }
1807   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1808   {
1809     EXPECT_STREQ(TRACE_EVENT_GET_SAMPLING_STATE(), "AAA");
1810     TRACE_EVENT_SET_SAMPLING_STATE("DDD", "name");
1811     TraceLog::GetInstance()->WaitSamplingEventForTesting();
1812     EXPECT_STREQ(TRACE_EVENT_GET_SAMPLING_STATE(), "DDD");
1813   }
1814   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1815   EXPECT_STREQ(TRACE_EVENT_GET_SAMPLING_STATE(), "DDD");
1816
1817   EndTraceAndFlush();
1818 }
1819
1820 TEST_F(TraceEventTestFixture, TraceContinuousSampling) {
1821   TraceLog::GetInstance()->SetEnabled(
1822     TraceConfig(kRecordAllCategoryFilter, "record-until-full,enable-sampling"),
1823     TraceLog::MONITORING_MODE);
1824
1825   TRACE_EVENT_SET_SAMPLING_STATE_FOR_BUCKET(1, "category", "AAA");
1826   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1827   TRACE_EVENT_SET_SAMPLING_STATE_FOR_BUCKET(1, "category", "BBB");
1828   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1829
1830   FlushMonitoring();
1831
1832   // Make sure we can get the profiled data.
1833   EXPECT_TRUE(FindNamePhase("AAA", "P"));
1834   EXPECT_TRUE(FindNamePhase("BBB", "P"));
1835
1836   Clear();
1837   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1838
1839   TRACE_EVENT_SET_SAMPLING_STATE_FOR_BUCKET(1, "category", "CCC");
1840   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1841   TRACE_EVENT_SET_SAMPLING_STATE_FOR_BUCKET(1, "category", "DDD");
1842   TraceLog::GetInstance()->WaitSamplingEventForTesting();
1843
1844   FlushMonitoring();
1845
1846   // Make sure the profiled data is accumulated.
1847   EXPECT_TRUE(FindNamePhase("AAA", "P"));
1848   EXPECT_TRUE(FindNamePhase("BBB", "P"));
1849   EXPECT_TRUE(FindNamePhase("CCC", "P"));
1850   EXPECT_TRUE(FindNamePhase("DDD", "P"));
1851
1852   Clear();
1853
1854   TraceLog::GetInstance()->SetDisabled();
1855
1856   // Make sure disabling the continuous sampling thread clears
1857   // the profiled data.
1858   EXPECT_FALSE(FindNamePhase("AAA", "P"));
1859   EXPECT_FALSE(FindNamePhase("BBB", "P"));
1860   EXPECT_FALSE(FindNamePhase("CCC", "P"));
1861   EXPECT_FALSE(FindNamePhase("DDD", "P"));
1862
1863   Clear();
1864 }
1865
1866 class MyData : public ConvertableToTraceFormat {
1867  public:
1868   MyData() {}
1869
1870   void AppendAsTraceFormat(std::string* out) const override {
1871     out->append("{\"foo\":1}");
1872   }
1873
1874  private:
1875   ~MyData() override {}
1876   DISALLOW_COPY_AND_ASSIGN(MyData);
1877 };
1878
1879 TEST_F(TraceEventTestFixture, ConvertableTypes) {
1880   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
1881                                       TraceLog::RECORDING_MODE);
1882
1883   scoped_refptr<ConvertableToTraceFormat> data(new MyData());
1884   scoped_refptr<ConvertableToTraceFormat> data1(new MyData());
1885   scoped_refptr<ConvertableToTraceFormat> data2(new MyData());
1886   TRACE_EVENT1("foo", "bar", "data", data);
1887   TRACE_EVENT2("foo", "baz",
1888                "data1", data1,
1889                "data2", data2);
1890
1891
1892   scoped_refptr<ConvertableToTraceFormat> convertData1(new MyData());
1893   scoped_refptr<ConvertableToTraceFormat> convertData2(new MyData());
1894   TRACE_EVENT2(
1895       "foo",
1896       "string_first",
1897       "str",
1898       "string value 1",
1899       "convert",
1900       convertData1);
1901   TRACE_EVENT2(
1902       "foo",
1903       "string_second",
1904       "convert",
1905       convertData2,
1906       "str",
1907       "string value 2");
1908   EndTraceAndFlush();
1909
1910   // One arg version.
1911   DictionaryValue* dict = FindNamePhase("bar", "X");
1912   ASSERT_TRUE(dict);
1913
1914   const DictionaryValue* args_dict = NULL;
1915   dict->GetDictionary("args", &args_dict);
1916   ASSERT_TRUE(args_dict);
1917
1918   const Value* value = NULL;
1919   const DictionaryValue* convertable_dict = NULL;
1920   EXPECT_TRUE(args_dict->Get("data", &value));
1921   ASSERT_TRUE(value->GetAsDictionary(&convertable_dict));
1922
1923   int foo_val;
1924   EXPECT_TRUE(convertable_dict->GetInteger("foo", &foo_val));
1925   EXPECT_EQ(1, foo_val);
1926
1927   // Two arg version.
1928   dict = FindNamePhase("baz", "X");
1929   ASSERT_TRUE(dict);
1930
1931   args_dict = NULL;
1932   dict->GetDictionary("args", &args_dict);
1933   ASSERT_TRUE(args_dict);
1934
1935   value = NULL;
1936   convertable_dict = NULL;
1937   EXPECT_TRUE(args_dict->Get("data1", &value));
1938   ASSERT_TRUE(value->GetAsDictionary(&convertable_dict));
1939
1940   value = NULL;
1941   convertable_dict = NULL;
1942   EXPECT_TRUE(args_dict->Get("data2", &value));
1943   ASSERT_TRUE(value->GetAsDictionary(&convertable_dict));
1944
1945   // Convertable with other types.
1946   dict = FindNamePhase("string_first", "X");
1947   ASSERT_TRUE(dict);
1948
1949   args_dict = NULL;
1950   dict->GetDictionary("args", &args_dict);
1951   ASSERT_TRUE(args_dict);
1952
1953   std::string str_value;
1954   EXPECT_TRUE(args_dict->GetString("str", &str_value));
1955   EXPECT_STREQ("string value 1", str_value.c_str());
1956
1957   value = NULL;
1958   convertable_dict = NULL;
1959   foo_val = 0;
1960   EXPECT_TRUE(args_dict->Get("convert", &value));
1961   ASSERT_TRUE(value->GetAsDictionary(&convertable_dict));
1962   EXPECT_TRUE(convertable_dict->GetInteger("foo", &foo_val));
1963   EXPECT_EQ(1, foo_val);
1964
1965   dict = FindNamePhase("string_second", "X");
1966   ASSERT_TRUE(dict);
1967
1968   args_dict = NULL;
1969   dict->GetDictionary("args", &args_dict);
1970   ASSERT_TRUE(args_dict);
1971
1972   EXPECT_TRUE(args_dict->GetString("str", &str_value));
1973   EXPECT_STREQ("string value 2", str_value.c_str());
1974
1975   value = NULL;
1976   convertable_dict = NULL;
1977   foo_val = 0;
1978   EXPECT_TRUE(args_dict->Get("convert", &value));
1979   ASSERT_TRUE(value->GetAsDictionary(&convertable_dict));
1980   EXPECT_TRUE(convertable_dict->GetInteger("foo", &foo_val));
1981   EXPECT_EQ(1, foo_val);
1982 }
1983
1984 TEST_F(TraceEventTestFixture, PrimitiveArgs) {
1985   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
1986                                       TraceLog::RECORDING_MODE);
1987
1988   TRACE_EVENT1("foo", "event1", "int_one", 1);
1989   TRACE_EVENT1("foo", "event2", "int_neg_ten", -10);
1990   TRACE_EVENT1("foo", "event3", "float_one", 1.0f);
1991   TRACE_EVENT1("foo", "event4", "float_half", .5f);
1992   TRACE_EVENT1("foo", "event5", "float_neghalf", -.5f);
1993   TRACE_EVENT1("foo", "event6", "float_infinity",
1994       std::numeric_limits<float>::infinity());
1995   TRACE_EVENT1("foo", "event6b", "float_neg_infinity",
1996       -std::numeric_limits<float>::infinity());
1997   TRACE_EVENT1("foo", "event7", "double_nan",
1998       std::numeric_limits<double>::quiet_NaN());
1999   void* p = 0;
2000   TRACE_EVENT1("foo", "event8", "pointer_null", p);
2001   p = reinterpret_cast<void*>(0xbadf00d);
2002   TRACE_EVENT1("foo", "event9", "pointer_badf00d", p);
2003   TRACE_EVENT1("foo", "event10", "bool_true", true);
2004   TRACE_EVENT1("foo", "event11", "bool_false", false);
2005   TRACE_EVENT1("foo", "event12", "time_null",
2006       base::Time());
2007   TRACE_EVENT1("foo", "event13", "time_one",
2008       base::Time::FromInternalValue(1));
2009   TRACE_EVENT1("foo", "event14", "timeticks_null",
2010       base::TimeTicks());
2011   TRACE_EVENT1("foo", "event15", "timeticks_one",
2012       base::TimeTicks::FromInternalValue(1));
2013   EndTraceAndFlush();
2014
2015   const DictionaryValue* args_dict = NULL;
2016   DictionaryValue* dict = NULL;
2017   const Value* value = NULL;
2018   std::string str_value;
2019   int int_value;
2020   double double_value;
2021   bool bool_value;
2022
2023   dict = FindNamePhase("event1", "X");
2024   ASSERT_TRUE(dict);
2025   dict->GetDictionary("args", &args_dict);
2026   ASSERT_TRUE(args_dict);
2027   EXPECT_TRUE(args_dict->GetInteger("int_one", &int_value));
2028   EXPECT_EQ(1, int_value);
2029
2030   dict = FindNamePhase("event2", "X");
2031   ASSERT_TRUE(dict);
2032   dict->GetDictionary("args", &args_dict);
2033   ASSERT_TRUE(args_dict);
2034   EXPECT_TRUE(args_dict->GetInteger("int_neg_ten", &int_value));
2035   EXPECT_EQ(-10, int_value);
2036
2037   // 1f must be serlized to JSON as "1.0" in order to be a double, not an int.
2038   dict = FindNamePhase("event3", "X");
2039   ASSERT_TRUE(dict);
2040   dict->GetDictionary("args", &args_dict);
2041   ASSERT_TRUE(args_dict);
2042   EXPECT_TRUE(args_dict->Get("float_one", &value));
2043   EXPECT_TRUE(value->IsType(Value::TYPE_DOUBLE));
2044   EXPECT_TRUE(value->GetAsDouble(&double_value));
2045   EXPECT_EQ(1, double_value);
2046
2047   // .5f must be serlized to JSON as "0.5".
2048   dict = FindNamePhase("event4", "X");
2049   ASSERT_TRUE(dict);
2050   dict->GetDictionary("args", &args_dict);
2051   ASSERT_TRUE(args_dict);
2052   EXPECT_TRUE(args_dict->Get("float_half", &value));
2053   EXPECT_TRUE(value->IsType(Value::TYPE_DOUBLE));
2054   EXPECT_TRUE(value->GetAsDouble(&double_value));
2055   EXPECT_EQ(0.5, double_value);
2056
2057   // -.5f must be serlized to JSON as "-0.5".
2058   dict = FindNamePhase("event5", "X");
2059   ASSERT_TRUE(dict);
2060   dict->GetDictionary("args", &args_dict);
2061   ASSERT_TRUE(args_dict);
2062   EXPECT_TRUE(args_dict->Get("float_neghalf", &value));
2063   EXPECT_TRUE(value->IsType(Value::TYPE_DOUBLE));
2064   EXPECT_TRUE(value->GetAsDouble(&double_value));
2065   EXPECT_EQ(-0.5, double_value);
2066
2067   // Infinity is serialized to JSON as a string.
2068   dict = FindNamePhase("event6", "X");
2069   ASSERT_TRUE(dict);
2070   dict->GetDictionary("args", &args_dict);
2071   ASSERT_TRUE(args_dict);
2072   EXPECT_TRUE(args_dict->GetString("float_infinity", &str_value));
2073   EXPECT_STREQ("Infinity", str_value.c_str());
2074   dict = FindNamePhase("event6b", "X");
2075   ASSERT_TRUE(dict);
2076   dict->GetDictionary("args", &args_dict);
2077   ASSERT_TRUE(args_dict);
2078   EXPECT_TRUE(args_dict->GetString("float_neg_infinity", &str_value));
2079   EXPECT_STREQ("-Infinity", str_value.c_str());
2080
2081   // NaN is serialized to JSON as a string.
2082   dict = FindNamePhase("event7", "X");
2083   ASSERT_TRUE(dict);
2084   dict->GetDictionary("args", &args_dict);
2085   ASSERT_TRUE(args_dict);
2086   EXPECT_TRUE(args_dict->GetString("double_nan", &str_value));
2087   EXPECT_STREQ("NaN", str_value.c_str());
2088
2089   // NULL pointers should be serialized as "0x0".
2090   dict = FindNamePhase("event8", "X");
2091   ASSERT_TRUE(dict);
2092   dict->GetDictionary("args", &args_dict);
2093   ASSERT_TRUE(args_dict);
2094   EXPECT_TRUE(args_dict->GetString("pointer_null", &str_value));
2095   EXPECT_STREQ("0x0", str_value.c_str());
2096
2097   // Other pointers should be serlized as a hex string.
2098   dict = FindNamePhase("event9", "X");
2099   ASSERT_TRUE(dict);
2100   dict->GetDictionary("args", &args_dict);
2101   ASSERT_TRUE(args_dict);
2102   EXPECT_TRUE(args_dict->GetString("pointer_badf00d", &str_value));
2103   EXPECT_STREQ("0xbadf00d", str_value.c_str());
2104
2105   dict = FindNamePhase("event10", "X");
2106   ASSERT_TRUE(dict);
2107   dict->GetDictionary("args", &args_dict);
2108   ASSERT_TRUE(args_dict);
2109   EXPECT_TRUE(args_dict->GetBoolean("bool_true", &bool_value));
2110   EXPECT_TRUE(bool_value);
2111
2112   dict = FindNamePhase("event11", "X");
2113   ASSERT_TRUE(dict);
2114   dict->GetDictionary("args", &args_dict);
2115   ASSERT_TRUE(args_dict);
2116   EXPECT_TRUE(args_dict->GetBoolean("bool_false", &bool_value));
2117   EXPECT_FALSE(bool_value);
2118
2119   dict = FindNamePhase("event12", "X");
2120   ASSERT_TRUE(dict);
2121   dict->GetDictionary("args", &args_dict);
2122   ASSERT_TRUE(args_dict);
2123   EXPECT_TRUE(args_dict->GetInteger("time_null", &int_value));
2124   EXPECT_EQ(0, int_value);
2125
2126   dict = FindNamePhase("event13", "X");
2127   ASSERT_TRUE(dict);
2128   dict->GetDictionary("args", &args_dict);
2129   ASSERT_TRUE(args_dict);
2130   EXPECT_TRUE(args_dict->GetInteger("time_one", &int_value));
2131   EXPECT_EQ(1, int_value);
2132
2133   dict = FindNamePhase("event14", "X");
2134   ASSERT_TRUE(dict);
2135   dict->GetDictionary("args", &args_dict);
2136   ASSERT_TRUE(args_dict);
2137   EXPECT_TRUE(args_dict->GetInteger("timeticks_null", &int_value));
2138   EXPECT_EQ(0, int_value);
2139
2140   dict = FindNamePhase("event15", "X");
2141   ASSERT_TRUE(dict);
2142   dict->GetDictionary("args", &args_dict);
2143   ASSERT_TRUE(args_dict);
2144   EXPECT_TRUE(args_dict->GetInteger("timeticks_one", &int_value));
2145   EXPECT_EQ(1, int_value);
2146 }
2147
2148 namespace {
2149
2150 bool IsTraceEventArgsWhitelisted(const char* category_group_name,
2151                                  const char* event_name) {
2152   if (MatchPattern(category_group_name, "toplevel") &&
2153       MatchPattern(event_name, "*")) {
2154     return true;
2155   }
2156
2157   return false;
2158 }
2159
2160 }  // namespace
2161
2162 TEST_F(TraceEventTestFixture, ArgsWhitelisting) {
2163   TraceLog::GetInstance()->SetArgumentFilterPredicate(
2164       base::Bind(&IsTraceEventArgsWhitelisted));
2165
2166   TraceLog::GetInstance()->SetEnabled(
2167     TraceConfig(kRecordAllCategoryFilter, "enable-argument-filter"),
2168     TraceLog::RECORDING_MODE);
2169
2170   TRACE_EVENT1("toplevel", "event1", "int_one", 1);
2171   TRACE_EVENT1("whitewashed", "event2", "int_two", 1);
2172   EndTraceAndFlush();
2173
2174   const DictionaryValue* args_dict = NULL;
2175   DictionaryValue* dict = NULL;
2176   int int_value;
2177
2178   dict = FindNamePhase("event1", "X");
2179   ASSERT_TRUE(dict);
2180   dict->GetDictionary("args", &args_dict);
2181   ASSERT_TRUE(args_dict);
2182   EXPECT_TRUE(args_dict->GetInteger("int_one", &int_value));
2183   EXPECT_EQ(1, int_value);
2184
2185   dict = FindNamePhase("event2", "X");
2186   ASSERT_TRUE(dict);
2187   dict->GetDictionary("args", &args_dict);
2188   ASSERT_TRUE(args_dict);
2189   EXPECT_FALSE(args_dict->GetInteger("int_two", &int_value));
2190
2191   std::string args_string;
2192   EXPECT_TRUE(dict->GetString("args", &args_string));
2193   EXPECT_EQ(args_string, "__stripped__");
2194 }
2195
2196 class TraceEventCallbackTest : public TraceEventTestFixture {
2197  public:
2198   void SetUp() override {
2199     TraceEventTestFixture::SetUp();
2200     ASSERT_EQ(NULL, s_instance);
2201     s_instance = this;
2202   }
2203   void TearDown() override {
2204     TraceLog::GetInstance()->SetDisabled();
2205     ASSERT_TRUE(s_instance);
2206     s_instance = NULL;
2207     TraceEventTestFixture::TearDown();
2208   }
2209
2210  protected:
2211   // For TraceEventCallbackAndRecordingX tests.
2212   void VerifyCallbackAndRecordedEvents(size_t expected_callback_count,
2213                                        size_t expected_recorded_count) {
2214     // Callback events.
2215     EXPECT_EQ(expected_callback_count, collected_events_names_.size());
2216     for (size_t i = 0; i < collected_events_names_.size(); ++i) {
2217       EXPECT_EQ("callback", collected_events_categories_[i]);
2218       EXPECT_EQ("yes", collected_events_names_[i]);
2219     }
2220
2221     // Recorded events.
2222     EXPECT_EQ(expected_recorded_count, trace_parsed_.GetSize());
2223     EXPECT_TRUE(FindTraceEntry(trace_parsed_, "recording"));
2224     EXPECT_FALSE(FindTraceEntry(trace_parsed_, "callback"));
2225     EXPECT_TRUE(FindTraceEntry(trace_parsed_, "yes"));
2226     EXPECT_FALSE(FindTraceEntry(trace_parsed_, "no"));
2227   }
2228
2229   void VerifyCollectedEvent(size_t i,
2230                             unsigned phase,
2231                             const std::string& category,
2232                             const std::string& name) {
2233     EXPECT_EQ(phase, collected_events_phases_[i]);
2234     EXPECT_EQ(category, collected_events_categories_[i]);
2235     EXPECT_EQ(name, collected_events_names_[i]);
2236   }
2237
2238   std::vector<std::string> collected_events_categories_;
2239   std::vector<std::string> collected_events_names_;
2240   std::vector<unsigned char> collected_events_phases_;
2241   std::vector<TraceTicks> collected_events_timestamps_;
2242
2243   static TraceEventCallbackTest* s_instance;
2244   static void Callback(TraceTicks timestamp,
2245                        char phase,
2246                        const unsigned char* category_group_enabled,
2247                        const char* name,
2248                        unsigned long long id,
2249                        int num_args,
2250                        const char* const arg_names[],
2251                        const unsigned char arg_types[],
2252                        const unsigned long long arg_values[],
2253                        unsigned char flags) {
2254     s_instance->collected_events_phases_.push_back(phase);
2255     s_instance->collected_events_categories_.push_back(
2256         TraceLog::GetCategoryGroupName(category_group_enabled));
2257     s_instance->collected_events_names_.push_back(name);
2258     s_instance->collected_events_timestamps_.push_back(timestamp);
2259   }
2260 };
2261
2262 TraceEventCallbackTest* TraceEventCallbackTest::s_instance;
2263
2264 TEST_F(TraceEventCallbackTest, TraceEventCallback) {
2265   TRACE_EVENT_INSTANT0("all", "before enable", TRACE_EVENT_SCOPE_THREAD);
2266   TraceLog::GetInstance()->SetEventCallbackEnabled(
2267       TraceConfig(kRecordAllCategoryFilter, ""), Callback);
2268   TRACE_EVENT_INSTANT0("all", "event1", TRACE_EVENT_SCOPE_GLOBAL);
2269   TRACE_EVENT_INSTANT0("all", "event2", TRACE_EVENT_SCOPE_GLOBAL);
2270   {
2271     TRACE_EVENT0("all", "duration");
2272     TRACE_EVENT_INSTANT0("all", "event3", TRACE_EVENT_SCOPE_GLOBAL);
2273   }
2274   TraceLog::GetInstance()->SetEventCallbackDisabled();
2275   TRACE_EVENT_INSTANT0("all", "after callback removed",
2276                        TRACE_EVENT_SCOPE_GLOBAL);
2277   ASSERT_EQ(5u, collected_events_names_.size());
2278   EXPECT_EQ("event1", collected_events_names_[0]);
2279   EXPECT_EQ(TRACE_EVENT_PHASE_INSTANT, collected_events_phases_[0]);
2280   EXPECT_EQ("event2", collected_events_names_[1]);
2281   EXPECT_EQ(TRACE_EVENT_PHASE_INSTANT, collected_events_phases_[1]);
2282   EXPECT_EQ("duration", collected_events_names_[2]);
2283   EXPECT_EQ(TRACE_EVENT_PHASE_BEGIN, collected_events_phases_[2]);
2284   EXPECT_EQ("event3", collected_events_names_[3]);
2285   EXPECT_EQ(TRACE_EVENT_PHASE_INSTANT, collected_events_phases_[3]);
2286   EXPECT_EQ("duration", collected_events_names_[4]);
2287   EXPECT_EQ(TRACE_EVENT_PHASE_END, collected_events_phases_[4]);
2288   for (size_t i = 1; i < collected_events_timestamps_.size(); i++) {
2289     EXPECT_LE(collected_events_timestamps_[i - 1],
2290               collected_events_timestamps_[i]);
2291   }
2292 }
2293
2294 TEST_F(TraceEventCallbackTest, TraceEventCallbackWhileFull) {
2295   TraceLog::GetInstance()->SetEnabled(TraceConfig(kRecordAllCategoryFilter, ""),
2296                                       TraceLog::RECORDING_MODE);
2297   do {
2298     TRACE_EVENT_INSTANT0("all", "badger badger", TRACE_EVENT_SCOPE_GLOBAL);
2299   } while (!TraceLog::GetInstance()->BufferIsFull());
2300   TraceLog::GetInstance()->SetEventCallbackEnabled(
2301       TraceConfig(kRecordAllCategoryFilter, ""), Callback);
2302   TRACE_EVENT_INSTANT0("all", "a snake", TRACE_EVENT_SCOPE_GLOBAL);
2303   TraceLog::GetInstance()->SetEventCallbackDisabled();
2304   ASSERT_EQ(1u, collected_events_names_.size());
2305   EXPECT_EQ("a snake", collected_events_names_[0]);
2306 }
2307
2308 // 1: Enable callback, enable recording, disable callback, disable recording.
2309 TEST_F(TraceEventCallbackTest, TraceEventCallbackAndRecording1) {
2310   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2311   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2312   TraceLog::GetInstance()->SetEventCallbackEnabled(TraceConfig("callback", ""),
2313                                                    Callback);
2314   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2315   TRACE_EVENT_INSTANT0("callback", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2316   TraceLog::GetInstance()->SetEnabled(TraceConfig("recording", ""),
2317                                       TraceLog::RECORDING_MODE);
2318   TRACE_EVENT_INSTANT0("recording", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2319   TRACE_EVENT_INSTANT0("callback", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2320   TraceLog::GetInstance()->SetEventCallbackDisabled();
2321   TRACE_EVENT_INSTANT0("recording", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2322   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2323   EndTraceAndFlush();
2324   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2325   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2326
2327   DropTracedMetadataRecords();
2328   VerifyCallbackAndRecordedEvents(2, 2);
2329 }
2330
2331 // 2: Enable callback, enable recording, disable recording, disable callback.
2332 TEST_F(TraceEventCallbackTest, TraceEventCallbackAndRecording2) {
2333   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2334   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2335   TraceLog::GetInstance()->SetEventCallbackEnabled(TraceConfig("callback", ""),
2336                                                    Callback);
2337   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2338   TRACE_EVENT_INSTANT0("callback", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2339   TraceLog::GetInstance()->SetEnabled(TraceConfig("recording", ""),
2340                                       TraceLog::RECORDING_MODE);
2341   TRACE_EVENT_INSTANT0("recording", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2342   TRACE_EVENT_INSTANT0("callback", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2343   EndTraceAndFlush();
2344   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2345   TRACE_EVENT_INSTANT0("callback", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2346   TraceLog::GetInstance()->SetEventCallbackDisabled();
2347   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2348   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2349
2350   DropTracedMetadataRecords();
2351   VerifyCallbackAndRecordedEvents(3, 1);
2352 }
2353
2354 // 3: Enable recording, enable callback, disable callback, disable recording.
2355 TEST_F(TraceEventCallbackTest, TraceEventCallbackAndRecording3) {
2356   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2357   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2358   TraceLog::GetInstance()->SetEnabled(TraceConfig("recording", ""),
2359                                       TraceLog::RECORDING_MODE);
2360   TRACE_EVENT_INSTANT0("recording", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2361   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2362   TraceLog::GetInstance()->SetEventCallbackEnabled(TraceConfig("callback", ""),
2363                                                    Callback);
2364   TRACE_EVENT_INSTANT0("recording", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2365   TRACE_EVENT_INSTANT0("callback", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2366   TraceLog::GetInstance()->SetEventCallbackDisabled();
2367   TRACE_EVENT_INSTANT0("recording", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2368   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2369   EndTraceAndFlush();
2370   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2371   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2372
2373   DropTracedMetadataRecords();
2374   VerifyCallbackAndRecordedEvents(1, 3);
2375 }
2376
2377 // 4: Enable recording, enable callback, disable recording, disable callback.
2378 TEST_F(TraceEventCallbackTest, TraceEventCallbackAndRecording4) {
2379   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2380   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2381   TraceLog::GetInstance()->SetEnabled(TraceConfig("recording", ""),
2382                                       TraceLog::RECORDING_MODE);
2383   TRACE_EVENT_INSTANT0("recording", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2384   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2385   TraceLog::GetInstance()->SetEventCallbackEnabled(TraceConfig("callback", ""),
2386                                                    Callback);
2387   TRACE_EVENT_INSTANT0("recording", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2388   TRACE_EVENT_INSTANT0("callback", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2389   EndTraceAndFlush();
2390   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2391   TRACE_EVENT_INSTANT0("callback", "yes", TRACE_EVENT_SCOPE_GLOBAL);
2392   TraceLog::GetInstance()->SetEventCallbackDisabled();
2393   TRACE_EVENT_INSTANT0("recording", "no", TRACE_EVENT_SCOPE_GLOBAL);
2394   TRACE_EVENT_INSTANT0("callback", "no", TRACE_EVENT_SCOPE_GLOBAL);
2395
2396   DropTracedMetadataRecords();
2397   VerifyCallbackAndRecordedEvents(2, 2);
2398 }
2399
2400 TEST_F(TraceEventCallbackTest, TraceEventCallbackAndRecordingDuration) {
2401   TraceLog::GetInstance()->SetEventCallbackEnabled(
2402       TraceConfig(kRecordAllCategoryFilter, ""), Callback);
2403   {
2404     TRACE_EVENT0("callback", "duration1");
2405     TraceLog::GetInstance()->SetEnabled(
2406         TraceConfig(kRecordAllCategoryFilter, ""), TraceLog::RECORDING_MODE);
2407     TRACE_EVENT0("callback", "duration2");
2408     EndTraceAndFlush();
2409     TRACE_EVENT0("callback", "duration3");
2410   }
2411   TraceLog::GetInstance()->SetEventCallbackDisabled();
2412
2413   ASSERT_EQ(6u, collected_events_names_.size());
2414   VerifyCollectedEvent(0, TRACE_EVENT_PHASE_BEGIN, "callback", "duration1");
2415   VerifyCollectedEvent(1, TRACE_EVENT_PHASE_BEGIN, "callback", "duration2");
2416   VerifyCollectedEvent(2, TRACE_EVENT_PHASE_BEGIN, "callback", "duration3");
2417   VerifyCollectedEvent(3, TRACE_EVENT_PHASE_END, "callback", "duration3");
2418   VerifyCollectedEvent(4, TRACE_EVENT_PHASE_END, "callback", "duration2");
2419   VerifyCollectedEvent(5, TRACE_EVENT_PHASE_END, "callback", "duration1");
2420 }
2421
2422 TEST_F(TraceEventTestFixture, TraceBufferVectorReportFull) {
2423   TraceLog* trace_log = TraceLog::GetInstance();
2424   trace_log->SetEnabled(
2425       TraceConfig(kRecordAllCategoryFilter, ""), TraceLog::RECORDING_MODE);
2426   trace_log->logged_events_.reset(
2427       trace_log->CreateTraceBufferVectorOfSize(100));
2428   do {
2429     TRACE_EVENT_BEGIN_WITH_ID_TID_AND_TIMESTAMP0(
2430         "all", "with_timestamp", 0, 0, TraceTicks::Now().ToInternalValue());
2431     TRACE_EVENT_END_WITH_ID_TID_AND_TIMESTAMP0(
2432         "all", "with_timestamp", 0, 0, TraceTicks::Now().ToInternalValue());
2433   } while (!trace_log->BufferIsFull());
2434
2435   EndTraceAndFlush();
2436
2437   const DictionaryValue* trace_full_metadata = NULL;
2438
2439   trace_full_metadata = FindTraceEntry(trace_parsed_,
2440                                        "overflowed_at_ts");
2441   std::string phase;
2442   double buffer_limit_reached_timestamp = 0;
2443
2444   EXPECT_TRUE(trace_full_metadata);
2445   EXPECT_TRUE(trace_full_metadata->GetString("ph", &phase));
2446   EXPECT_EQ("M", phase);
2447   EXPECT_TRUE(trace_full_metadata->GetDouble(
2448       "args.overflowed_at_ts", &buffer_limit_reached_timestamp));
2449   EXPECT_DOUBLE_EQ(
2450       static_cast<double>(
2451           trace_log->buffer_limit_reached_timestamp_.ToInternalValue()),
2452       buffer_limit_reached_timestamp);
2453
2454   // Test that buffer_limit_reached_timestamp's value is between the timestamp
2455   // of the last trace event and current time.
2456   DropTracedMetadataRecords();
2457   const DictionaryValue* last_trace_event = NULL;
2458   double last_trace_event_timestamp = 0;
2459   EXPECT_TRUE(trace_parsed_.GetDictionary(trace_parsed_.GetSize() - 1,
2460                                           &last_trace_event));
2461   EXPECT_TRUE(last_trace_event->GetDouble("ts", &last_trace_event_timestamp));
2462   EXPECT_LE(last_trace_event_timestamp, buffer_limit_reached_timestamp);
2463   EXPECT_LE(buffer_limit_reached_timestamp,
2464             trace_log->OffsetNow().ToInternalValue());
2465 }
2466
2467 TEST_F(TraceEventTestFixture, TraceBufferRingBufferGetReturnChunk) {
2468   TraceLog::GetInstance()->SetEnabled(
2469       TraceConfig(kRecordAllCategoryFilter, RECORD_CONTINUOUSLY),
2470       TraceLog::RECORDING_MODE);
2471   TraceBuffer* buffer = TraceLog::GetInstance()->trace_buffer();
2472   size_t capacity = buffer->Capacity();
2473   size_t num_chunks = capacity / TraceBufferChunk::kTraceBufferChunkSize;
2474   uint32 last_seq = 0;
2475   size_t chunk_index;
2476   EXPECT_EQ(0u, buffer->Size());
2477
2478   scoped_ptr<TraceBufferChunk*[]> chunks(new TraceBufferChunk*[num_chunks]);
2479   for (size_t i = 0; i < num_chunks; ++i) {
2480     chunks[i] = buffer->GetChunk(&chunk_index).release();
2481     EXPECT_TRUE(chunks[i]);
2482     EXPECT_EQ(i, chunk_index);
2483     EXPECT_GT(chunks[i]->seq(), last_seq);
2484     EXPECT_EQ((i + 1) * TraceBufferChunk::kTraceBufferChunkSize,
2485               buffer->Size());
2486     last_seq = chunks[i]->seq();
2487   }
2488
2489   // Ring buffer is never full.
2490   EXPECT_FALSE(buffer->IsFull());
2491
2492   // Return all chunks in original order.
2493   for (size_t i = 0; i < num_chunks; ++i)
2494     buffer->ReturnChunk(i, scoped_ptr<TraceBufferChunk>(chunks[i]));
2495
2496   // Should recycle the chunks in the returned order.
2497   for (size_t i = 0; i < num_chunks; ++i) {
2498     chunks[i] = buffer->GetChunk(&chunk_index).release();
2499     EXPECT_TRUE(chunks[i]);
2500     EXPECT_EQ(i, chunk_index);
2501     EXPECT_GT(chunks[i]->seq(), last_seq);
2502     last_seq = chunks[i]->seq();
2503   }
2504
2505   // Return all chunks in reverse order.
2506   for (size_t i = 0; i < num_chunks; ++i) {
2507     buffer->ReturnChunk(
2508         num_chunks - i - 1,
2509         scoped_ptr<TraceBufferChunk>(chunks[num_chunks - i - 1]));
2510   }
2511
2512   // Should recycle the chunks in the returned order.
2513   for (size_t i = 0; i < num_chunks; ++i) {
2514     chunks[i] = buffer->GetChunk(&chunk_index).release();
2515     EXPECT_TRUE(chunks[i]);
2516     EXPECT_EQ(num_chunks - i - 1, chunk_index);
2517     EXPECT_GT(chunks[i]->seq(), last_seq);
2518     last_seq = chunks[i]->seq();
2519   }
2520
2521   for (size_t i = 0; i < num_chunks; ++i)
2522     buffer->ReturnChunk(i, scoped_ptr<TraceBufferChunk>(chunks[i]));
2523
2524   TraceLog::GetInstance()->SetDisabled();
2525 }
2526
2527 TEST_F(TraceEventTestFixture, TraceBufferRingBufferHalfIteration) {
2528   TraceLog::GetInstance()->SetEnabled(
2529       TraceConfig(kRecordAllCategoryFilter, RECORD_CONTINUOUSLY),
2530       TraceLog::RECORDING_MODE);
2531   TraceBuffer* buffer = TraceLog::GetInstance()->trace_buffer();
2532   size_t capacity = buffer->Capacity();
2533   size_t num_chunks = capacity / TraceBufferChunk::kTraceBufferChunkSize;
2534   size_t chunk_index;
2535   EXPECT_EQ(0u, buffer->Size());
2536   EXPECT_FALSE(buffer->NextChunk());
2537
2538   size_t half_chunks = num_chunks / 2;
2539   scoped_ptr<TraceBufferChunk*[]> chunks(new TraceBufferChunk*[half_chunks]);
2540
2541   for (size_t i = 0; i < half_chunks; ++i) {
2542     chunks[i] = buffer->GetChunk(&chunk_index).release();
2543     EXPECT_TRUE(chunks[i]);
2544     EXPECT_EQ(i, chunk_index);
2545   }
2546   for (size_t i = 0; i < half_chunks; ++i)
2547     buffer->ReturnChunk(i, scoped_ptr<TraceBufferChunk>(chunks[i]));
2548
2549   for (size_t i = 0; i < half_chunks; ++i)
2550     EXPECT_EQ(chunks[i], buffer->NextChunk());
2551   EXPECT_FALSE(buffer->NextChunk());
2552   TraceLog::GetInstance()->SetDisabled();
2553 }
2554
2555 TEST_F(TraceEventTestFixture, TraceBufferRingBufferFullIteration) {
2556   TraceLog::GetInstance()->SetEnabled(
2557       TraceConfig(kRecordAllCategoryFilter, RECORD_CONTINUOUSLY),
2558       TraceLog::RECORDING_MODE);
2559   TraceBuffer* buffer = TraceLog::GetInstance()->trace_buffer();
2560   size_t capacity = buffer->Capacity();
2561   size_t num_chunks = capacity / TraceBufferChunk::kTraceBufferChunkSize;
2562   size_t chunk_index;
2563   EXPECT_EQ(0u, buffer->Size());
2564   EXPECT_FALSE(buffer->NextChunk());
2565
2566   scoped_ptr<TraceBufferChunk*[]> chunks(new TraceBufferChunk*[num_chunks]);
2567
2568   for (size_t i = 0; i < num_chunks; ++i) {
2569     chunks[i] = buffer->GetChunk(&chunk_index).release();
2570     EXPECT_TRUE(chunks[i]);
2571     EXPECT_EQ(i, chunk_index);
2572   }
2573   for (size_t i = 0; i < num_chunks; ++i)
2574     buffer->ReturnChunk(i, scoped_ptr<TraceBufferChunk>(chunks[i]));
2575
2576   for (size_t i = 0; i < num_chunks; ++i)
2577     EXPECT_TRUE(chunks[i] == buffer->NextChunk());
2578   EXPECT_FALSE(buffer->NextChunk());
2579   TraceLog::GetInstance()->SetDisabled();
2580 }
2581
2582 TEST_F(TraceEventTestFixture, TraceRecordAsMuchAsPossibleMode) {
2583   TraceLog::GetInstance()->SetEnabled(
2584     TraceConfig(kRecordAllCategoryFilter, RECORD_AS_MUCH_AS_POSSIBLE),
2585     TraceLog::RECORDING_MODE);
2586   TraceBuffer* buffer = TraceLog::GetInstance()->trace_buffer();
2587   EXPECT_EQ(512000000UL, buffer->Capacity());
2588   TraceLog::GetInstance()->SetDisabled();
2589 }
2590
2591 void BlockUntilStopped(WaitableEvent* task_start_event,
2592                        WaitableEvent* task_stop_event) {
2593   task_start_event->Signal();
2594   task_stop_event->Wait();
2595 }
2596
2597 TEST_F(TraceEventTestFixture, SetCurrentThreadBlocksMessageLoopBeforeTracing) {
2598   BeginTrace();
2599
2600   Thread thread("1");
2601   WaitableEvent task_complete_event(false, false);
2602   thread.Start();
2603   thread.task_runner()->PostTask(
2604       FROM_HERE, Bind(&TraceLog::SetCurrentThreadBlocksMessageLoop,
2605                       Unretained(TraceLog::GetInstance())));
2606
2607   thread.task_runner()->PostTask(
2608       FROM_HERE, Bind(&TraceWithAllMacroVariants, &task_complete_event));
2609   task_complete_event.Wait();
2610
2611   WaitableEvent task_start_event(false, false);
2612   WaitableEvent task_stop_event(false, false);
2613   thread.task_runner()->PostTask(
2614       FROM_HERE, Bind(&BlockUntilStopped, &task_start_event, &task_stop_event));
2615   task_start_event.Wait();
2616
2617   EndTraceAndFlush();
2618   ValidateAllTraceMacrosCreatedData(trace_parsed_);
2619
2620   task_stop_event.Signal();
2621   thread.Stop();
2622 }
2623
2624 TEST_F(TraceEventTestFixture, ConvertTraceConfigToInternalOptions) {
2625   TraceLog* trace_log = TraceLog::GetInstance();
2626   EXPECT_EQ(TraceLog::kInternalRecordUntilFull,
2627             trace_log->GetInternalOptionsFromTraceConfig(
2628                 TraceConfig(kRecordAllCategoryFilter, RECORD_UNTIL_FULL)));
2629
2630   EXPECT_EQ(TraceLog::kInternalRecordContinuously,
2631             trace_log->GetInternalOptionsFromTraceConfig(
2632                 TraceConfig(kRecordAllCategoryFilter, RECORD_CONTINUOUSLY)));
2633
2634   EXPECT_EQ(TraceLog::kInternalEchoToConsole,
2635             trace_log->GetInternalOptionsFromTraceConfig(
2636                 TraceConfig(kRecordAllCategoryFilter, ECHO_TO_CONSOLE)));
2637
2638   EXPECT_EQ(
2639       TraceLog::kInternalRecordUntilFull | TraceLog::kInternalEnableSampling,
2640       trace_log->GetInternalOptionsFromTraceConfig(
2641           TraceConfig(kRecordAllCategoryFilter,
2642                       "record-until-full,enable-sampling")));
2643
2644   EXPECT_EQ(
2645       TraceLog::kInternalRecordContinuously | TraceLog::kInternalEnableSampling,
2646       trace_log->GetInternalOptionsFromTraceConfig(
2647           TraceConfig(kRecordAllCategoryFilter,
2648                       "record-continuously,enable-sampling")));
2649
2650   EXPECT_EQ(
2651       TraceLog::kInternalEchoToConsole | TraceLog::kInternalEnableSampling,
2652       trace_log->GetInternalOptionsFromTraceConfig(
2653           TraceConfig(kRecordAllCategoryFilter,
2654                       "trace-to-console,enable-sampling")));
2655
2656   EXPECT_EQ(
2657       TraceLog::kInternalEchoToConsole | TraceLog::kInternalEnableSampling,
2658       trace_log->GetInternalOptionsFromTraceConfig(
2659           TraceConfig("*",
2660                       "trace-to-console,enable-sampling,enable-systrace")));
2661 }
2662
2663 void SetBlockingFlagAndBlockUntilStopped(WaitableEvent* task_start_event,
2664                                          WaitableEvent* task_stop_event) {
2665   TraceLog::GetInstance()->SetCurrentThreadBlocksMessageLoop();
2666   BlockUntilStopped(task_start_event, task_stop_event);
2667 }
2668
2669 TEST_F(TraceEventTestFixture, SetCurrentThreadBlocksMessageLoopAfterTracing) {
2670   BeginTrace();
2671
2672   Thread thread("1");
2673   WaitableEvent task_complete_event(false, false);
2674   thread.Start();
2675
2676   thread.task_runner()->PostTask(
2677       FROM_HERE, Bind(&TraceWithAllMacroVariants, &task_complete_event));
2678   task_complete_event.Wait();
2679
2680   WaitableEvent task_start_event(false, false);
2681   WaitableEvent task_stop_event(false, false);
2682   thread.task_runner()->PostTask(
2683       FROM_HERE, Bind(&SetBlockingFlagAndBlockUntilStopped, &task_start_event,
2684                       &task_stop_event));
2685   task_start_event.Wait();
2686
2687   EndTraceAndFlush();
2688   ValidateAllTraceMacrosCreatedData(trace_parsed_);
2689
2690   task_stop_event.Signal();
2691   thread.Stop();
2692 }
2693
2694 TEST_F(TraceEventTestFixture, ThreadOnceBlocking) {
2695   BeginTrace();
2696
2697   Thread thread("1");
2698   WaitableEvent task_complete_event(false, false);
2699   thread.Start();
2700
2701   thread.task_runner()->PostTask(
2702       FROM_HERE, Bind(&TraceWithAllMacroVariants, &task_complete_event));
2703   task_complete_event.Wait();
2704   task_complete_event.Reset();
2705
2706   WaitableEvent task_start_event(false, false);
2707   WaitableEvent task_stop_event(false, false);
2708   thread.task_runner()->PostTask(
2709       FROM_HERE, Bind(&BlockUntilStopped, &task_start_event, &task_stop_event));
2710   task_start_event.Wait();
2711
2712   // The thread will timeout in this flush.
2713   EndTraceAndFlushInThreadWithMessageLoop();
2714   Clear();
2715
2716   // Let the thread's message loop continue to spin.
2717   task_stop_event.Signal();
2718
2719   // The following sequence ensures that the FlushCurrentThread task has been
2720   // executed in the thread before continuing.
2721   task_start_event.Reset();
2722   task_stop_event.Reset();
2723   thread.task_runner()->PostTask(
2724       FROM_HERE, Bind(&BlockUntilStopped, &task_start_event, &task_stop_event));
2725   task_start_event.Wait();
2726   task_stop_event.Signal();
2727   Clear();
2728
2729   // TraceLog should discover the generation mismatch and recover the thread
2730   // local buffer for the thread without any error.
2731   BeginTrace();
2732   thread.task_runner()->PostTask(
2733       FROM_HERE, Bind(&TraceWithAllMacroVariants, &task_complete_event));
2734   task_complete_event.Wait();
2735   task_complete_event.Reset();
2736   EndTraceAndFlushInThreadWithMessageLoop();
2737   ValidateAllTraceMacrosCreatedData(trace_parsed_);
2738 }
2739
2740 std::string* g_log_buffer = NULL;
2741 bool MockLogMessageHandler(int, const char*, int, size_t,
2742                            const std::string& str) {
2743   if (!g_log_buffer)
2744     g_log_buffer = new std::string();
2745   g_log_buffer->append(str);
2746   return false;
2747 }
2748
2749 TEST_F(TraceEventTestFixture, EchoToConsole) {
2750   logging::LogMessageHandlerFunction old_log_message_handler =
2751       logging::GetLogMessageHandler();
2752   logging::SetLogMessageHandler(MockLogMessageHandler);
2753
2754   TraceLog::GetInstance()->SetEnabled(
2755       TraceConfig(kRecordAllCategoryFilter, ECHO_TO_CONSOLE),
2756       TraceLog::RECORDING_MODE);
2757   TRACE_EVENT_BEGIN0("a", "begin_end");
2758   {
2759     TRACE_EVENT0("b", "duration");
2760     TRACE_EVENT0("b1", "duration1");
2761   }
2762   TRACE_EVENT_INSTANT0("c", "instant", TRACE_EVENT_SCOPE_GLOBAL);
2763   TRACE_EVENT_END0("a", "begin_end");
2764
2765   EXPECT_NE(std::string::npos, g_log_buffer->find("begin_end[a]\x1b"));
2766   EXPECT_NE(std::string::npos, g_log_buffer->find("| duration[b]\x1b"));
2767   EXPECT_NE(std::string::npos, g_log_buffer->find("| | duration1[b1]\x1b"));
2768   EXPECT_NE(std::string::npos, g_log_buffer->find("| | duration1[b1] ("));
2769   EXPECT_NE(std::string::npos, g_log_buffer->find("| duration[b] ("));
2770   EXPECT_NE(std::string::npos, g_log_buffer->find("| instant[c]\x1b"));
2771   EXPECT_NE(std::string::npos, g_log_buffer->find("begin_end[a] ("));
2772
2773   EndTraceAndFlush();
2774   delete g_log_buffer;
2775   logging::SetLogMessageHandler(old_log_message_handler);
2776   g_log_buffer = NULL;
2777 }
2778
2779 bool LogMessageHandlerWithTraceEvent(int, const char*, int, size_t,
2780                                      const std::string&) {
2781   TRACE_EVENT0("log", "trace_event");
2782   return false;
2783 }
2784
2785 TEST_F(TraceEventTestFixture, EchoToConsoleTraceEventRecursion) {
2786   logging::LogMessageHandlerFunction old_log_message_handler =
2787       logging::GetLogMessageHandler();
2788   logging::SetLogMessageHandler(LogMessageHandlerWithTraceEvent);
2789
2790   TraceLog::GetInstance()->SetEnabled(
2791       TraceConfig(kRecordAllCategoryFilter, ECHO_TO_CONSOLE),
2792       TraceLog::RECORDING_MODE);
2793   {
2794     // This should not cause deadlock or infinite recursion.
2795     TRACE_EVENT0("b", "duration");
2796   }
2797
2798   EndTraceAndFlush();
2799   logging::SetLogMessageHandler(old_log_message_handler);
2800 }
2801
2802 TEST_F(TraceEventTestFixture, TimeOffset) {
2803   BeginTrace();
2804   // Let TraceLog timer start from 0.
2805   TimeDelta time_offset = TraceTicks::Now() - TraceTicks();
2806   TraceLog::GetInstance()->SetTimeOffset(time_offset);
2807
2808   {
2809     TRACE_EVENT0("all", "duration1");
2810     TRACE_EVENT0("all", "duration2");
2811   }
2812   TRACE_EVENT_BEGIN_WITH_ID_TID_AND_TIMESTAMP0(
2813       "all", "with_timestamp", 0, 0, TraceTicks::Now().ToInternalValue());
2814   TRACE_EVENT_END_WITH_ID_TID_AND_TIMESTAMP0(
2815       "all", "with_timestamp", 0, 0, TraceTicks::Now().ToInternalValue());
2816
2817   EndTraceAndFlush();
2818   DropTracedMetadataRecords();
2819
2820   double end_time = static_cast<double>(
2821       (TraceTicks::Now() - time_offset).ToInternalValue());
2822   double last_timestamp = 0;
2823   for (size_t i = 0; i < trace_parsed_.GetSize(); ++i) {
2824     const DictionaryValue* item;
2825     EXPECT_TRUE(trace_parsed_.GetDictionary(i, &item));
2826     double timestamp;
2827     EXPECT_TRUE(item->GetDouble("ts", &timestamp));
2828     EXPECT_GE(timestamp, last_timestamp);
2829     EXPECT_LE(timestamp, end_time);
2830     last_timestamp = timestamp;
2831   }
2832 }
2833
2834 TEST_F(TraceEventTestFixture, ConfigureSyntheticDelays) {
2835   BeginSpecificTrace("DELAY(test.Delay;0.05)");
2836
2837   base::TimeTicks start = base::TimeTicks::Now();
2838   {
2839     TRACE_EVENT_SYNTHETIC_DELAY("test.Delay");
2840   }
2841   base::TimeDelta duration = base::TimeTicks::Now() - start;
2842   EXPECT_GE(duration.InMilliseconds(), 50);
2843
2844   EndTraceAndFlush();
2845 }
2846
2847 TEST_F(TraceEventTestFixture, BadSyntheticDelayConfigurations) {
2848   const char* const filters[] = {
2849     "",
2850     "DELAY(",
2851     "DELAY(;",
2852     "DELAY(;)",
2853     "DELAY(test.Delay)",
2854     "DELAY(test.Delay;)"
2855   };
2856   for (size_t i = 0; i < arraysize(filters); i++) {
2857     BeginSpecificTrace(filters[i]);
2858     EndTraceAndFlush();
2859     TraceConfig trace_config = TraceLog::GetInstance()->GetCurrentTraceConfig();
2860     EXPECT_EQ(0u, trace_config.GetSyntheticDelayValues().size());
2861   }
2862 }
2863
2864 TEST_F(TraceEventTestFixture, SyntheticDelayConfigurationMerging) {
2865   TraceConfig config1("DELAY(test.Delay1;16)", "");
2866   TraceConfig config2("DELAY(test.Delay2;32)", "");
2867   config1.Merge(config2);
2868   EXPECT_EQ(2u, config1.GetSyntheticDelayValues().size());
2869 }
2870
2871 TEST_F(TraceEventTestFixture, SyntheticDelayConfigurationToString) {
2872   const char filter[] = "DELAY(test.Delay;16;oneshot)";
2873   TraceConfig config(filter, "");
2874   EXPECT_EQ(filter, config.ToCategoryFilterString());
2875 }
2876
2877 }  // namespace trace_event
2878 }  // namespace base