From 33c1f642729871e0f768c2690aafa438afc04c17 Mon Sep 17 00:00:00 2001 From: Khanh Ngo Date: Wed, 18 Dec 2019 23:48:46 +0700 Subject: [PATCH] Update mock test --- tests/unit/zone/test_admin_apikey.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/unit/zone/test_admin_apikey.py b/tests/unit/zone/test_admin_apikey.py index 751b037..6f3a9c3 100644 --- a/tests/unit/zone/test_admin_apikey.py +++ b/tests/unit/zone/test_admin_apikey.py @@ -51,6 +51,9 @@ class TestUnitApiZoneAdminApiKey(object): self.mock_hist_patcher = patch( 'powerdnsadmin.routes.api.History', spec=powerdnsadmin.models.history.History) + self.mock_setting_patcher = patch( + 'powerdnsadmin.routes.api.Setting', + spec=powerdnsadmin.models.setting.Setting) data = admin_apikey_data() api_key = ApiKey(desc=data['description'], @@ -71,6 +74,7 @@ class TestUnitApiZoneAdminApiKey(object): ) self.mock_apikey = self.mock_apikey_patcher.start() self.mock_hist = self.mock_hist_patcher.start() + self.mock_setting = self.mock_setting_patcher.start() self.mock_google_setting.return_value.get.side_effect = load_data self.mock_github_setting.return_value.get.side_effect = load_data