diff --git a/wpilibNewCommands/src/test/native/cpp/frc2/command/CommandTestBase.cpp b/wpilibNewCommands/src/test/native/cpp/frc2/command/CommandTestBase.cpp index 6e27d248c0..6e3bde02a9 100644 --- a/wpilibNewCommands/src/test/native/cpp/frc2/command/CommandTestBase.cpp +++ b/wpilibNewCommands/src/test/native/cpp/frc2/command/CommandTestBase.cpp @@ -11,22 +11,20 @@ CommandTestBase::CommandTestBase() { scheduler.CancelAll(); scheduler.Enable(); scheduler.GetActiveButtonLoop()->Clear(); + + SetDSEnabled(true); +} + +CommandTestBase::~CommandTestBase() { + CommandScheduler::GetInstance().GetActiveButtonLoop()->Clear(); } CommandScheduler CommandTestBase::GetScheduler() { return CommandScheduler(); } -void CommandTestBase::SetUp() { - frc::sim::DriverStationSim::SetEnabled(true); - frc::sim::DriverStationSim::NotifyNewData(); -} - -void CommandTestBase::TearDown() { - CommandScheduler::GetInstance().GetActiveButtonLoop()->Clear(); -} - void CommandTestBase::SetDSEnabled(bool enabled) { + frc::sim::DriverStationSim::SetDsAttached(true); frc::sim::DriverStationSim::SetEnabled(enabled); frc::sim::DriverStationSim::NotifyNewData(); } diff --git a/wpilibNewCommands/src/test/native/cpp/frc2/command/CommandTestBase.h b/wpilibNewCommands/src/test/native/cpp/frc2/command/CommandTestBase.h index 6ccc90428d..035f5dedbf 100644 --- a/wpilibNewCommands/src/test/native/cpp/frc2/command/CommandTestBase.h +++ b/wpilibNewCommands/src/test/native/cpp/frc2/command/CommandTestBase.h @@ -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 { 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(); } }; diff --git a/wpilibNewCommands/src/test/native/cpp/frc2/command/SchedulingRecursionTest.cpp b/wpilibNewCommands/src/test/native/cpp/frc2/command/SchedulingRecursionTest.cpp index 4aa5199230..733117c7bc 100644 --- a/wpilibNewCommands/src/test/native/cpp/frc2/command/SchedulingRecursionTest.cpp +++ b/wpilibNewCommands/src/test/native/cpp/frc2/command/SchedulingRecursionTest.cpp @@ -57,7 +57,7 @@ TEST_F(SchedulingRecursionTest, CancelFromInitialize) { } TEST_P(SchedulingRecursionTest, - DISABLED_DefaultCommandGetsRescheduledAfterSelfCanceling) { + DefaultCommandGetsRescheduledAfterSelfCanceling) { CommandScheduler scheduler = GetScheduler(); bool hasOtherRun = false; TestSubsystem requirement;