mirror of
https://github.com/valitydev/redash.git
synced 2024-11-07 09:28:51 +00:00
f7b57fa580
This is one huge change for the permissions system and related: * (Backward incompatible:) Remove the table based permissions in favour of the new model. * Manage permission to view or query datasources based on groups. * Add the concept of Organization. It's irrelevant for most deployments, but allows for multi-tenant support in re:dash. * Replace ActivityLog with Event based rows (old data in activity_log table is retained). * Enforce permissions on the server-side. There were some permissions that were only enforced on the client side. This is no more. All permissions are enforced by the server. * Added new permission: 'super-admin' to access the status and Flask-Admin interface. * Make sure that html is never cached by the browser - this is to make sure that the browser will always ask for the new Javascript/CSS resources (if such are available).
91 lines
4.7 KiB
Python
91 lines
4.7 KiB
Python
import datetime
|
|
from mock import patch, call, ANY
|
|
from tests import BaseTestCase
|
|
from redash.utils import utcnow
|
|
from redash.tasks import refresh_queries
|
|
|
|
|
|
# TODO: this test should be split into two:
|
|
# 1. tests for Query.outdated_queries method
|
|
# 2. test for the refresh_query task
|
|
class TestRefreshQueries(BaseTestCase):
|
|
def test_enqueues_outdated_queries(self):
|
|
query = self.factory.create_query(schedule="60")
|
|
retrieved_at = utcnow() - datetime.timedelta(minutes=10)
|
|
query_result = self.factory.create_query_result(retrieved_at=retrieved_at, query=query.query,
|
|
query_hash=query.query_hash)
|
|
query.latest_query_data = query_result
|
|
query.save()
|
|
|
|
with patch('redash.tasks.QueryTask.add_task') as add_job_mock:
|
|
refresh_queries()
|
|
add_job_mock.assert_called_with(query.query, query.data_source, scheduled=True, metadata=ANY)
|
|
|
|
def test_skips_fresh_queries(self):
|
|
query = self.factory.create_query(schedule="1200")
|
|
retrieved_at = utcnow() - datetime.timedelta(minutes=10)
|
|
query_result = self.factory.create_query_result(retrieved_at=retrieved_at, query=query.query,
|
|
query_hash=query.query_hash)
|
|
|
|
with patch('redash.tasks.QueryTask.add_task') as add_job_mock:
|
|
refresh_queries()
|
|
self.assertFalse(add_job_mock.called)
|
|
|
|
def test_skips_queries_with_no_ttl(self):
|
|
query = self.factory.create_query(schedule=None)
|
|
retrieved_at = utcnow() - datetime.timedelta(minutes=10)
|
|
query_result = self.factory.create_query_result(retrieved_at=retrieved_at, query=query.query,
|
|
query_hash=query.query_hash)
|
|
|
|
with patch('redash.tasks.QueryTask.add_task') as add_job_mock:
|
|
refresh_queries()
|
|
self.assertFalse(add_job_mock.called)
|
|
|
|
def test_enqueues_query_only_once(self):
|
|
query = self.factory.create_query(schedule="60")
|
|
query2 = self.factory.create_query(schedule="60", query=query.query, query_hash=query.query_hash)
|
|
retrieved_at = utcnow() - datetime.timedelta(minutes=10)
|
|
query_result = self.factory.create_query_result(retrieved_at=retrieved_at, query=query.query,
|
|
query_hash=query.query_hash)
|
|
query.latest_query_data = query_result
|
|
query2.latest_query_data = query_result
|
|
query.save()
|
|
query2.save()
|
|
|
|
with patch('redash.tasks.QueryTask.add_task') as add_job_mock:
|
|
refresh_queries()
|
|
add_job_mock.assert_called_once_with(query.query, query.data_source, scheduled=True, metadata=ANY)#{'Query ID': query.id, 'Username': 'Scheduled'})
|
|
|
|
def test_enqueues_query_with_correct_data_source(self):
|
|
query = self.factory.create_query(schedule="60", data_source=self.factory.create_data_source())
|
|
query2 = self.factory.create_query(schedule="60", query=query.query, query_hash=query.query_hash)
|
|
retrieved_at = utcnow() - datetime.timedelta(minutes=10)
|
|
query_result = self.factory.create_query_result(retrieved_at=retrieved_at, query=query.query,
|
|
query_hash=query.query_hash)
|
|
query.latest_query_data = query_result
|
|
query2.latest_query_data = query_result
|
|
query.save()
|
|
query2.save()
|
|
|
|
with patch('redash.tasks.QueryTask.add_task') as add_job_mock:
|
|
refresh_queries()
|
|
add_job_mock.assert_has_calls([call(query2.query, query2.data_source, scheduled=True, metadata=ANY),
|
|
call(query.query, query.data_source, scheduled=True, metadata=ANY)],
|
|
any_order=True)
|
|
self.assertEquals(2, add_job_mock.call_count)
|
|
|
|
def test_enqueues_only_for_relevant_data_source(self):
|
|
query = self.factory.create_query(schedule="60")
|
|
query2 = self.factory.create_query(schedule="3600", query=query.query, query_hash=query.query_hash)
|
|
import psycopg2
|
|
retrieved_at = utcnow().replace(tzinfo=psycopg2.tz.FixedOffsetTimezone(offset=0, name=None)) - datetime.timedelta(minutes=10)
|
|
query_result = self.factory.create_query_result(retrieved_at=retrieved_at, query=query.query,
|
|
query_hash=query.query_hash)
|
|
query.latest_query_data = query_result
|
|
query2.latest_query_data = query_result
|
|
query.save()
|
|
query2.save()
|
|
|
|
with patch('redash.tasks.QueryTask.add_task') as add_job_mock:
|
|
refresh_queries()
|
|
add_job_mock.assert_called_once_with(query.query, query.data_source, scheduled=True, metadata=ANY) |