mirror of
https://github.com/getredash/redash.git
synced 2025-12-29 11:00:14 -05:00
37 lines
967 B
Python
37 lines
967 B
Python
import logging
|
|
from unittest import TestCase
|
|
import datetime
|
|
from redash import settings
|
|
settings.DATABASE_CONFIG = {
|
|
'name': 'circle_test',
|
|
'threadlocals': True
|
|
}
|
|
|
|
from redash import models
|
|
|
|
logging.getLogger('peewee').setLevel(logging.INFO)
|
|
|
|
|
|
class BaseTestCase(TestCase):
|
|
def setUp(self):
|
|
models.create_db(True, True)
|
|
models.init_db()
|
|
|
|
def tearDown(self):
|
|
models.db.close_db(None)
|
|
models.create_db(False, True)
|
|
|
|
def assertResponseEqual(self, expected, actual):
|
|
for k, v in expected.iteritems():
|
|
if isinstance(v, datetime.datetime) or isinstance(actual[k], datetime.datetime):
|
|
continue
|
|
|
|
if isinstance(v, list):
|
|
continue
|
|
|
|
if isinstance(v, dict):
|
|
self.assertResponseEqual(v, actual[k])
|
|
continue
|
|
|
|
self.assertEqual(v, actual[k], "{} not equal (expected: {}, actual: {}).".format(k, v, actual[k]))
|