// Copyright (c) 2012 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 "content/browser/tracing/trace_subscriber_stdio.h"

#include "base/file_util.h"
#include "base/files/scoped_temp_dir.h"
#include "base/threading/sequenced_worker_pool.h"
#include "content/public/browser/browser_thread.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace content {

class TraceSubscriberStdioTest : public ::testing::Test {};

TEST_F(TraceSubscriberStdioTest, CanWriteArray) {
  base::ScopedTempDir trace_dir;
  ASSERT_TRUE(trace_dir.CreateUniqueTempDir());
  base::FilePath trace_file(trace_dir.path().AppendASCII("trace.txt"));
  {
    TraceSubscriberStdio subscriber(trace_file,
                                    TraceSubscriberStdio::FILE_TYPE_ARRAY,
                                    false);

    std::string foo("foo");
    subscriber.OnTraceDataCollected(
        make_scoped_refptr(base::RefCountedString::TakeString(&foo)));

    std::string bar("bar");
    subscriber.OnTraceDataCollected(
        make_scoped_refptr(base::RefCountedString::TakeString(&bar)));

    subscriber.OnEndTracingComplete();
  }
  BrowserThread::GetBlockingPool()->FlushForTesting();
  std::string result;
  EXPECT_TRUE(base::ReadFileToString(trace_file, &result));
  EXPECT_EQ("[foo,bar]", result);
}

TEST_F(TraceSubscriberStdioTest, CanWritePropertyList) {
  base::ScopedTempDir trace_dir;
  ASSERT_TRUE(trace_dir.CreateUniqueTempDir());
  base::FilePath trace_file(trace_dir.path().AppendASCII("trace.txt"));
  {
    TraceSubscriberStdio subscriber(
        trace_file,
        TraceSubscriberStdio::FILE_TYPE_PROPERTY_LIST,
        false);

    std::string foo("foo");
    subscriber.OnTraceDataCollected(
        make_scoped_refptr(base::RefCountedString::TakeString(&foo)));

    std::string bar("bar");
    subscriber.OnTraceDataCollected(
        make_scoped_refptr(base::RefCountedString::TakeString(&bar)));

    subscriber.OnEndTracingComplete();
  }
  BrowserThread::GetBlockingPool()->FlushForTesting();
  std::string result;
  EXPECT_TRUE(base::ReadFileToString(trace_file, &result));
  EXPECT_EQ("{\"traceEvents\":[foo,bar]}", result);
}

TEST_F(TraceSubscriberStdioTest, CanWriteSystemDataFirst) {
  base::ScopedTempDir trace_dir;
  ASSERT_TRUE(trace_dir.CreateUniqueTempDir());
  base::FilePath trace_file(trace_dir.path().AppendASCII("trace.txt"));
  {
    TraceSubscriberStdio subscriber(
        trace_file,
        TraceSubscriberStdio::FILE_TYPE_PROPERTY_LIST,
        true);

    std::string foo("foo");
    subscriber.OnTraceDataCollected(
        make_scoped_refptr(base::RefCountedString::TakeString(&foo)));

    std::string bar("bar");
    subscriber.OnTraceDataCollected(
        make_scoped_refptr(base::RefCountedString::TakeString(&bar)));

    std::string systemTrace("event1\nev\"ent\"2\n");
    subscriber.OnEndSystemTracing(
        make_scoped_refptr(base::RefCountedString::TakeString(&systemTrace)));
    subscriber.OnEndTracingComplete();
  }
  BrowserThread::GetBlockingPool()->FlushForTesting();
  std::string result;
  EXPECT_TRUE(base::ReadFileToString(trace_file, &result));
  EXPECT_EQ(
    "{\"traceEvents\":[foo,bar],\""
    "systemTraceEvents\":\"event1\\nev\\\"ent\\\"2\\n\"}",
    result);
}

TEST_F(TraceSubscriberStdioTest, CanWriteSystemDataLast) {
  base::ScopedTempDir trace_dir;
  ASSERT_TRUE(trace_dir.CreateUniqueTempDir());
  base::FilePath trace_file(trace_dir.path().AppendASCII("trace.txt"));
  {
    TraceSubscriberStdio subscriber(
        trace_file,
        TraceSubscriberStdio::FILE_TYPE_PROPERTY_LIST,
        true);

    std::string foo("foo");
    subscriber.OnTraceDataCollected(
        make_scoped_refptr(base::RefCountedString::TakeString(&foo)));

    std::string bar("bar");
    subscriber.OnTraceDataCollected(
        make_scoped_refptr(base::RefCountedString::TakeString(&bar)));

    std::string systemTrace("event1\nev\"ent\"2\n");
    subscriber.OnEndTracingComplete();
    subscriber.OnEndSystemTracing(
        make_scoped_refptr(base::RefCountedString::TakeString(&systemTrace)));
  }
  BrowserThread::GetBlockingPool()->FlushForTesting();
  std::string result;
  EXPECT_TRUE(base::ReadFileToString(trace_file, &result));
  EXPECT_EQ(
    "{\"traceEvents\":[foo,bar],\""
    "systemTraceEvents\":\"event1\\nev\\\"ent\\\"2\\n\"}",
    result);
}

}  // namespace content