/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "NSPRLogModulesParser.h"
#include "gtest/gtest.h"

using namespace mozilla;

TEST(NSPRLogModulesParser, Empty)
{
  bool callbackInvoked = false;
  auto callback = [&](const char*, mozilla::LogLevel, int32_t) mutable {
    callbackInvoked = true;
  };

  mozilla::NSPRLogModulesParser(nullptr, callback);
  EXPECT_FALSE(callbackInvoked);

  mozilla::NSPRLogModulesParser("", callback);
  EXPECT_FALSE(callbackInvoked);
}

TEST(NSPRLogModulesParser, DefaultLevel)
{
  bool callbackInvoked = false;
  auto callback = [&](const char* aName, mozilla::LogLevel aLevel, int32_t) {
    EXPECT_STREQ("Foo", aName);
    EXPECT_EQ(mozilla::LogLevel::Error, aLevel);
    callbackInvoked = true;
  };

  mozilla::NSPRLogModulesParser("Foo", callback);
  EXPECT_TRUE(callbackInvoked);

  callbackInvoked = false;
  mozilla::NSPRLogModulesParser("Foo:", callback);
  EXPECT_TRUE(callbackInvoked);
}

TEST(NSPRLogModulesParser, LevelSpecified)
{
  std::pair<const char*, mozilla::LogLevel> expected[] = {
      {"Foo:0", mozilla::LogLevel::Disabled},
      {"Foo:1", mozilla::LogLevel::Error},
      {"Foo:2", mozilla::LogLevel::Warning},
      {"Foo:3", mozilla::LogLevel::Info},
      {"Foo:4", mozilla::LogLevel::Debug},
      {"Foo:5", mozilla::LogLevel::Verbose},
      {"Foo:25", mozilla::LogLevel::Verbose},   // too high
      {"Foo:-12", mozilla::LogLevel::Disabled}  // too low
  };

  auto* currTest = expected;

  for (size_t i = 0; i < std::size(expected); i++) {
    bool callbackInvoked = false;
    mozilla::NSPRLogModulesParser(
        currTest->first,
        [&](const char* aName, mozilla::LogLevel aLevel, int32_t) {
          EXPECT_STREQ("Foo", aName);
          EXPECT_EQ(currTest->second, aLevel);
          callbackInvoked = true;
        });
    EXPECT_TRUE(callbackInvoked);
    currTest++;
  }
}

TEST(NSPRLogModulesParser, Multiple)
{
  std::pair<const char*, mozilla::LogLevel> expected[] = {
      {"timestamp", mozilla::LogLevel::Error},
      {"Foo", mozilla::LogLevel::Info},
      {"Bar", mozilla::LogLevel::Error},
      {"Baz", mozilla::LogLevel::Warning},
      {"Qux", mozilla::LogLevel::Verbose},
  };

  const size_t kExpectedCount = std::size(expected);

  auto* currTest = expected;

  size_t count = 0;
  mozilla::NSPRLogModulesParser(
      "timestamp,Foo:3, Bar,Baz:2,    Qux:5",
      [&](const char* aName, mozilla::LogLevel aLevel, int32_t) mutable {
        ASSERT_LT(count, kExpectedCount);
        EXPECT_STREQ(currTest->first, aName);
        EXPECT_EQ(currTest->second, aLevel);
        currTest++;
        count++;
      });

  EXPECT_EQ(kExpectedCount, count);
}

TEST(NSPRLogModulesParser, Characters)
{
  std::pair<const char*, mozilla::LogLevel> expected[] = {
      {"valid.name", mozilla::LogLevel::Verbose},
      {"valid_name", mozilla::LogLevel::Debug},
      {"invalid", mozilla::LogLevel::Error},
  };

  const size_t kExpectedCount = std::size(expected);

  auto* currTest = expected;

  size_t count = 0;
  mozilla::NSPRLogModulesParser(
      "valid.name:5,valid_name:4,invalid/name:3,aborts-everything:2",
      [&](const char* aName, mozilla::LogLevel aLevel, int32_t) mutable {
        ASSERT_LT(count, kExpectedCount);
        EXPECT_STREQ(currTest->first, aName);
        EXPECT_EQ(currTest->second, aLevel);
        currTest++;
        count++;
      });

  EXPECT_EQ(kExpectedCount, count);
}

TEST(NSPRLogModulesParser, RawArg)
{
  bool callbackInvoked = false;
  auto callback = [&](const char* aName, mozilla::LogLevel aLevel,
                      int32_t aRawValue) {
    EXPECT_STREQ("Foo", aName);
    EXPECT_EQ(mozilla::LogLevel::Verbose, aLevel);
    EXPECT_EQ(1000, aRawValue);
    callbackInvoked = true;
  };

  mozilla::NSPRLogModulesParser("Foo:1000", callback);
  EXPECT_TRUE(callbackInvoked);
}

TEST(NSPRLogModulesParser, RustModules)
{
  std::pair<const char*, mozilla::LogLevel> expected[] = {
      {"timestamp", mozilla::LogLevel::Error},
      {"crate::mod::file1", mozilla::LogLevel::Error},
      {"crate::mod::file2", mozilla::LogLevel::Verbose},
      {"crate::mod::*", mozilla::LogLevel::Info},
  };

  const size_t kExpectedCount = std::size(expected);

  auto* currTest = expected;

  size_t count = 0;
  mozilla::NSPRLogModulesParser(
      "timestamp,crate::mod::file1, crate::mod::file2:5, crate::mod::*:3",
      [&](const char* aName, mozilla::LogLevel aLevel, int32_t) mutable {
        ASSERT_LT(count, kExpectedCount);
        EXPECT_STREQ(currTest->first, aName);
        EXPECT_EQ(currTest->second, aLevel);
        currTest++;
        count++;
      });

  EXPECT_EQ(kExpectedCount, count);
}
