diff options
Diffstat (limited to 'src/components/policy/policy_external/test')
4 files changed, 11 insertions, 0 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 816b99cc2f..40b57f9ed3 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 @@ -117,6 +117,7 @@ class PolicyManagerImplTest : public ::testing::Test { MockUpdateStatusManager update_manager_; NiceMock<MockPolicyListener> listener_; NiceMock<MockPTURetryHandler> ptu_retry_handler_; + PolicyTableSPtr default_pt_snapshot_; void SetUp() OVERRIDE; diff --git a/src/components/policy/policy_external/test/policy_manager_impl_ptu_test.cc b/src/components/policy/policy_external/test/policy_manager_impl_ptu_test.cc index 920d2e1272..d5be757139 100644 --- a/src/components/policy/policy_external/test/policy_manager_impl_ptu_test.cc +++ b/src/components/policy/policy_external/test/policy_manager_impl_ptu_test.cc @@ -980,6 +980,8 @@ TEST_F(PolicyManagerImplTest2, TEST_F(PolicyManagerImplTest, LoadPT_SetInvalidUpdatePT_PTIsNotLoaded) { // Arrange + EXPECT_CALL(*cache_manager_, GenerateSnapshot()) + .WillOnce(Return(default_pt_snapshot_)); policy_manager_->ForcePTExchange(); policy_manager_->OnUpdateStarted(); Json::Value table(Json::objectValue); diff --git a/src/components/policy/policy_external/test/policy_manager_impl_test.cc b/src/components/policy/policy_external/test/policy_manager_impl_test.cc index f824d37dfb..388630a68f 100644 --- a/src/components/policy/policy_external/test/policy_manager_impl_test.cc +++ b/src/components/policy/policy_external/test/policy_manager_impl_test.cc @@ -106,6 +106,8 @@ TEST_F(PolicyManagerImplTest, LoadPT_SetPT_PTIsLoaded) { // Arrange EXPECT_CALL(*cache_manager_, DaysBeforeExchange(_)) .WillOnce(Return(kNonZero)); + ON_CALL(*cache_manager_, GenerateSnapshot()) + .WillByDefault(Return(default_pt_snapshot_)); policy_manager_->ForcePTExchange(); policy_manager_->SetSendOnUpdateFlags(true); policy_manager_->OnUpdateStarted(); @@ -254,6 +256,8 @@ TEST_F(PolicyManagerImplTest2, GetPolicyTableStatus_ExpectUpToDate) { TEST_F(PolicyManagerImplTest, SetUpdateStarted_GetPolicyTableStatus_Expect_Updating) { // Arrange + EXPECT_CALL(*cache_manager_, GenerateSnapshot()) + .WillOnce(Return(default_pt_snapshot_)); policy_manager_->ForcePTExchange(); EXPECT_CALL(*cache_manager_, SaveUpdateRequired(true)); policy_manager_->OnUpdateStarted(); @@ -306,6 +310,8 @@ TEST_F(PolicyManagerImplTest, MarkUnpairedDevice) { EXPECT_CALL(*cache_manager_, IgnitionCyclesBeforeExchange()); EXPECT_CALL(*cache_manager_, DaysBeforeExchange(_)); // Act + EXPECT_CALL(*cache_manager_, GenerateSnapshot()) + .WillOnce(Return(default_pt_snapshot_)); policy_manager_->MarkUnpairedDevice(unpaired_device_id_); } diff --git a/src/components/policy/policy_external/test/policy_manager_impl_test_base.cc b/src/components/policy/policy_external/test/policy_manager_impl_test_base.cc index dc1d7a06ef..7c2d231708 100644 --- a/src/components/policy/policy_external/test/policy_manager_impl_test_base.cc +++ b/src/components/policy/policy_external/test/policy_manager_impl_test_base.cc @@ -220,6 +220,8 @@ void PolicyManagerImplTest::SetUp() { ON_CALL(listener_, GetRegisteredLinks(_)).WillByDefault(Return()); ON_CALL(listener_, ptu_retry_handler()) .WillByDefault(ReturnRef(ptu_retry_handler_)); + Json::Value table = createPTforLoad(); + default_pt_snapshot_ = std::make_shared<policy_table::Table>(&table); } void PolicyManagerImplTest::TearDown() { |