mirror of https://gitlab.com/SIGBUS/nyaa.git
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
""" 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
|