summaryrefslogtreecommitdiff
path: root/src/components/include/test/transport_manager/mock_transport_manager.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/components/include/test/transport_manager/mock_transport_manager.h')
-rw-r--r--src/components/include/test/transport_manager/mock_transport_manager.h4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/components/include/test/transport_manager/mock_transport_manager.h b/src/components/include/test/transport_manager/mock_transport_manager.h
index 884bde50bf..c853ceb512 100644
--- a/src/components/include/test/transport_manager/mock_transport_manager.h
+++ b/src/components/include/test/transport_manager/mock_transport_manager.h
@@ -35,7 +35,7 @@
#include <string>
#include "gmock/gmock.h"
-#include "resumption/last_state.h"
+#include "resumption/last_state_wrapper.h"
#include "telemetry_monitor/telemetry_observable.h"
#include "transport_manager/transport_adapter/transport_adapter_event.h"
#include "transport_manager/transport_manager.h"
@@ -56,6 +56,8 @@ class MockTransportManager : public ::transport_manager::TransportManager,
public ::telemetry_monitor::TelemetryObservable<
transport_manager::TMTelemetryObserver> {
public:
+ MOCK_METHOD1(Init, int(resumption::LastStateWrapperPtr last_state));
+ DEPRECATED
MOCK_METHOD1(Init, int(resumption::LastState& last_state));
MOCK_METHOD0(Reinit, int());
MOCK_METHOD0(Deinit, void());