mirror of
https://github.com/getredash/redash.git
synced 2025-12-30 05:00:12 -05:00
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
import os
|
|
os.environ['REDASH_REDIS_URL'] = "redis://localhost:6379/5"
|
|
|
|
import logging
|
|
from unittest import TestCase
|
|
import datetime
|
|
from redash import settings
|
|
|
|
settings.DATABASE_CONFIG = {
|
|
'name': 'circle_test',
|
|
'threadlocals': True
|
|
}
|
|
|
|
from redash import models, redis_connection
|
|
|
|
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)
|
|
redis_connection.flushdb()
|
|
|
|
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]))
|