From a81be828d0609c98c7fb95428447503dd2011740 Mon Sep 17 00:00:00 2001 From: Jonathan Math Date: Tue, 13 Oct 2020 22:12:40 +0700 Subject: [PATCH] fix testrunner issues --- utils/testrunner.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/utils/testrunner.py b/utils/testrunner.py index b829f17da..b35e16140 100644 --- a/utils/testrunner.py +++ b/utils/testrunner.py @@ -1,21 +1,21 @@ from django.test.runner import DiscoverRunner from django.test.utils import setup_databases -from mongoengine import connect - +from mongoengine.connection import disconnect, connect class TestRunner(DiscoverRunner): def setup_databases(self, **kwargs): db_name = 'newsblur_test' + disconnect() connect(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): import pymongo conn = pymongo.MongoClient() db_name = 'newsblur_test' 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)