// Copyright 2018 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 "chrome/chrome_cleaner/logging/scoped_timed_task_logger.h"

#include <vector>

#include "base/bind.h"
#include "base/logging.h"
#include "chrome/chrome_cleaner/test/test_util.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace chrome_cleaner {

namespace {

class ScopedTimedTaskLoggerTest : public testing::Test {
 public:
  // Intercepts all log messages.
  static bool LogMessageHandler(int severity,
                                const char* file,
                                int line,
                                size_t message_start,
                                const std::string& str) {
    DCHECK(active_logging_messages_);
    active_logging_messages_->push_back(str);
    return false;
  }

  bool LoggingMessagesContain(const std::string& sub_string) {
    std::vector<std::string>::const_iterator line = logging_messages_.begin();
    for (; line != logging_messages_.end(); ++line) {
      if (StringContainsCaseInsensitive(*line, sub_string))
        return true;
    }
    return false;
  }

  void FlushMessages() { logging_messages_.clear(); }

  void SetUp() override {
    DCHECK(active_logging_messages_ == nullptr);
    active_logging_messages_ = &logging_messages_;
    logging::SetLogMessageHandler(&LogMessageHandler);
  }

  void TearDown() override {
    logging::SetLogMessageHandler(nullptr);
    logging_messages_.clear();
    DCHECK(active_logging_messages_ == &logging_messages_);
    active_logging_messages_ = nullptr;
  }

  void timer_callback(const base::TimeDelta&) { callback_called_ = true; }

  std::vector<std::string> logging_messages_;
  static std::vector<std::string>* active_logging_messages_;
  bool callback_called_ = false;
};

std::vector<std::string>* ScopedTimedTaskLoggerTest::active_logging_messages_ =
    nullptr;

}  // namespace

TEST_F(ScopedTimedTaskLoggerTest, CallbackCalled) {
  {
    ScopedTimedTaskLogger timer(base::BindRepeating(
        &ScopedTimedTaskLoggerTest::timer_callback, base::Unretained(this)));
  }
  EXPECT_TRUE(callback_called_);
}

TEST_F(ScopedTimedTaskLoggerTest, NoLog) {
  static const char kNoShow[] = "Should not show up";
  {
    ScopedTimedTaskLogger no_logs(
        base::BindRepeating(ScopedTimedTaskLogger::LogIfExceedThreshold,
                            kNoShow, base::TimeDelta::FromDays(1)));
  }
  EXPECT_FALSE(LoggingMessagesContain(kNoShow));
}

TEST_F(ScopedTimedTaskLoggerTest, Log) {
  static const char kShow[] = "Should show up";
  {
    ScopedTimedTaskLogger logs(
        base::BindRepeating(ScopedTimedTaskLogger::LogIfExceedThreshold, kShow,
                            base::TimeDelta::FromMilliseconds(0)));
    ::Sleep(2);
  }
  EXPECT_TRUE(LoggingMessagesContain(kShow));
}

}  // namespace chrome_cleaner