// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <memory>

#include "base/base64.h"
#include "base/json/json_reader.h"
#include "base/json/json_writer.h"
#include "base/trace_event/memory_dump_manager.h"
#include "base/trace_event/trace_config.h"
#include "base/values.h"
#include "content/browser/tracing/tracing_ui.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace content {

class TracingUITest : public testing::Test {
 public:
  TracingUITest() {}
};

std::string GetConfig() {
  std::unique_ptr<base::DictionaryValue> dict(new base::DictionaryValue());
  std::unique_ptr<base::Value> filter1(
      new base::Value(base::trace_event::MemoryDumpManager::kTraceCategory));
  std::unique_ptr<base::Value> filter2(new base::Value("filter2"));
  std::unique_ptr<base::ListValue> included(new base::ListValue);
  included->Append(std::move(filter1));
  std::unique_ptr<base::ListValue> excluded(new base::ListValue);
  excluded->Append(std::move(filter2));

  dict->SetList("included_categories", std::move(included));
  dict->SetList("excluded_categories", std::move(excluded));
  dict->SetString("record_mode", "record-continuously");
  dict->SetBoolean("enable_systrace", true);
  dict->SetString("stream_format", "protobuf");

  std::unique_ptr<base::DictionaryValue> memory_config(
      new base::DictionaryValue());
  std::unique_ptr<base::DictionaryValue> trigger(new base::DictionaryValue());
  trigger->SetString("mode", "detailed");
  trigger->SetInteger("periodic_interval_ms", 10000);
  std::unique_ptr<base::ListValue> triggers(new base::ListValue);
  triggers->Append(std::move(trigger));
  memory_config->SetList("triggers", std::move(triggers));
  dict->SetDictionary("memory_dump_config", std::move(memory_config));

  std::string results;
  if (!base::JSONWriter::Write(*dict.get(), &results))
    return "";

  std::string data;
  base::Base64Encode(results, &data);
  return data;
}

TEST_F(TracingUITest, ConfigParsing) {
  base::trace_event::TraceConfig config;
  std::string stream_format;
  ASSERT_TRUE(TracingUI::GetTracingOptions(GetConfig(), config, stream_format));
  EXPECT_EQ(config.GetTraceRecordMode(),
            base::trace_event::RECORD_CONTINUOUSLY);
  std::string expected(base::trace_event::MemoryDumpManager::kTraceCategory);
  expected += ",-filter2";
  EXPECT_EQ(config.ToCategoryFilterString(), expected);
  EXPECT_EQ(stream_format, "protobuf");
  EXPECT_TRUE(config.IsSystraceEnabled());

  ASSERT_EQ(config.memory_dump_config().triggers.size(), 1u);
  EXPECT_EQ(config.memory_dump_config().triggers[0].min_time_between_dumps_ms,
            10000u);
  EXPECT_EQ(config.memory_dump_config().triggers[0].level_of_detail,
            base::trace_event::MemoryDumpLevelOfDetail::DETAILED);
}

}  // namespace content