import os import pytest from unittest.mock import patch, MagicMock import sys import json from base64 import b64encode 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 Setting, Domain, ApiKey, Role from app.schema import DomainSchema, ApiKeySchema from tests.fixtures import client, initial_data, created_zone_data from tests.fixtures import user_apikey, zone_data from tests.fixtures import user_apikey_data, load_data class TestUnitApiZoneUserApiKey(object): @pytest.fixture def common_data_mock(self): self.oauth_setting_patcher = patch( 'app.oauth.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.mock_apikey_patcher = patch( 'app.decorators.ApiKey', spec=app.models.ApiKey ) 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.mock_apikey = self.mock_apikey_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 data = user_apikey_data() domain = Domain(name=data['domains'][0]) api_key = ApiKey( desc=data['description'], role_name=data['role'], domains=[domain] ) api_key.role = Role(name=data['role']) self.mock_apikey.return_value.is_validate.return_value = api_key def test_create_zone( self, client, common_data_mock, zone_data, user_apikey, 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/servers/localhost/zones", headers=user_apikey, 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, user_apikey ): 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/servers/localhost/zones", headers=user_apikey ) 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, user_apikey ): 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_delete.return_value.status_code = 204 mock_delete.return_value.content = '' zone_url_format = "/api/v1/servers/localhost/zones/{0}" zone_url = zone_url_format.format(zone_data['name'].rstrip(".")) res = client.delete( zone_url, headers=user_apikey ) assert res.status_code == 204