mirror of
https://github.com/valitydev/redash.git
synced 2024-11-07 09:28:51 +00:00
72 lines
3.2 KiB
Python
72 lines
3.2 KiB
Python
from tests import BaseTestCase
|
|
from redash import settings
|
|
|
|
|
|
class TestEmbedVisualization(BaseTestCase):
|
|
def test_sucesss(self):
|
|
vis = self.factory.create_visualization()
|
|
vis.query.latest_query_data = self.factory.create_query_result()
|
|
vis.query.save()
|
|
|
|
res = self.make_request("get", "/embed/query/{}/visualization/{}".format(vis.query.id, vis.id), is_json=False)
|
|
self.assertEqual(res.status_code, 200)
|
|
|
|
def test_parameters_on_embeds(self):
|
|
previous = settings.ALLOW_PARAMETERS_IN_EMBEDS
|
|
# set configuration
|
|
settings.ALLOW_PARAMETERS_IN_EMBEDS = True
|
|
|
|
try:
|
|
vis = self.factory.create_visualization_with_params()
|
|
param1_name = "param1"
|
|
param1_value = "12345"
|
|
|
|
res = self.make_request("get", "/embed/query/{}/visualization/{}?p_{}={}".format(vis.query.id, vis.id, param1_name, param1_value), is_json=False)
|
|
|
|
# Currently we are expecting a 503 error which indicates that
|
|
# the database is unavailable. This ensures that the code in embed.py
|
|
# reaches the point where a DB query is made, where we then fail
|
|
# intentionally (because DB connection is not available in the tests).
|
|
self.assertEqual(res.status_code, 503)
|
|
|
|
# run embed query with maxAge to test caching
|
|
res = self.make_request("get", "/embed/query/{}/visualization/{}?p_{}={}&maxAge=60".format(vis.query.id, vis.id, param1_name, param1_value), is_json=False)
|
|
# If the 'maxAge' parameter is set and the query fails (because DB connection
|
|
# is not available in the tests), we're expecting a 404 error here.
|
|
self.assertEqual(res.status_code, 404)
|
|
|
|
finally:
|
|
# reset configuration
|
|
settings.ALLOW_PARAMETERS_IN_EMBEDS = previous
|
|
|
|
|
|
class TestPublicDashboard(BaseTestCase):
|
|
def test_success(self):
|
|
dashboard = self.factory.create_dashboard()
|
|
api_key = self.factory.create_api_key(object=dashboard)
|
|
|
|
res = self.make_request('get', '/public/dashboards/{}'.format(api_key.api_key), user=False, is_json=False)
|
|
self.assertEqual(res.status_code, 200)
|
|
|
|
def test_works_for_logged_in_user(self):
|
|
dashboard = self.factory.create_dashboard()
|
|
api_key = self.factory.create_api_key(object=dashboard)
|
|
|
|
res = self.make_request('get', '/public/dashboards/{}'.format(api_key.api_key), is_json=False)
|
|
self.assertEqual(res.status_code, 200)
|
|
|
|
def test_bad_token(self):
|
|
res = self.make_request('get', '/public/dashboards/bad-token', user=False, is_json=False)
|
|
self.assertEqual(res.status_code, 302)
|
|
|
|
def test_inactive_token(self):
|
|
dashboard = self.factory.create_dashboard()
|
|
api_key = self.factory.create_api_key(object=dashboard, active=False)
|
|
res = self.make_request('get', '/public/dashboards/{}'.format(api_key.api_key), user=False, is_json=False)
|
|
self.assertEqual(res.status_code, 302)
|
|
|
|
# Not relevant for now, as tokens in api_keys table are only created for dashboards. Once this changes, we should
|
|
# add this test.
|
|
# def test_token_doesnt_belong_to_dashboard(self):
|
|
# pass
|