168 lines
5.8 KiB
Python
168 lines
5.8 KiB
Python
import os
|
|
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
import sys
|
|
import json
|
|
from collections import namedtuple
|
|
import logging as logger
|
|
sys.path.append(os.getcwd())
|
|
import app
|
|
from app.validators import validate_zone
|
|
from app.models import User, Domain, Role
|
|
from app.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.oauth_setting_patcher = patch(
|
|
'app.oauth.Setting',
|
|
spec=app.models.Setting
|
|
)
|
|
self.api_setting_patcher = patch(
|
|
'app.blueprints.api.Setting',
|
|
spec=app.models.Setting
|
|
)
|
|
self.views_setting_patcher = patch(
|
|
'app.views.Setting',
|
|
spec=app.models.Setting
|
|
)
|
|
self.helpers_setting_patcher = patch(
|
|
'app.lib.helper.Setting',
|
|
spec=app.models.Setting
|
|
)
|
|
self.models_setting_patcher = patch(
|
|
'app.models.Setting',
|
|
spec=app.models.Setting
|
|
)
|
|
self.decorators_setting_patcher = patch(
|
|
'app.decorators.Setting'
|
|
)
|
|
self.mock_user_patcher = patch(
|
|
'app.decorators.User'
|
|
)
|
|
self.mock_hist_patcher = patch(
|
|
'app.blueprints.api.History',
|
|
spec=app.models.History
|
|
)
|
|
|
|
self.mock_oauth_setting = self.oauth_setting_patcher.start()
|
|
self.mock_views_setting = self.views_setting_patcher.start()
|
|
self.mock_helpers_setting = self.helpers_setting_patcher.start()
|
|
self.mock_models_setting = self.models_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_oauth_setting.return_value.get.side_effect = load_data
|
|
self.mock_views_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.decorators_setting.return_value.get.side_effect = load_data
|
|
self.api_setting.return_value.get.side_effect = load_data
|
|
mockk = MagicMock()
|
|
mockk.role.name = "Administrator"
|
|
self.mock_user.query.filter.return_value.first.return_value = mockk
|
|
self.mock_user.return_value.is_validate.return_value = True
|
|
|
|
def test_empty_get(
|
|
self,
|
|
client,
|
|
common_data_mock,
|
|
basic_auth_admin_headers
|
|
):
|
|
with patch('app.blueprints.api.Domain') as mock_domain, \
|
|
patch('app.lib.utils.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('app.lib.helper.requests.request') as mock_post, \
|
|
patch('app.blueprints.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('app.blueprints.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('app.lib.utils.requests.request') as mock_delete, \
|
|
patch('app.blueprints.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
|