mirror of
https://github.com/getredash/redash.git
synced 2026-03-21 16:00:09 -04:00
d333660473
* Fix python query runner crashing the GUI due to missing 'rows' and/or 'columns' in the JSON returned data. * Fix typo of previous commit. * Throw exception when python query runner has invalid result. * Update test_python.py --------- Co-authored-by: YuhengChen <yuheng.chen@imaygou.com>
104 lines
3.8 KiB
Python
104 lines
3.8 KiB
Python
from datetime import datetime
|
|
from unittest import TestCase
|
|
|
|
import mock
|
|
|
|
from redash.query_runner.python import Python
|
|
|
|
|
|
class TestPythonQueryRunner(TestCase):
|
|
def setUp(self):
|
|
self.python = Python({})
|
|
|
|
@mock.patch("datetime.datetime")
|
|
def test_print_in_query_string_success(self, mock_dt):
|
|
query_string = "print('test')"
|
|
mock_dt.utcnow = mock.Mock(return_value=datetime(1901, 12, 21))
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(result[0], '{"rows": [], "columns": [], "log": ["[1901-12-21T00:00:00] test"]}')
|
|
|
|
def test_empty_result(self):
|
|
query_string = "result={}"
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(result[0], None)
|
|
|
|
def test_none_result(self):
|
|
query_string = "result=None"
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(result[0], None)
|
|
|
|
def test_invalid_result_type_string(self):
|
|
query_string = "result='string'"
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(result[0], None)
|
|
|
|
def test_invalid_result_type_int(self):
|
|
query_string = "result=100"
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(result[0], None)
|
|
|
|
def test_invalid_result_missing_rows(self):
|
|
query_string = "result={'columns': []}"
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(result[0], None)
|
|
|
|
def test_invalid_result_not_list_rows(self):
|
|
query_string = "result={'rows': {}, 'columns': []}"
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(result[0], None)
|
|
|
|
def test_invalid_result_missing_columns(self):
|
|
query_string = "result={'rows': []}"
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(result[0], None)
|
|
|
|
def test_invalid_result_not_list_columns(self):
|
|
query_string = "result={'rows': [], 'columns': {}}"
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(result[0], None)
|
|
|
|
def test_valid_result_type(self):
|
|
query_string = (
|
|
"result="
|
|
'{"columns": [{"name": "col1", "type": TYPE_STRING},'
|
|
'{"name": "col2", "type": TYPE_INTEGER}],'
|
|
'"rows": [{"col1": "foo", "col2": 100},'
|
|
'{"col1": "bar", "col2": 200}]}'
|
|
)
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(
|
|
result[0],
|
|
'{"columns": [{"name": "col1", "type": "string"},'
|
|
' {"name": "col2", "type": "integer"}],'
|
|
' "rows": [{"col1": "foo", "col2": 100},'
|
|
' {"col1": "bar", "col2": 200}],'
|
|
' "log": []}',
|
|
)
|
|
|
|
@mock.patch("datetime.datetime")
|
|
def test_valid_result_type_with_print(self, mock_dt):
|
|
mock_dt.utcnow = mock.Mock(return_value=datetime(1901, 12, 21))
|
|
query_string = (
|
|
'print("test")\n'
|
|
"result="
|
|
'{"columns": [{"name": "col1", "type": TYPE_STRING},'
|
|
'{"name": "col2", "type": TYPE_INTEGER}],'
|
|
'"rows": [{"col1": "foo", "col2": 100},'
|
|
'{"col1": "bar", "col2": 200}]}'
|
|
)
|
|
result = self.python.run_query(query_string, "user")
|
|
self.assertEqual(
|
|
result[0],
|
|
'{"columns": [{"name": "col1", "type": "string"},'
|
|
' {"name": "col2", "type": "integer"}],'
|
|
' "rows": [{"col1": "foo", "col2": 100},'
|
|
' {"col1": "bar", "col2": 200}],'
|
|
' "log": ["[1901-12-21T00:00:00] test"]}',
|
|
)
|
|
|
|
|
|
class TestPython(TestCase):
|
|
def test_sorted_safe_builtins(self):
|
|
src = list(Python.safe_builtins)
|
|
assert src == sorted(src), "Python safe_builtins package not sorted."
|