mirror of
https://github.com/valitydev/redash.git
synced 2024-11-07 09:28:51 +00:00
67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
import os
|
|
os.environ['REDASH_REDIS_URL'] = "redis://localhost:6379/5"
|
|
# Use different url for Celery to avoid DB being cleaned up:
|
|
os.environ['REDASH_CELERY_BROKER'] = "redis://localhost:6379/6"
|
|
|
|
# Dummy values for oauth login
|
|
os.environ['REDASH_GOOGLE_CLIENT_ID'] = "dummy"
|
|
os.environ['REDASH_GOOGLE_CLIENT_SECRET'] = "dummy"
|
|
os.environ['REDASH_MULTI_ORG'] = "true"
|
|
|
|
|
|
import logging
|
|
from unittest import TestCase
|
|
import datetime
|
|
from redash import settings
|
|
from factories import Factory
|
|
|
|
settings.DATABASE_CONFIG = {
|
|
'name': 'circle_test',
|
|
'threadlocals': True
|
|
}
|
|
|
|
from redash import redis_connection
|
|
import redash.models
|
|
from tests.handlers import make_request
|
|
|
|
logging.disable("INFO")
|
|
logging.getLogger("metrics").setLevel("ERROR")
|
|
logging.getLogger('peewee').setLevel(logging.INFO)
|
|
|
|
|
|
class BaseTestCase(TestCase):
|
|
def setUp(self):
|
|
redash.models.create_db(True, True)
|
|
self.factory = Factory()
|
|
|
|
def tearDown(self):
|
|
redash.models.db.close_db(None)
|
|
redash.models.create_db(False, True)
|
|
redis_connection.flushdb()
|
|
|
|
def make_request(self, method, path, org=None, user=None, data=None, is_json=True):
|
|
if user is None:
|
|
user = self.factory.user
|
|
|
|
if org is None:
|
|
org = self.factory.org
|
|
|
|
if org is not False:
|
|
path = "/{}{}".format(org.slug, path)
|
|
|
|
return make_request(method, path, user, data, is_json)
|
|
|
|
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]))
|