Updated mock classes

2 files changed, 7 insertions, 5 deletions

diff --git a/src/components/application_manager/test/include/application_manager/mock_resume_ctrl.h b/src/components/application_manager/test/include/application_manager/mock_resume_ctrl.h
index 73fa97c2c4..94ebb396fe 100644
--- a/src/components/application_manager/test/include/application_manager/mock_resume_ctrl.h
+++ b/src/components/application_manager/test/include/application_manager/mock_resume_ctrl.h
@@ -63,7 +63,9 @@ class MockResumeCtrl : public resumption::ResumeCtrl {
                bool(app_mngr::ApplicationSharedPtr application,
                     const std::string& hash,
                     resumption::ResumeCtrl::ResumptionCallBack));
-  MOCK_METHOD1(HandleOnTimeOut, void(const int32_t app_id));
+  MOCK_METHOD2(HandleOnTimeOut,
+               void(const uint32_t cor_id,
+                    const hmi_apis::FunctionID::eType function_id));
   MOCK_METHOD1(StartResumptionOnlyHMILevel,
                bool(app_mngr::ApplicationSharedPtr application));
   MOCK_METHOD1(RetryResumption, void(const uint32_t app_id));
diff --git a/src/components/include/test/application_manager/mock_app_extension.h b/src/components/include/test/application_manager/mock_app_extension.h
index a42c9229e4..e98c15b1d8 100644
--- a/src/components/include/test/application_manager/mock_app_extension.h
+++ b/src/components/include/test/application_manager/mock_app_extension.h
@@ -47,10 +47,10 @@ static unsigned MockAppExtensionUID = 123;
 class MockAppExtension : public application_manager::AppExtension {
  public:
   MockAppExtension() : AppExtension(MockAppExtensionUID) {}
-  MOCK_METHOD1(
-      SaveResumptionData,
-      void(ns_smart_device_link::ns_smart_objects::SmartObject& resumption_data));
-  MOCK_METHOD1(ProcessResumption,
+  MOCK_METHOD1(SaveResumptionData,
+               void(ns_smart_device_link::ns_smart_objects::SmartObject&
+                        resumption_data));
+  MOCK_METHOD2(ProcessResumption,
                void(const ns_smart_device_link::ns_smart_objects::SmartObject&
                         resumption_data,
                     resumption::Subscriber subscriber));