mirror of
https://github.com/viq/NewsBlur.git
synced 2025-04-13 09:38:09 +00:00
25 lines
789 B
Python
25 lines
789 B
Python
from django.test.simple import DjangoTestSuiteRunner
|
|
from django.test import TransactionTestCase
|
|
|
|
from mongoengine import connect
|
|
|
|
class TestRunner(DjangoTestSuiteRunner):
|
|
def setup_databases(self, **kwargs):
|
|
db_name = 'newsblur_test'
|
|
connect(db_name)
|
|
print 'Creating test-database: ' + db_name
|
|
|
|
return super(TestRunner, self).setup_databases(**kwargs)
|
|
|
|
def teardown_databases(self, db_name, **kwargs):
|
|
import pymongo
|
|
conn = pymongo.MongoClient()
|
|
db_name = 'newsblur_test'
|
|
conn.drop_database(db_name)
|
|
print 'Dropping test-database: ' + db_name
|
|
return super(TestRunner, self).teardown_databases(db_name, **kwargs)
|
|
|
|
|
|
# class TestCase(TransactionTestCase):
|
|
# def _fixture_setup(self):
|
|
# pass
|