// Copyright (c) FIRST and other WPILib contributors.
// Open Source Software; you can modify and/or share it under the terms of
// the WPILib BSD license file in the root directory of this project.

#include "wpi/system/Watchdog.hpp"

#include <stdint.h>

#include <gtest/gtest.h>

#include "wpi/simulation/SimHooks.hpp"

using namespace wpi;

namespace {
class WatchdogTest : public ::testing::Test {
 protected:
  void SetUp() override { wpi::sim::PauseTiming(); }

  void TearDown() override { wpi::sim::ResumeTiming(); }
};

}  // namespace

TEST_F(WatchdogTest, EnableDisable) {
  uint32_t watchdogCounter = 0;

  Watchdog watchdog(0.4_s, [&] { watchdogCounter++; });

  // Run 1
  watchdog.Enable();
  wpi::sim::StepTiming(0.2_s);
  watchdog.Disable();

  EXPECT_EQ(0u, watchdogCounter) << "Watchdog triggered early";

  // Run 2
  watchdogCounter = 0;
  watchdog.Enable();
  wpi::sim::StepTiming(0.4_s);
  watchdog.Disable();

  EXPECT_EQ(1u, watchdogCounter)
      << "Watchdog either didn't trigger or triggered more than once";

  // Run 3
  watchdogCounter = 0;
  watchdog.Enable();
  wpi::sim::StepTiming(1_s);
  watchdog.Disable();

  EXPECT_EQ(1u, watchdogCounter)
      << "Watchdog either didn't trigger or triggered more than once";
}

TEST_F(WatchdogTest, Reset) {
  uint32_t watchdogCounter = 0;

  Watchdog watchdog(0.4_s, [&] { watchdogCounter++; });

  watchdog.Enable();
  wpi::sim::StepTiming(0.2_s);
  watchdog.Reset();
  wpi::sim::StepTiming(0.2_s);
  watchdog.Disable();

  EXPECT_EQ(0u, watchdogCounter) << "Watchdog triggered early";
}

TEST_F(WatchdogTest, SetTimeout) {
  uint32_t watchdogCounter = 0;

  Watchdog watchdog(1_s, [&] { watchdogCounter++; });

  watchdog.Enable();
  wpi::sim::StepTiming(0.2_s);
  watchdog.SetTimeout(0.2_s);

  EXPECT_EQ(0.2_s, watchdog.GetTimeout());
  EXPECT_EQ(0u, watchdogCounter) << "Watchdog triggered early";

  wpi::sim::StepTiming(0.3_s);
  watchdog.Disable();

  EXPECT_EQ(1u, watchdogCounter)
      << "Watchdog either didn't trigger or triggered more than once";
}

TEST_F(WatchdogTest, IsExpired) {
  Watchdog watchdog(0.2_s, [] {});
  EXPECT_FALSE(watchdog.IsExpired());
  watchdog.Enable();

  EXPECT_FALSE(watchdog.IsExpired());
  wpi::sim::StepTiming(0.3_s);
  EXPECT_TRUE(watchdog.IsExpired());

  watchdog.Disable();
  EXPECT_TRUE(watchdog.IsExpired());

  watchdog.Reset();
  EXPECT_FALSE(watchdog.IsExpired());
}

TEST_F(WatchdogTest, Epochs) {
  uint32_t watchdogCounter = 0;

  Watchdog watchdog(0.4_s, [&] { watchdogCounter++; });

  // Run 1
  watchdog.Enable();
  watchdog.AddEpoch("Epoch 1");
  wpi::sim::StepTiming(0.1_s);
  watchdog.AddEpoch("Epoch 2");
  wpi::sim::StepTiming(0.1_s);
  watchdog.AddEpoch("Epoch 3");
  watchdog.Disable();

  EXPECT_EQ(0u, watchdogCounter) << "Watchdog triggered early";

  // Run 2
  watchdog.Enable();
  watchdog.AddEpoch("Epoch 1");
  wpi::sim::StepTiming(0.2_s);
  watchdog.Reset();
  wpi::sim::StepTiming(0.2_s);
  watchdog.AddEpoch("Epoch 2");
  watchdog.Disable();

  EXPECT_EQ(0u, watchdogCounter) << "Watchdog triggered early";
}

TEST_F(WatchdogTest, MultiWatchdog) {
  uint32_t watchdogCounter1 = 0;
  uint32_t watchdogCounter2 = 0;

  Watchdog watchdog1(0.2_s, [&] { watchdogCounter1++; });
  Watchdog watchdog2(0.6_s, [&] { watchdogCounter2++; });

  watchdog2.Enable();
  wpi::sim::StepTiming(0.25_s);
  EXPECT_EQ(0u, watchdogCounter1) << "Watchdog triggered early";
  EXPECT_EQ(0u, watchdogCounter2) << "Watchdog triggered early";

  // Sleep enough such that only the watchdog enabled later times out first
  watchdog1.Enable();
  wpi::sim::StepTiming(0.25_s);
  watchdog1.Disable();
  watchdog2.Disable();

  EXPECT_EQ(1u, watchdogCounter1)
      << "Watchdog either didn't trigger or triggered more than once";
  EXPECT_EQ(0u, watchdogCounter2) << "Watchdog triggered early";
}
