diff options
Diffstat (limited to 'src/components/protocol_handler/test/include/protocol_handler/mock_telemetry_observer.h')
-rw-r--r-- | src/components/protocol_handler/test/include/protocol_handler/mock_telemetry_observer.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/components/protocol_handler/test/include/protocol_handler/mock_telemetry_observer.h b/src/components/protocol_handler/test/include/protocol_handler/mock_telemetry_observer.h index 9a91a5d1f7..f303e8ed57 100644 --- a/src/components/protocol_handler/test/include/protocol_handler/mock_telemetry_observer.h +++ b/src/components/protocol_handler/test/include/protocol_handler/mock_telemetry_observer.h @@ -34,8 +34,7 @@ #define SRC_COMPONENTS_PROTOCOL_HANDLER_TEST_INCLUDE_PROTOCOL_HANDLER_MOCK_TELEMETRY_OBSERVER_H_ #include "gmock/gmock.h" -#include "protocol_handler/time_metric_observer.h" -#include "utils/shared_ptr.h" +#include "protocol_handler/telemetry_observer.h" namespace test { namespace components { @@ -44,8 +43,9 @@ namespace protocol_handler_test { class MockPHTelemetryObserver : public ::protocol_handler::PHTelemetryObserver { public: MOCK_METHOD2(StartMessageProcess, - void(uint32_t, const date_time::TimeDuration&)); - MOCK_METHOD2(EndMessageProcess, void(std::shared_ptr<MessageMetric>)); + void(uint32_t message_id, + const date_time::TimeDuration& start_time)); + MOCK_METHOD1(EndMessageProcess, void(std::shared_ptr<MessageMetric> m)); }; } // namespace protocol_handler_test |