tests.views

Complexity

Total Complexity

12

Size/Duplication

Total Lines	82
Duplicated Lines	0 %

Importance

Changes

0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	TCBase.tearDown()	0	2	1
A	TCBase._get_tokens()	0	3	1
A	TCBase.__init__()	0	4	1
A	TCBase.setUp()	0	3	1
A	TCBase._create_fake_account()	0	2	1
B	TCBase.json_request()	0	24	3
A	TCBase.request()	0	20	2
A	TCBase.decode_response_data()	0	2	1
A	TCBase.get_response_data()	0	2	1

import ujson
from unittest import TestCase as TC

from flask import Response

from app import app


class TCBase(TC):
    def __init__(self, *args, **kwargs):
        TC.__init__(self, *args, **kwargs)

        self.client = app.test_client()

    def _create_fake_account(self):
        pass

    def _get_tokens(self):
        self.access_token = None
        self.refresh_token = None

    def setUp(self):
        self._create_fake_account()
        self._get_tokens()

    def tearDown(self):
        pass

    def json_request(self, method, target_url_rule, token=None, *args, **kwargs):
        """
        Helper for json request

        Args:
            method (func): Request method
            target_url_rule (str): URL rule for request
            token (str) : JWT or OAuth's access token with prefix(Bearer, JWT, ...)

        Returns:
            Response
        """
        if token is None:
            token = self.access_token

        data = kwargs.pop('data')

        return method(
            target_url_rule,
            data=ujson.dumps(data) if data else None,
            content_type='application/json',
            headers={'Authorization': token},
            *args,
            **kwargs
        )

    def request(self, method, target_url_rule, token=None, *args, **kwargs):
        """
        Helper for common request

        Args:
            method (func): Request method
            target_url_rule (str): URL rule for request
            token (str) : JWT or OAuth's access token with prefix(Bearer, JWT, ...)

        Returns:
            Response
        """
        if token is None:
            token = self.access_token

        return method(
            target_url_rule,
            headers={'Authorization': token},
            *args,
            **kwargs
        )

    def decode_response_data(self, resp):
        return resp.data.decode()

    def get_response_data(self, resp):
        return ujson.loads(self.decode_response_data(resp))