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.role import Role from powerdnsadmin.models.domain import Domain from powerdnsadmin.lib.validators import validate_zone from powerdnsadmin.lib.schema import DomainSchema from tests.fixtures import client, basic_auth_admin_headers from tests.fixtures import zone_data, created_zone_data, load_data class TestUnitApiZoneAdminUser(object): @pytest.fixture def common_data_mock(self): 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.api_setting_patcher = patch( 'powerdnsadmin.routes.api.Setting', spec=powerdnsadmin.models.setting.Setting) self.base_route_user_patcher = patch( 'powerdnsadmin.routes.base.User', spec=powerdnsadmin.models.user.User) self.helpers_setting_patcher = patch( 'powerdnsadmin.lib.helper.Setting', spec=powerdnsadmin.models.setting.Setting) self.models_setting_patcher = patch( 'powerdnsadmin.models.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.decorators_setting_patcher = patch( 'powerdnsadmin.decorators.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_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.decorators_setting = self.decorators_setting_patcher.start() self.api_setting = self.api_setting_patcher.start() self.mock_user = self.mock_user_patcher.start() self.mock_hist = self.mock_hist_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.decorators_setting.return_value.get.side_effect = load_data self.api_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.decorators_setting_patcher, self.api_setting_patcher, self.mock_user_patcher, self.mock_hist_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