nyaa/tests/__init__.py

""" Sets up helper class for testing """

import os.path as op
import unittest

from nyaa import create_app

USE_MYSQL = True


class NyaaTestCase(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        app = create_app('config')
        app.config['TESTING'] = True

        # Use a seperate database for testing
        # if USE_MYSQL:
        #     cls.db_name = 'nyaav2_tests'
        #     db_uri = 'mysql://root:@localhost/{}?charset=utf8mb4'.format(cls.db_name)
        # else:
        #     cls.db_name = op.abspath(op.join(op.dirname(__file__)), 'test.db')
        #     db_uri = 'sqlite:///{}?check_same_thread=False'.format(cls.db_name)

        # if not os.environ.get('TRAVIS'):  # Travis doesn't need a seperate DB
        #     app.config['USE_MYSQL'] = USE_MYSQL
        #     app.config['SQLALCHEMY_DATABASE_URI'] = db_uri

        cls.app = app
        cls.app_context = app.app_context()
        cls.request_context = app.test_request_context

        with cls.app_context:
            cls.client = app.test_client()

    @classmethod
    def tearDownClass(cls):
        with cls.app_context:
            pass