Team2890/allwpilib
4
0
Fork 0
mirror of https://github.com/wpilibsuite/allwpilib synced 2026-06-19 00:41:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

[commands] Notify DriverStationSim in CommandTestBaseWithParam (#5400)

Browse Source
This commit is contained in:
Joseph Eng
2023-06-19 22:56:56 -07:00
committed by GitHub
parent d4ea5fa902
commit 316cd2a453
3 changed files with 18 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
wpilibNewCommands/src/test/native/cpp/frc2/command/CommandTestBase.h
View File

@@ -83,13 +83,11 @@ class CommandTestBase : public ::testing::Test {
public:
CommandTestBase();
~CommandTestBase() override;
protected:
CommandScheduler GetScheduler();
void SetUp() override;
void TearDown() override;
void SetDSEnabled(bool enabled);
};
@@ -101,19 +99,21 @@ class CommandTestBaseWithParam : public ::testing::TestWithParam<T> {
scheduler.CancelAll();
scheduler.Enable();
scheduler.GetActiveButtonLoop()->Clear();
SetDSEnabled(true);
}
~CommandTestBaseWithParam() override {
CommandScheduler::GetInstance().GetActiveButtonLoop()->Clear();
}
protected:
CommandScheduler GetScheduler() { return CommandScheduler(); }
void SetUp() override { frc::sim::DriverStationSim::SetEnabled(true); }
void TearDown() override {
CommandScheduler::GetInstance().GetActiveButtonLoop()->Clear();
}
void SetDSEnabled(bool enabled) {
frc::sim::DriverStationSim::SetDsAttached(true);
frc::sim::DriverStationSim::SetEnabled(enabled);
frc::sim::DriverStationSim::NotifyNewData();
}
};