nyaa/tests/__init__.py

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

import os
import unittest

from nyaa import app

USE_MYSQL = True


class NyaaTestCase(unittest.TestCase):

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

        # 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 = os.path.join(os.path.abspath(os.path.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

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

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