// Copyright 2015 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 "components/tracing/child/child_trace_message_filter.h"

#include <memory>
#include "base/run_loop.h"

#include "base/memory/ref_counted.h"
#include "base/metrics/histogram_macros.h"
#include "base/test/scoped_task_environment.h"
#include "base/threading/thread_task_runner_handle.h"
#include "components/tracing/common/tracing_messages.h"
#include "ipc/ipc_message.h"
#include "ipc/ipc_sender.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace tracing {

class FakeSender : public IPC::Sender {
 public:
  FakeSender() {}

  ~FakeSender() override {}

  bool Send(IPC::Message* msg) override {
    last_message_.reset(msg);
    return true;
  }

  std::unique_ptr<IPC::Message> last_message_;
};

class ChildTraceMessageFilterTest : public testing::Test {
 public:
  ChildTraceMessageFilterTest() {
    message_filter_ = new tracing::ChildTraceMessageFilter(
        base::ThreadTaskRunnerHandle::Get().get());
    message_filter_->SetSenderForTesting(&fake_sender_);
  }

  void OnSetUMACallback(const std::string& histogram,
                        int low,
                        int high,
                        bool repeat) {
    fake_sender_.last_message_.reset();
    message_filter_->OnSetUMACallback(histogram, low, high, repeat);
  }

  base::test::ScopedTaskEnvironment task_environment_;
  FakeSender fake_sender_;
  scoped_refptr<tracing::ChildTraceMessageFilter> message_filter_;
};

TEST_F(ChildTraceMessageFilterTest, TestHistogramDoesNotTrigger) {
  LOCAL_HISTOGRAM_COUNTS("foo1", 10);

  OnSetUMACallback("foo1", 20000, 25000, true);

  base::RunLoop().RunUntilIdle();

  EXPECT_FALSE(fake_sender_.last_message_);
}

TEST_F(ChildTraceMessageFilterTest, TestHistogramTriggers) {
  LOCAL_HISTOGRAM_COUNTS("foo2", 2);

  OnSetUMACallback("foo2", 1, 3, true);

  base::RunLoop().RunUntilIdle();

  EXPECT_TRUE(fake_sender_.last_message_);
  EXPECT_EQ(fake_sender_.last_message_->type(),
            static_cast<uint32_t>(TracingHostMsg_TriggerBackgroundTrace::ID));
}

TEST_F(ChildTraceMessageFilterTest, TestHistogramAborts) {
  LOCAL_HISTOGRAM_COUNTS("foo3", 10);

  OnSetUMACallback("foo3", 1, 3, false);

  base::RunLoop().RunUntilIdle();

  EXPECT_TRUE(fake_sender_.last_message_);
  EXPECT_EQ(fake_sender_.last_message_->type(),
            static_cast<uint32_t>(TracingHostMsg_AbortBackgroundTrace::ID));
}

}  // namespace tracing