Adding mocking for the new function.

mockingbirdnest · Dec 18, 2022 · 8af84d1 · 8af84d1
1 parent 7e9c30a
commit 8af84d1
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/ksp_plugin/plugin.hpp b/ksp_plugin/plugin.hpp
@@ -349,7 +349,7 @@ class Plugin {
 
   // If there is a target vessel, increases the size of the prediction to try
   // and cover the actual final time of the flight plan.
-  void ExtendPredictionForFlightPlan(GUID const& vessel_guid) const;
+  virtual void ExtendPredictionForFlightPlan(GUID const& vessel_guid) const;
 
   // Computes the apsides of the trajectory defined by |begin| and |end| with
   // respect to the celestial with index |celestial_index|.

diff --git a/ksp_plugin_test/interface_flight_plan_test.cpp b/ksp_plugin_test/interface_flight_plan_test.cpp
@@ -78,6 +78,7 @@ using quantities::si::Second;
 using quantities::si::Tonne;
 using testing_utilities::AlmostEquals;
 using ::testing::AllOf;
+using ::testing::AnyNumber;
 using ::testing::ByMove;
 using ::testing::DoAll;
 using ::testing::Invoke;
@@ -147,6 +148,8 @@ TEST_F(InterfaceFlightPlanTest, FlightPlan) {
       .WillRepeatedly(Return(true));
   EXPECT_CALL(vessel, flight_plan())
       .WillRepeatedly(ReturnRef(flight_plan));
+  EXPECT_CALL(*plugin_, ExtendPredictionForFlightPlan(vessel_guid))
+      .Times(AnyNumber());
 
   EXPECT_TRUE(principia__FlightPlanExists(plugin_.get(), vessel_guid));
 

diff --git a/ksp_plugin_test/mock_plugin.hpp b/ksp_plugin_test/mock_plugin.hpp
@@ -83,6 +83,11 @@ class MockPlugin : public Plugin {
                Mass const& initial_mass),
               (const, override));
 
+  MOCK_METHOD(void,
+              ExtendPredictionForFlightPlan,
+              (GUID const& vessel_guid),
+              (const, override));
+
   MOCK_METHOD(void,
               SetPredictionAdaptiveStepParameters,
               (GUID const& vessel_guid,