import json import pytest from unittest.mock import patch, MagicMock from collections import namedtuple import powerdnsadmin from powerdnsadmin.models.user import User from powerdnsadmin.models.domain import Domain from powerdnsadmin.lib.validators import validate_zone from powerdnsadmin.lib.schema import DomainSchema from tests.conftest import load_data class TestUnitApiZoneAdminUser(object): @pytest.fixture def common_data_mock(self, app, initial_data): self.google_setting_patcher = patch( 'powerdnsadmin.services.google.Setting', spec=powerdnsadmin.models.setting.Setting) self.github_setting_patcher = patch( 'powerdnsadmin.services.github.Setting', spec=powerdnsadmin.models.setting.Setting) self.azure_setting_patcher = patch( 'powerdnsadmin.services.azure.Setting', spec=powerdnsadmin.models.setting.Setting) self.oidc_setting_patcher = patch( 'powerdnsadmin.services.oidc.Setting', spec=powerdnsadmin.models.setting.Setting) self.helpers_setting_patcher = patch( 'powerdnsadmin.lib.helper.Setting', spec=powerdnsadmin.models.setting.Setting) self.models_setting_patcher = patch( 'powerdnsadmin.models.setting.Setting', spec=powerdnsadmin.models.setting.Setting) self.domain_model_setting_patcher = patch( 'powerdnsadmin.models.domain.Setting', spec=powerdnsadmin.models.setting.Setting) self.record_model_setting_patcher = patch( 'powerdnsadmin.models.record.Setting', spec=powerdnsadmin.models.setting.Setting) self.server_model_setting_patcher = patch( 'powerdnsadmin.models.server.Setting', spec=powerdnsadmin.models.setting.Setting) self.mock_user_patcher = patch( 'powerdnsadmin.decorators.User', spec=powerdnsadmin.models.user.User) 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) self.mock_decorators_setting_patcher = patch( 'powerdnsadmin.decorators.Setting', spec=powerdnsadmin.models.setting.Setting) self.base_route_user_patcher = patch( 'powerdnsadmin.routes.base.User', spec=powerdnsadmin.models.user.User) with app.app_context(): self.mock_google_setting = self.google_setting_patcher.start() self.mock_github_setting = self.github_setting_patcher.start() self.mock_azure_setting = self.azure_setting_patcher.start() self.mock_oidc_setting = self.oidc_setting_patcher.start() self.mock_base_route_user = self.base_route_user_patcher.start() self.mock_helpers_setting = self.helpers_setting_patcher.start() self.mock_models_setting = self.models_setting_patcher.start() self.mock_domain_model_setting = self.domain_model_setting_patcher.start( ) self.mock_record_model_setting = self.record_model_setting_patcher.start( ) self.mock_server_model_setting = self.server_model_setting_patcher.start( ) self.mock_user = self.mock_user_patcher.start() self.mock_hist = self.mock_hist_patcher.start() self.mock_setting = self.mock_setting_patcher.start() self.mock_decorators_setting = self.mock_decorators_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 self.mock_azure_setting.return_value.get.side_effect = load_data self.mock_oidc_setting.return_value.get.side_effect = load_data self.mock_helpers_setting.return_value.get.side_effect = load_data self.mock_models_setting.return_value.get.side_effect = load_data self.mock_domain_model_setting.return_value.get.side_effect = load_data self.mock_record_model_setting.return_value.get.side_effect = load_data self.mock_server_model_setting.return_value.get.side_effect = load_data self.mock_decorators_setting.return_value.get.side_effect = load_data self.mock_setting.return_value.get.side_effect = load_data self.mockk = MagicMock() self.mockk.role.name = "Administrator" self.mock_user.query.filter.return_value.first.return_value = self.mockk self.mock_user.return_value.is_validate.return_value = True self.mock_base_route_user.query.filter.return_value.first.return_value = self.mockk self.mock_base_route_user.return_value.is_validate.return_value = True yield for patcher in [ self.google_setting_patcher, self.github_setting_patcher, self.azure_setting_patcher, self.oidc_setting_patcher, self.base_route_user_patcher, self.helpers_setting_patcher, self.models_setting_patcher, self.domain_model_setting_patcher, self.record_model_setting_patcher, self.server_model_setting_patcher, self.mock_user_patcher, self.mock_hist_patcher, self.mock_setting_patcher, self.mock_decorators_setting_patcher, ]: patcher.stop() def test_empty_get(self, client, common_data_mock, basic_auth_admin_headers): with patch('powerdnsadmin.routes.api.Domain') as mock_domain, \ patch('powerdnsadmin.lib.helper.requests.get') as mock_get: mock_domain.return_value.domains.return_value = [] mock_domain.query.all.return_value = [] mock_get.return_value.json.return_value = [] mock_get.return_value.status_code = 200 res = client.get("/api/v1/pdnsadmin/zones", headers=basic_auth_admin_headers) data = res.get_json(force=True) assert res.status_code == 200 assert data == [] def test_create_zone(self, client, common_data_mock, zone_data, basic_auth_admin_headers, created_zone_data): with patch('powerdnsadmin.lib.helper.requests.request') as mock_post, \ patch('powerdnsadmin.routes.api.Domain') as mock_domain: mock_post.return_value.status_code = 201 mock_post.return_value.content = json.dumps(created_zone_data) mock_post.return_value.headers = {} mock_domain.return_value.update.return_value = True res = client.post("/api/v1/pdnsadmin/zones", headers=basic_auth_admin_headers, data=json.dumps(zone_data), content_type="application/json") data = res.get_json(force=True) data['rrsets'] = [] validate_zone(data) assert res.status_code == 201 def test_get_multiple_zones(self, client, common_data_mock, zone_data, basic_auth_admin_headers): with patch('powerdnsadmin.routes.api.Domain') as mock_domain: test_domain = Domain(1, name=zone_data['name'].rstrip(".")) mock_domain.query.all.return_value = [test_domain] res = client.get("/api/v1/pdnsadmin/zones", headers=basic_auth_admin_headers) data = res.get_json(force=True) fake_domain = namedtuple("Domain", data[0].keys())(*data[0].values()) domain_schema = DomainSchema(many=True) json.dumps(domain_schema.dump([fake_domain])) assert res.status_code == 200 def test_delete_zone(self, client, common_data_mock, zone_data, basic_auth_admin_headers): with patch('powerdnsadmin.lib.helper.requests.request') as mock_delete, \ patch('powerdnsadmin.routes.api.Domain') as mock_domain: mock_domain.return_value.update.return_value = True mock_domain.query.filter.return_value = True mock_delete.return_value.status_code = 204 mock_delete.return_value.content = '' zone_url_format = "/api/v1/pdnsadmin/zones/{0}" zone_url = zone_url_format.format(zone_data['name'].rstrip(".")) res = client.delete(zone_url, headers=basic_auth_admin_headers) assert res.status_code == 204