tests.views

Complexity

Total Complexity

8

Size/Duplication

Total Lines	56
Duplicated Lines	0 %

Importance

Changes

0

8 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
A	TCBase.request()	0	17	1
A	TCBase.get_response_data_as_json()	0	2	1
A	TCBase.decode_response_data()	0	2	1

import ujson
from unittest import TestCase as TC

from flask import Response

from app import create_app
from config.test import TestConfig

app = create_app(TestConfig)


class TCBase(TC):
    def __init__(self, *args, **kwargs):
        self.client = app.test_client()

        super(TCBase, self).__init__(*args, **kwargs)

    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 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
        """
        return method(
            target_url_rule,
            headers={'Authorization': token or self.access_token},
            *args,
            **kwargs
        )

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

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