Walle/tests/test_01_api_environment.py

# -*- coding: utf-8 -*-
"""Test Apis."""
from flask import json
import types
import urllib
import pytest
from utils import *


@pytest.mark.usefixtures('db')
class TestApiEnv:

    """api role testing"""
    uri_prefix = '/api/environment'

    env_id = {}

    env_data = {
        'env_name': u'测试环境',
        'space_id': 1,
    }

    user_name_2 = u'Production'

    env_data_2 = {
        'env_name': u'Production',
        'space_id': 1,
    }

    env_data_remove = {
        'env_name': u'environment_remove',
        'space_id': 1,
    }

    def test_create(self, user, testapp, client, db):
        """create successful."""
        # 1.create another role
        resp = client.post('%s/' % (self.uri_prefix), data=self.env_data)

        response_success(resp)
        compare_req_resp(self.env_data, resp)

        self.env_data['id'] = resp_json(resp)['data']['id']

        # f.write(str(self.env_data))
        # f.write(str(resp_json(resp)['data']['id']))


        # 2.create another role
        resp = client.post('%s/' % (self.uri_prefix), data=self.env_data_2)

        response_success(resp)
        compare_req_resp(self.env_data_2, resp)

        self.env_data_2['id'] = resp_json(resp)['data']['id']

    def test_one(self, user, testapp, client, db):
        """item successful."""
        # Goes to homepage

        resp = client.get('%s/%d' % (self.uri_prefix, self.env_data['id']))

        response_success(resp)
        compare_req_resp(self.env_data, resp)

    def test_get_list_page_size(self, user, testapp, client):
        """test list should create 2 users at least, due to test pagination, searching."""

        query = {
            'page': 1,
            'size': 1,
        }
        response = {
            'count': 2,
        }
        resp = client.get('%s/?%s' % (self.uri_prefix, urllib.urlencode(query)))
        response_success(resp)
        resp_dict = resp_json(resp)

        compare_in(self.env_data_2, resp_dict['data']['list'].pop())
        compare_req_resp(response, resp)

    def test_get_list_query(self, user, testapp, client):
        """test list should create 2 users at least, due to test pagination, searching."""
        query = {
            'page': 1,
            'size': 1,
            'kw': self.user_name_2
        }
        response = {
            'count': 1,
        }
        resp = client.get('%s/?%s' % (self.uri_prefix, urllib.urlencode(query)))
        response_success(resp)
        resp_dict = resp_json(resp)

        compare_in(self.env_data_2, resp_dict['data']['list'].pop())
        compare_req_resp(response, resp)

    def test_get_update(self, user, testapp, client):
        """Login successful."""
        # 1.update
        env_data_2 = self.env_data_2
        env_data_2['env_name'] = 'Tester_edit'
        resp = client.put('%s/%d' % (self.uri_prefix, self.env_data_2['id']), data=env_data_2)

        response_success(resp)
        compare_req_resp(env_data_2, resp)

        # 3.get it
        resp = client.get('%s/%d' % (self.uri_prefix, self.env_data_2['id']))
        response_success(resp)
        compare_req_resp(env_data_2, resp)

    def test_get_remove(self, user, testapp, client):
        """Login successful."""
        # 1.create another role
        resp = client.post('%s/' % (self.uri_prefix), data=self.env_data_remove)
        env_id = resp_json(resp)['data']['id']
        response_success(resp)

        # 2.delete
        resp = client.delete('%s/%d' % (self.uri_prefix, env_id))
        response_success(resp)

        # 3.get it
        resp = client.get('%s/%d' % (self.uri_prefix, env_id))
        response_error(resp)