mirror of
https://github.com/cwinfo/powerdns-admin.git
synced 2024-09-19 23:02:31 +00:00
8dd03a4d85
Fix migration init_db 'id' Handle app context when needed Fix conftest fixtures Rearrange test Dockerfiles Hide DeprecationWarning during pytest execution Upgrade all python packages
173 lines
8.1 KiB
Python
173 lines
8.1 KiB
Python
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 TestUnitApiZoneUser(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 = "User"
|
|
|
|
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_create_zone(self, client, common_data_mock, zone_data,
|
|
basic_auth_user_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_user_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_user_headers):
|
|
with patch('powerdnsadmin.routes.api.get_user_domains') as mock_user_domains:
|
|
test_domain = Domain(1, name=zone_data['name'].rstrip("."))
|
|
mock_user_domains.return_value = [test_domain]
|
|
|
|
res = client.get("/api/v1/pdnsadmin/zones",
|
|
headers=basic_auth_user_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_user_headers):
|
|
with patch('powerdnsadmin.lib.helper.requests.request') as mock_delete, \
|
|
patch('powerdnsadmin.routes.api.Domain') as mock_domain, \
|
|
patch('powerdnsadmin.routes.api.get_user_domains') as mock_user_domains:
|
|
test_domain = Domain(1, name=zone_data['name'].rstrip("."))
|
|
|
|
mock_domain.return_value.update.return_value = True
|
|
mock_user_domains.return_value = [test_domain]
|
|
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_user_headers)
|
|
|
|
assert res.status_code == 204
|