LBRYCommunity/comment-server
1
0
Fork 0
Code Issues18 Pull requests1 Projects Releases Packages Wiki Activity

fixes import statements

Browse source
This commit is contained in:
Oleg Silkin 2019-05-21 08:58:43 -04:00
parent 6230a8ae49
commit e9ba0cd843
4 changed files with 51 additions and 50 deletions

10
tests/testcase.py
View file

@ -4,8 +4,8 @@ from unittest.case import _Outcome
import asyncio
import lbry_comment_server.database as db
import schema.db_helpers as schema
from lbry_comment_server.database import obtain_connection
from schema.db_helpers import setup_database, teardown_database
from lbry_comment_server.settings import config
@ -120,11 +120,11 @@ class AsyncioTestCase(unittest.TestCase):
class DatabaseTestCase(unittest.TestCase):
def setUp(self) -> None:
super().setUp()
schema.setup_database(config['PATH']['TEST'])
self.conn = db.obtain_connection(config['PATH']['TEST'])
setup_database(config['PATH']['TEST'])
self.conn = obtain_connection(config['PATH']['TEST'])
def tearDown(self) -> None:
self.conn.close()
schema.teardown_database(config['PATH']['TEST'])
teardown_database(config['PATH']['TEST'])