1 files changed, 6 insertions, 3 deletions
diff --git a/src/components/policy/policy_external/test/include/policy/policy_manager_impl_test_base.h b/src/components/policy/policy_external/test/include/policy/policy_manager_impl_test_base.h
index 23701e86de..816b99cc2f 100644
--- a/src/components/policy/policy_external/test/include/policy/policy_manager_impl_test_base.h
+++ b/src/components/policy/policy_external/test/include/policy/policy_manager_impl_test_base.h
@@ -40,16 +40,17 @@
 #include "policy/policy_manager_impl.h"
 
 #include "policy/mock_cache_manager.h"
-#include "policy/mock_update_status_manager.h"
 #include "policy/mock_policy_listener.h"
 #include "policy/mock_policy_settings.h"
+#include "policy/mock_ptu_retry_handler.h"
+#include "policy/mock_update_status_manager.h"
 
 namespace test {
 namespace components {
 namespace policy_test {
 
-using ::testing::NiceMock;
 using ::policy::PolicyManagerImpl;
+using ::testing::NiceMock;
 
 typedef std::multimap<std::string, policy_table::Rpcs&>
     UserConsentPromptToRpcsConnections;
@@ -112,9 +113,10 @@ class PolicyManagerImplTest : public ::testing::Test {
   const std::string unpaired_device_id_;
 
   PolicyManagerImpl* policy_manager_;
-  MockCacheManagerInterface* cache_manager_;
+  NiceMock<MockCacheManagerInterface>* cache_manager_;
   MockUpdateStatusManager update_manager_;
   NiceMock<MockPolicyListener> listener_;
+  NiceMock<MockPTURetryHandler> ptu_retry_handler_;
 
   void SetUp() OVERRIDE;
 
@@ -140,6 +142,7 @@ class PolicyManagerImplTest2 : public ::testing::Test {
 
   PolicyManagerImpl* policy_manager_;
   NiceMock<MockPolicyListener> listener_;
+  NiceMock<MockPTURetryHandler> ptu_retry_handler_;
   ::policy::StringArray hmi_level_;
   ::policy::StringArray pt_request_types_;
   size_t ptu_request_types_size_;