y2024_swerve/swerve_publisher_lib_test.cc

#include "y2024_swerve/swerve_publisher_lib.h"

#include "gtest/gtest.h"

#include "aos/events/simulated_event_loop.h"

namespace y2024_swerve::testing {
class SwervePublisherTest : public ::testing::Test {
 public:
  SwervePublisherTest()
      : config_(aos::configuration::ReadConfig("y2024_swerve/aos_config.json")),
        event_loop_factory_(&config_.message()),
        roborio_(aos::configuration::GetNode(
            event_loop_factory_.configuration(), "roborio")),
        event_loop_(
            event_loop_factory_.MakeEventLoop("swerve_publisher", roborio_)),
        exit_handle_(event_loop_factory_.MakeExitHandle()),
        drivetrain_swerve_goal_fetcher_(
            event_loop_->MakeFetcher<frc971::control_loops::swerve::Goal>(
                "/drivetrain")),
        swerve_publisher_(event_loop_.get(), exit_handle_.get(),
                          "y2024_swerve/swerve_drivetrain_goal.json", 100) {}

  void SendOutput() { event_loop_factory_.Run(); }

  void CheckOutput() {
    drivetrain_swerve_goal_fetcher_.Fetch();

    ASSERT_TRUE(drivetrain_swerve_goal_fetcher_.get() != nullptr)
        << ": No drivetrain goal.";
  }

 private:
  aos::FlatbufferDetachedBuffer<aos::Configuration> config_;
  aos::SimulatedEventLoopFactory event_loop_factory_;
  const aos::Node *const roborio_;

  std::unique_ptr<aos::EventLoop> event_loop_;
  std::unique_ptr<aos::ExitHandle> exit_handle_;

  aos::Fetcher<frc971::control_loops::swerve::Goal>
      drivetrain_swerve_goal_fetcher_;

  y2024_swerve::SwervePublisher swerve_publisher_;
};

TEST_F(SwervePublisherTest, CheckSentFb) {
  SendOutput();
  CheckOutput();
}
}  // namespace y2024_swerve::testing