mirror of
https://github.com/getredash/redash.git
synced 2025-12-19 17:37:19 -05:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import unittest
|
|
|
|
from redash.query_runner.big_query import BigQuery
|
|
|
|
|
|
class TestBigQueryQueryRunner(unittest.TestCase):
|
|
def test_annotate_query_with_use_query_annotation_option(self):
|
|
query_runner = BigQuery({"useQueryAnnotation": True})
|
|
|
|
self.assertTrue(query_runner.should_annotate_query)
|
|
|
|
metadata = {
|
|
"Username": "username",
|
|
"query_id": "adhoc",
|
|
"Job ID": "job-id",
|
|
"Query Hash": "query-hash",
|
|
"Scheduled": False,
|
|
}
|
|
|
|
query = "SELECT a FROM tbl"
|
|
expect = (
|
|
"/* Username: username, query_id: adhoc, "
|
|
"Query Hash: query-hash, "
|
|
"Scheduled: False */ SELECT a FROM tbl"
|
|
)
|
|
|
|
self.assertEqual(query_runner.annotate_query(query, metadata), expect)
|
|
|
|
def test_annotate_query_without_use_query_annotation_option(self):
|
|
query_runner = BigQuery({"useQueryAnnotation": False})
|
|
|
|
self.assertFalse(query_runner.should_annotate_query)
|
|
|
|
metadata = {
|
|
"Username": "user-name",
|
|
"query_id": "adhoc",
|
|
"Job ID": "job-id",
|
|
"Query Hash": "query-hash",
|
|
"Scheduled": False,
|
|
}
|
|
|
|
query = "SELECT a FROM tbl"
|
|
expect = query
|
|
|
|
self.assertEqual(query_runner.annotate_query(query, metadata), expect)
|