/**
 *
 * \section COPYRIGHT
 *
 * Copyright 2013-2020 Software Radio Systems Limited
 *
 * By using this file, you agree to the terms and conditions set
 * forth in the LICENSE file which can be found at the top level of
 * the distribution.
 *
 */

#include "srslte/srslog/logger.h"
#include "srslte/srslog/sink.h"
#include "testing_helpers.h"

using namespace srslog;

static constexpr char logger_id[] = "TestLogger";

namespace {

/// A Dummy implementation of a sink.
class sink_dummy : public sink
{
public:
  detail::error_string write(detail::memory_buffer buffer) override
  {
    return {};
  }

  detail::error_string flush() override { return {}; }
};

/// A Dummy implementation of the log backend.
class backend_dummy : public detail::log_backend
{
public:
  void start() override {}

  void push(detail::log_entry&& entry) override {}

  bool is_running() const override { return true; }
};

/// Definition of a three level logger
enum class test_logger_levels { error, warning, info, LAST };
struct test_logger_channels {
  log_channel& error;
  log_channel& warning;
  log_channel& info;
};
using test_logger = build_logger_type<test_logger_channels, test_logger_levels>;

} // namespace

static bool when_logger_is_created_then_id_matches_expected_value()
{
  backend_dummy backend;
  sink_dummy s;
  log_channel error("err", s, backend);
  log_channel warning("warning", s, backend);
  log_channel info("info", s, backend);

  test_logger logger(logger_id, error, warning, info);

  ASSERT_EQ(logger.id(), logger_id);

  return true;
}

static bool when_level_is_set_to_error_then_info_and_warning_is_disabled()
{
  backend_dummy backend;
  sink_dummy s;
  log_channel error("err", s, backend);
  log_channel warning("warning", s, backend);
  log_channel info("info", s, backend);

  test_logger logger(logger_id, error, warning, info);
  logger.set_level(test_logger_levels::error);

  ASSERT_EQ(logger.error.enabled(), true);
  ASSERT_EQ(logger.warning.enabled(), false);
  ASSERT_EQ(logger.info.enabled(), false);

  return true;
}

static bool when_level_is_set_to_warning_then_info_is_disabled()
{
  backend_dummy backend;
  sink_dummy s;
  log_channel error("err", s, backend);
  log_channel warning("warning", s, backend);
  log_channel info("info", s, backend);

  test_logger logger(logger_id, error, warning, info);
  logger.set_level(test_logger_levels::warning);

  ASSERT_EQ(logger.error.enabled(), true);
  ASSERT_EQ(logger.warning.enabled(), true);
  ASSERT_EQ(logger.info.enabled(), false);

  return true;
}

static bool when_level_is_set_to_info_then_all_are_enabled()
{
  backend_dummy backend;
  sink_dummy s;
  log_channel error("err", s, backend);
  log_channel warning("warning", s, backend);
  log_channel info("info", s, backend);

  test_logger logger(logger_id, error, warning, info);
  logger.set_level(test_logger_levels::info);

  ASSERT_EQ(logger.error.enabled(), true);
  ASSERT_EQ(logger.warning.enabled(), true);
  ASSERT_EQ(logger.info.enabled(), true);

  return true;
}

int main()
{
  TEST_FUNCTION(when_logger_is_created_then_id_matches_expected_value);
  TEST_FUNCTION(when_level_is_set_to_error_then_info_and_warning_is_disabled);
  TEST_FUNCTION(when_level_is_set_to_warning_then_info_is_disabled);
  TEST_FUNCTION(when_level_is_set_to_info_then_all_are_enabled);

  return 0;
}