mirror of
https://github.com/getredash/redash.git
synced 2026-05-14 19:00:06 -04:00
c66f63d7a5
* avoid Query's updated_at from changing when it is linked to new query results * move comment to previous line * move QueryResult tests to their own module * add test which verifies that updated_at is not changed on query data updates * tests were false positives - they compared HH:MM:SS, but that never changed because the original time was 1 week behind. * remove redundant constructor * remove hack and use a proper event to prevent updated_at from changing * use self.assertEqual instead of assert
70 lines
2.8 KiB
Python
70 lines
2.8 KiB
Python
#encoding: utf8
|
|
import datetime
|
|
|
|
from tests import BaseTestCase
|
|
|
|
from redash import models
|
|
from redash.utils import utcnow
|
|
|
|
|
|
class QueryResultTest(BaseTestCase):
|
|
def test_get_latest_returns_none_if_not_found(self):
|
|
found_query_result = models.QueryResult.get_latest(self.factory.data_source, "SELECT 1", 60)
|
|
self.assertIsNone(found_query_result)
|
|
|
|
def test_get_latest_returns_when_found(self):
|
|
qr = self.factory.create_query_result()
|
|
found_query_result = models.QueryResult.get_latest(qr.data_source, qr.query_text, 60)
|
|
|
|
self.assertEqual(qr, found_query_result)
|
|
|
|
def test_get_latest_doesnt_return_query_from_different_data_source(self):
|
|
qr = self.factory.create_query_result()
|
|
data_source = self.factory.create_data_source()
|
|
found_query_result = models.QueryResult.get_latest(data_source, qr.query_text, 60)
|
|
|
|
self.assertIsNone(found_query_result)
|
|
|
|
def test_get_latest_doesnt_return_if_ttl_expired(self):
|
|
yesterday = utcnow() - datetime.timedelta(days=1)
|
|
qr = self.factory.create_query_result(retrieved_at=yesterday)
|
|
|
|
found_query_result = models.QueryResult.get_latest(qr.data_source, qr.query_text, max_age=60)
|
|
|
|
self.assertIsNone(found_query_result)
|
|
|
|
def test_get_latest_returns_if_ttl_not_expired(self):
|
|
yesterday = utcnow() - datetime.timedelta(seconds=30)
|
|
qr = self.factory.create_query_result(retrieved_at=yesterday)
|
|
|
|
found_query_result = models.QueryResult.get_latest(qr.data_source, qr.query_text, max_age=120)
|
|
|
|
self.assertEqual(found_query_result, qr)
|
|
|
|
def test_get_latest_returns_the_most_recent_result(self):
|
|
yesterday = utcnow() - datetime.timedelta(seconds=30)
|
|
old_qr = self.factory.create_query_result(retrieved_at=yesterday)
|
|
qr = self.factory.create_query_result()
|
|
|
|
found_query_result = models.QueryResult.get_latest(qr.data_source, qr.query_text, 60)
|
|
|
|
self.assertEqual(found_query_result.id, qr.id)
|
|
|
|
def test_get_latest_returns_the_last_cached_result_for_negative_ttl(self):
|
|
yesterday = utcnow() + datetime.timedelta(days=-100)
|
|
very_old = self.factory.create_query_result(retrieved_at=yesterday)
|
|
|
|
yesterday = utcnow() + datetime.timedelta(days=-1)
|
|
qr = self.factory.create_query_result(retrieved_at=yesterday)
|
|
found_query_result = models.QueryResult.get_latest(qr.data_source, qr.query_text, -1)
|
|
|
|
self.assertEqual(found_query_result.id, qr.id)
|
|
|
|
def test_store_result_does_not_modify_query_update_at(self):
|
|
original_updated_at = utcnow() - datetime.timedelta(hours=1)
|
|
query = self.factory.create_query(updated_at=original_updated_at)
|
|
|
|
models.QueryResult.store_result(query.org_id, query.data_source, query.query_hash, query.query_text, "", 0, utcnow())
|
|
|
|
self.assertEqual(original_updated_at, query.updated_at)
|