powerdns-admin/tests/unit/zone/test_admin_apikey.py

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, admin_apikey, zone_data
from tests.fixtures import admin_apikey_data, load_data


class TestUnitApiZoneAdminApiKey(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
        )

        data = admin_apikey_data()
        api_key = ApiKey(
            desc=data['description'],
            role_name=data['role'],
            domains=[]
        )
        api_key.role = Role(name=data['role'])

        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
        self.mock_apikey.return_value.is_validate.return_value = api_key

    def test_empty_get(
        self,
        client,
        common_data_mock,
        admin_apikey
    ):
        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/servers/localhost/zones",
                headers=admin_apikey
            )

            data = res.get_json(force=True)
            assert res.status_code == 200
            assert data == []

    def test_create_zone(
        self,
        client,
        common_data_mock,
        zone_data,
        admin_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=admin_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,
        admin_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=admin_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,
        admin_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=admin_apikey
            )

            assert res.status_code == 204