fix testrunner issues

This commit is contained in:
Jonathan Math 2020-10-13 22:12:40 +07:00
parent 39518e265f
commit a81be828d0

View file

@ -1,14 +1,14 @@
from django.test.runner import DiscoverRunner from django.test.runner import DiscoverRunner
from django.test.utils import setup_databases from django.test.utils import setup_databases
from mongoengine import connect from mongoengine.connection import disconnect, connect
class TestRunner(DiscoverRunner): class TestRunner(DiscoverRunner):
def setup_databases(self, **kwargs): def setup_databases(self, **kwargs):
db_name = 'newsblur_test' db_name = 'newsblur_test'
disconnect()
connect(db_name) connect(db_name)
print('Creating test-database: ' + db_name) print('Creating test-database: ' + db_name)
return setup_databases(**kwargs) return setup_databases(self.verbosity, self.interactive, **kwargs)
def teardown_databases(self, old_config, **kwargs): def teardown_databases(self, old_config, **kwargs):
import pymongo import pymongo