public-mirrors/NewsBlur-viq
1
0
Fork 0
mirror of https://github.com/viq/NewsBlur.git synced 2025-09-18 21:43:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix testrunner issues

Browse source
This commit is contained in:
Jonathan Math 2020-10-13 22:12:40 +07:00
parent 39518e265f
commit a81be828d0
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
utils/testrunner.py
View file

@ -1,21 +1,21 @@
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
conn = pymongo.MongoClient() conn = pymongo.MongoClient()
db_name = 'newsblur_test' db_name = 'newsblur_test'
conn.drop_database(db_name) conn.drop_database(db_name)
print ('Dropping test-database: %s' % db_name) print('Dropping test-database: %s' % db_name)
return super(TestRunner, self).teardown_databases(old_config, **kwargs) return super(TestRunner, self).teardown_databases(old_config, **kwargs)