2017-11-07 23:23:40 -05:00
|
|
|
from datetime import date, datetime, time, timedelta
|
2017-01-05 12:56:07 -05:00
|
|
|
from decimal import Decimal
|
|
|
|
import unittest
|
|
|
|
import uuid
|
|
|
|
|
2017-10-04 15:43:29 -04:00
|
|
|
from mock import patch
|
2017-01-05 12:56:07 -05:00
|
|
|
import numpy
|
|
|
|
|
2018-03-12 01:07:51 -04:00
|
|
|
from superset.exceptions import SupersetException
|
2018-10-16 20:59:34 -04:00
|
|
|
from superset.utils.core import (
|
2018-07-16 17:27:40 -04:00
|
|
|
base_json_conv,
|
2018-07-31 16:52:20 -04:00
|
|
|
convert_legacy_filters_into_adhoc,
|
2018-07-16 17:27:40 -04:00
|
|
|
datetime_f,
|
|
|
|
get_since_until,
|
|
|
|
json_int_dttm_ser,
|
|
|
|
json_iso_dttm_ser,
|
|
|
|
JSONEncodedDict,
|
|
|
|
memoized,
|
|
|
|
merge_extra_filters,
|
|
|
|
merge_request_params,
|
|
|
|
parse_human_timedelta,
|
|
|
|
validate_json,
|
|
|
|
zlib_compress,
|
|
|
|
zlib_decompress_to_string,
|
2017-11-07 23:23:40 -05:00
|
|
|
)
|
2017-01-05 12:56:07 -05:00
|
|
|
|
2017-11-10 20:52:34 -05:00
|
|
|
|
2018-07-16 17:27:40 -04:00
|
|
|
def mock_parse_human_datetime(s):
|
|
|
|
if s in ['now', 'today']:
|
|
|
|
return datetime(2016, 11, 7)
|
|
|
|
elif s == 'yesterday':
|
|
|
|
return datetime(2016, 11, 6)
|
|
|
|
elif s == 'tomorrow':
|
|
|
|
return datetime(2016, 11, 8)
|
|
|
|
elif s == 'Last year':
|
|
|
|
return datetime(2015, 11, 7)
|
|
|
|
elif s == 'Last 5 months':
|
|
|
|
return datetime(2016, 6, 7)
|
|
|
|
elif s == 'Next 5 months':
|
|
|
|
return datetime(2017, 4, 7)
|
|
|
|
elif s in ['5 days', '5 days ago']:
|
|
|
|
return datetime(2016, 11, 2)
|
|
|
|
elif s == '2018-01-01T00:00:00':
|
|
|
|
return datetime(2018, 1, 1)
|
|
|
|
elif s == '2018-12-31T23:59:59':
|
|
|
|
return datetime(2018, 12, 31, 23, 59, 59)
|
|
|
|
|
|
|
|
|
2018-07-31 16:52:20 -04:00
|
|
|
def mock_to_adhoc(filt, expressionType='SIMPLE', clause='where'):
|
|
|
|
result = {
|
|
|
|
'clause': clause.upper(),
|
|
|
|
'expressionType': expressionType,
|
|
|
|
}
|
|
|
|
|
|
|
|
if expressionType == 'SIMPLE':
|
|
|
|
result.update({
|
|
|
|
'comparator': filt['val'],
|
|
|
|
'operator': filt['op'],
|
|
|
|
'subject': filt['col'],
|
|
|
|
})
|
|
|
|
elif expressionType == 'SQL':
|
|
|
|
result.update({
|
|
|
|
'sqlExpression': filt[clause],
|
|
|
|
})
|
|
|
|
|
|
|
|
return result
|
|
|
|
|
|
|
|
|
2017-01-05 12:56:07 -05:00
|
|
|
class UtilsTestCase(unittest.TestCase):
|
|
|
|
def test_json_int_dttm_ser(self):
|
|
|
|
dttm = datetime(2020, 1, 1)
|
|
|
|
ts = 1577836800000.0
|
|
|
|
assert json_int_dttm_ser(dttm) == ts
|
|
|
|
assert json_int_dttm_ser(date(2020, 1, 1)) == ts
|
|
|
|
assert json_int_dttm_ser(datetime(1970, 1, 1)) == 0
|
|
|
|
assert json_int_dttm_ser(date(1970, 1, 1)) == 0
|
|
|
|
assert json_int_dttm_ser(dttm + timedelta(milliseconds=1)) == (ts + 1)
|
|
|
|
|
|
|
|
with self.assertRaises(TypeError):
|
2017-11-14 00:06:51 -05:00
|
|
|
json_int_dttm_ser('this is not a date')
|
2017-01-05 12:56:07 -05:00
|
|
|
|
|
|
|
def test_json_iso_dttm_ser(self):
|
|
|
|
dttm = datetime(2020, 1, 1)
|
|
|
|
dt = date(2020, 1, 1)
|
|
|
|
t = time()
|
|
|
|
assert json_iso_dttm_ser(dttm) == dttm.isoformat()
|
|
|
|
assert json_iso_dttm_ser(dt) == dt.isoformat()
|
|
|
|
assert json_iso_dttm_ser(t) == t.isoformat()
|
|
|
|
|
|
|
|
with self.assertRaises(TypeError):
|
2017-11-14 00:06:51 -05:00
|
|
|
json_iso_dttm_ser('this is not a date')
|
2017-01-05 12:56:07 -05:00
|
|
|
|
|
|
|
def test_base_json_conv(self):
|
2017-02-08 14:52:58 -05:00
|
|
|
assert isinstance(base_json_conv(numpy.bool_(1)), bool) is True
|
|
|
|
assert isinstance(base_json_conv(numpy.int64(1)), int) is True
|
|
|
|
assert isinstance(base_json_conv(set([1])), list) is True
|
|
|
|
assert isinstance(base_json_conv(Decimal('1.0')), float) is True
|
|
|
|
assert isinstance(base_json_conv(uuid.uuid4()), str) is True
|
2017-01-05 12:56:07 -05:00
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.datetime')
|
2017-01-05 12:56:07 -05:00
|
|
|
def test_parse_human_timedelta(self, mock_now):
|
|
|
|
mock_now.return_value = datetime(2016, 12, 1)
|
|
|
|
self.assertEquals(parse_human_timedelta('now'), timedelta(0))
|
2017-04-06 12:42:43 -04:00
|
|
|
|
|
|
|
def test_zlib_compression(self):
|
2017-11-14 00:06:51 -05:00
|
|
|
json_str = '{"test": 1}'
|
2017-04-06 12:42:43 -04:00
|
|
|
blob = zlib_compress(json_str)
|
|
|
|
got_str = zlib_decompress_to_string(blob)
|
|
|
|
self.assertEquals(json_str, got_str)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2017-10-04 15:43:29 -04:00
|
|
|
def test_merge_extra_filters(self):
|
|
|
|
# does nothing if no extra filters
|
|
|
|
form_data = {'A': 1, 'B': 2, 'c': 'test'}
|
|
|
|
expected = {'A': 1, 'B': 2, 'c': 'test'}
|
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
2017-11-15 00:12:26 -05:00
|
|
|
# empty extra_filters
|
2017-10-04 15:43:29 -04:00
|
|
|
form_data = {'A': 1, 'B': 2, 'c': 'test', 'extra_filters': []}
|
2018-07-31 16:52:20 -04:00
|
|
|
expected = {'A': 1, 'B': 2, 'c': 'test', 'adhoc_filters': []}
|
2017-10-04 15:43:29 -04:00
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
# copy over extra filters into empty filters
|
|
|
|
form_data = {'extra_filters': [
|
|
|
|
{'col': 'a', 'op': 'in', 'val': 'someval'},
|
2017-11-08 00:32:45 -05:00
|
|
|
{'col': 'B', 'op': '==', 'val': ['c1', 'c2']},
|
2017-10-04 15:43:29 -04:00
|
|
|
]}
|
2018-07-31 16:52:20 -04:00
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': 'someval',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
2017-10-04 15:43:29 -04:00
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
# adds extra filters to existing filters
|
2018-07-31 16:52:20 -04:00
|
|
|
form_data = {
|
|
|
|
'extra_filters': [
|
|
|
|
{'col': 'a', 'op': 'in', 'val': 'someval'},
|
|
|
|
{'col': 'B', 'op': '==', 'val': ['c1', 'c2']},
|
|
|
|
],
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['G1', 'g2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '!=',
|
|
|
|
'subject': 'D',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['G1', 'g2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '!=',
|
|
|
|
'subject': 'D',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': 'someval',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
2017-10-04 15:43:29 -04:00
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
# adds extra filters to existing filters and sets time options
|
|
|
|
form_data = {'extra_filters': [
|
2018-07-16 17:27:40 -04:00
|
|
|
{'col': '__time_range', 'op': 'in', 'val': '1 year ago :'},
|
2017-10-04 15:43:29 -04:00
|
|
|
{'col': '__time_col', 'op': 'in', 'val': 'birth_year'},
|
|
|
|
{'col': '__time_grain', 'op': 'in', 'val': 'years'},
|
|
|
|
{'col': 'A', 'op': 'like', 'val': 'hello'},
|
|
|
|
{'col': '__time_origin', 'op': 'in', 'val': 'now'},
|
|
|
|
{'col': '__granularity', 'op': 'in', 'val': '90 seconds'},
|
|
|
|
]}
|
|
|
|
expected = {
|
2018-07-31 16:52:20 -04:00
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': 'hello',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'like',
|
|
|
|
'subject': 'A',
|
|
|
|
},
|
|
|
|
],
|
2018-07-16 17:27:40 -04:00
|
|
|
'time_range': '1 year ago :',
|
2017-10-04 15:43:29 -04:00
|
|
|
'granularity_sqla': 'birth_year',
|
|
|
|
'time_grain_sqla': 'years',
|
|
|
|
'granularity': '90 seconds',
|
|
|
|
'druid_time_origin': 'now',
|
|
|
|
}
|
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2017-11-11 00:33:31 -05:00
|
|
|
def test_merge_extra_filters_ignores_empty_filters(self):
|
|
|
|
form_data = {'extra_filters': [
|
|
|
|
{'col': 'a', 'op': 'in', 'val': ''},
|
|
|
|
{'col': 'B', 'op': '==', 'val': []},
|
|
|
|
]}
|
2018-07-31 16:52:20 -04:00
|
|
|
expected = {'adhoc_filters': []}
|
2017-11-11 00:33:31 -05:00
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2017-12-20 19:22:43 -05:00
|
|
|
def test_merge_extra_filters_ignores_nones(self):
|
|
|
|
form_data = {
|
2018-07-31 16:52:20 -04:00
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': '',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': None,
|
|
|
|
},
|
2017-12-20 19:22:43 -05:00
|
|
|
],
|
|
|
|
'extra_filters': [
|
|
|
|
{'col': 'B', 'op': '==', 'val': []},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
expected = {
|
2018-07-31 16:52:20 -04:00
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': '',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': None,
|
|
|
|
},
|
2017-12-20 19:22:43 -05:00
|
|
|
],
|
|
|
|
}
|
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2017-11-11 00:33:31 -05:00
|
|
|
def test_merge_extra_filters_ignores_equal_filters(self):
|
|
|
|
form_data = {
|
|
|
|
'extra_filters': [
|
|
|
|
{'col': 'a', 'op': 'in', 'val': 'someval'},
|
|
|
|
{'col': 'B', 'op': '==', 'val': ['c1', 'c2']},
|
|
|
|
],
|
2018-07-31 16:52:20 -04:00
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': 'someval',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': 'someval',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
2017-11-11 00:33:31 -05:00
|
|
|
],
|
|
|
|
}
|
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2017-11-11 00:33:31 -05:00
|
|
|
def test_merge_extra_filters_merges_different_val_types(self):
|
|
|
|
form_data = {
|
|
|
|
'extra_filters': [
|
|
|
|
{'col': 'a', 'op': 'in', 'val': ['g1', 'g2']},
|
|
|
|
{'col': 'B', 'op': '==', 'val': ['c1', 'c2']},
|
|
|
|
],
|
2018-07-31 16:52:20 -04:00
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': 'someval',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': 'someval',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['g1', 'g2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
2017-11-11 00:33:31 -05:00
|
|
|
],
|
|
|
|
}
|
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
form_data = {
|
|
|
|
'extra_filters': [
|
|
|
|
{'col': 'a', 'op': 'in', 'val': 'someval'},
|
|
|
|
{'col': 'B', 'op': '==', 'val': ['c1', 'c2']},
|
|
|
|
],
|
2018-07-31 16:52:20 -04:00
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['g1', 'g2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['g1', 'g2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': 'someval',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
2017-11-11 00:33:31 -05:00
|
|
|
],
|
|
|
|
}
|
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2017-11-11 00:33:31 -05:00
|
|
|
def test_merge_extra_filters_adds_unequal_lists(self):
|
|
|
|
form_data = {
|
|
|
|
'extra_filters': [
|
|
|
|
{'col': 'a', 'op': 'in', 'val': ['g1', 'g2', 'g3']},
|
|
|
|
{'col': 'B', 'op': '==', 'val': ['c1', 'c2', 'c3']},
|
|
|
|
],
|
2018-07-31 16:52:20 -04:00
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['g1', 'g2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['g1', 'g2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['g1', 'g2', 'g3'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': ['c1', 'c2', 'c3'],
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'B',
|
|
|
|
},
|
2017-11-11 00:33:31 -05:00
|
|
|
],
|
|
|
|
}
|
|
|
|
merge_extra_filters(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-03-08 16:19:41 -05:00
|
|
|
def test_merge_request_params(self):
|
|
|
|
form_data = {
|
|
|
|
'since': '2000',
|
|
|
|
'until': 'now',
|
|
|
|
}
|
|
|
|
url_params = {
|
|
|
|
'form_data': form_data,
|
|
|
|
'dashboard_ids': '(1,2,3,4,5)',
|
|
|
|
}
|
|
|
|
merge_request_params(form_data, url_params)
|
|
|
|
self.assertIn('url_params', form_data.keys())
|
|
|
|
self.assertIn('dashboard_ids', form_data['url_params'])
|
|
|
|
self.assertNotIn('form_data', form_data.keys())
|
|
|
|
|
2017-09-26 12:03:03 -04:00
|
|
|
def test_datetime_f(self):
|
2017-11-10 15:06:22 -05:00
|
|
|
self.assertEquals(
|
|
|
|
datetime_f(datetime(1990, 9, 21, 19, 11, 19, 626096)),
|
|
|
|
'<nobr>1990-09-21T19:11:19.626096</nobr>',
|
|
|
|
)
|
2017-09-26 12:03:03 -04:00
|
|
|
self.assertEquals(len(datetime_f(datetime.now())), 28)
|
|
|
|
self.assertEquals(datetime_f(None), '<nobr>None</nobr>')
|
|
|
|
iso = datetime.now().isoformat()[:10].split('-')
|
|
|
|
[a, b, c] = [int(v) for v in iso]
|
2017-11-11 00:33:31 -05:00
|
|
|
self.assertEquals(
|
|
|
|
datetime_f(datetime(a, b, c)), '<nobr>00:00:00</nobr>',
|
|
|
|
)
|
2017-09-26 12:03:03 -04:00
|
|
|
|
|
|
|
def test_json_encoded_obj(self):
|
|
|
|
obj = {'a': 5, 'b': ['a', 'g', 5]}
|
|
|
|
val = '{"a": 5, "b": ["a", "g", 5]}'
|
|
|
|
jsonObj = JSONEncodedDict()
|
|
|
|
resp = jsonObj.process_bind_param(obj, 'dialect')
|
|
|
|
self.assertIn('"a": 5', resp)
|
|
|
|
self.assertIn('"b": ["a", "g", 5]', resp)
|
|
|
|
self.assertEquals(jsonObj.process_result_value(val, 'dialect'), obj)
|
|
|
|
|
|
|
|
def test_validate_json(self):
|
|
|
|
invalid = '{"a": 5, "b": [1, 5, ["g", "h]]}'
|
|
|
|
with self.assertRaises(SupersetException):
|
|
|
|
validate_json(invalid)
|
2017-12-17 13:35:00 -05:00
|
|
|
|
|
|
|
def test_memoized_on_functions(self):
|
|
|
|
watcher = {'val': 0}
|
|
|
|
|
|
|
|
@memoized
|
|
|
|
def test_function(a, b, c):
|
|
|
|
watcher['val'] += 1
|
|
|
|
return a * b * c
|
|
|
|
result1 = test_function(1, 2, 3)
|
|
|
|
result2 = test_function(1, 2, 3)
|
|
|
|
self.assertEquals(result1, result2)
|
|
|
|
self.assertEquals(watcher['val'], 1)
|
|
|
|
|
|
|
|
def test_memoized_on_methods(self):
|
|
|
|
|
|
|
|
class test_class:
|
|
|
|
def __init__(self, num):
|
|
|
|
self.num = num
|
|
|
|
self.watcher = 0
|
|
|
|
|
|
|
|
@memoized
|
|
|
|
def test_method(self, a, b, c):
|
|
|
|
self.watcher += 1
|
|
|
|
return a * b * c * self.num
|
|
|
|
|
|
|
|
instance = test_class(5)
|
|
|
|
result1 = instance.test_method(1, 2, 3)
|
|
|
|
result2 = instance.test_method(1, 2, 3)
|
|
|
|
self.assertEquals(result1, result2)
|
|
|
|
self.assertEquals(instance.watcher, 1)
|
|
|
|
instance.num = 10
|
|
|
|
self.assertEquals(result2, instance.test_method(1, 2, 3))
|
|
|
|
|
|
|
|
def test_memoized_on_methods_with_watches(self):
|
|
|
|
|
|
|
|
class test_class:
|
|
|
|
def __init__(self, x, y):
|
|
|
|
self.x = x
|
|
|
|
self.y = y
|
|
|
|
self.watcher = 0
|
|
|
|
|
|
|
|
@memoized(watch=('x', 'y'))
|
|
|
|
def test_method(self, a, b, c):
|
|
|
|
self.watcher += 1
|
|
|
|
return a * b * c * self.x * self.y
|
|
|
|
|
|
|
|
instance = test_class(3, 12)
|
|
|
|
result1 = instance.test_method(1, 2, 3)
|
|
|
|
result2 = instance.test_method(1, 2, 3)
|
|
|
|
self.assertEquals(result1, result2)
|
|
|
|
self.assertEquals(instance.watcher, 1)
|
|
|
|
result3 = instance.test_method(2, 3, 4)
|
|
|
|
self.assertEquals(instance.watcher, 2)
|
|
|
|
result4 = instance.test_method(2, 3, 4)
|
|
|
|
self.assertEquals(instance.watcher, 2)
|
|
|
|
self.assertEquals(result3, result4)
|
|
|
|
self.assertNotEqual(result3, result1)
|
|
|
|
instance.x = 1
|
|
|
|
result5 = instance.test_method(2, 3, 4)
|
|
|
|
self.assertEqual(instance.watcher, 3)
|
|
|
|
self.assertNotEqual(result5, result4)
|
|
|
|
result6 = instance.test_method(2, 3, 4)
|
|
|
|
self.assertEqual(instance.watcher, 3)
|
|
|
|
self.assertEqual(result6, result5)
|
|
|
|
instance.x = 10
|
|
|
|
instance.y = 10
|
|
|
|
result7 = instance.test_method(2, 3, 4)
|
|
|
|
self.assertEqual(instance.watcher, 4)
|
|
|
|
self.assertNotEqual(result7, result6)
|
|
|
|
instance.x = 3
|
|
|
|
instance.y = 12
|
|
|
|
result8 = instance.test_method(1, 2, 3)
|
|
|
|
self.assertEqual(instance.watcher, 4)
|
|
|
|
self.assertEqual(result1, result8)
|
2018-07-16 17:27:40 -04:00
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.parse_human_datetime', mock_parse_human_datetime)
|
2018-07-16 17:27:40 -04:00
|
|
|
def test_get_since_until(self):
|
2018-11-07 20:23:52 -05:00
|
|
|
result = get_since_until()
|
2018-07-16 17:27:40 -04:00
|
|
|
expected = None, datetime(2016, 11, 7)
|
|
|
|
self.assertEqual(result, expected)
|
|
|
|
|
2018-11-07 20:23:52 -05:00
|
|
|
result = get_since_until(' : now')
|
2018-07-16 17:27:40 -04:00
|
|
|
expected = None, datetime(2016, 11, 7)
|
|
|
|
self.assertEqual(result, expected)
|
|
|
|
|
2018-11-07 20:23:52 -05:00
|
|
|
result = get_since_until('yesterday : tomorrow')
|
2018-07-16 17:27:40 -04:00
|
|
|
expected = datetime(2016, 11, 6), datetime(2016, 11, 8)
|
|
|
|
self.assertEqual(result, expected)
|
|
|
|
|
2018-11-07 20:23:52 -05:00
|
|
|
result = get_since_until('2018-01-01T00:00:00 : 2018-12-31T23:59:59')
|
2018-07-16 17:27:40 -04:00
|
|
|
expected = datetime(2018, 1, 1), datetime(2018, 12, 31, 23, 59, 59)
|
|
|
|
self.assertEqual(result, expected)
|
|
|
|
|
2018-11-07 20:23:52 -05:00
|
|
|
result = get_since_until('Last year')
|
2018-07-16 17:27:40 -04:00
|
|
|
expected = datetime(2015, 11, 7), datetime(2016, 11, 7)
|
|
|
|
self.assertEqual(result, expected)
|
|
|
|
|
2018-11-07 20:23:52 -05:00
|
|
|
result = get_since_until('Last 5 months')
|
2018-07-16 17:27:40 -04:00
|
|
|
expected = datetime(2016, 6, 7), datetime(2016, 11, 7)
|
|
|
|
self.assertEqual(result, expected)
|
|
|
|
|
2018-11-07 20:23:52 -05:00
|
|
|
result = get_since_until('Next 5 months')
|
2018-07-16 17:27:40 -04:00
|
|
|
expected = datetime(2016, 11, 7), datetime(2017, 4, 7)
|
|
|
|
self.assertEqual(result, expected)
|
|
|
|
|
2018-11-07 20:23:52 -05:00
|
|
|
result = get_since_until(since='5 days')
|
2018-07-16 17:27:40 -04:00
|
|
|
expected = datetime(2016, 11, 2), datetime(2016, 11, 7)
|
|
|
|
self.assertEqual(result, expected)
|
|
|
|
|
2018-11-07 20:23:52 -05:00
|
|
|
result = get_since_until(since='5 days ago', until='tomorrow')
|
2018-07-16 17:27:40 -04:00
|
|
|
expected = datetime(2016, 11, 2), datetime(2016, 11, 8)
|
|
|
|
self.assertEqual(result, expected)
|
2018-07-31 16:52:20 -04:00
|
|
|
|
2018-11-07 20:23:52 -05:00
|
|
|
result = get_since_until(time_range='yesterday : tomorrow', time_shift='1 day')
|
|
|
|
expected = datetime(2016, 11, 5), datetime(2016, 11, 7)
|
|
|
|
self.assertEqual(result, expected)
|
|
|
|
|
2018-11-19 14:29:46 -05:00
|
|
|
result = get_since_until(time_range='5 days : now')
|
|
|
|
expected = datetime(2016, 11, 2), datetime(2016, 11, 7)
|
|
|
|
self.assertEqual(result, expected)
|
|
|
|
|
2018-11-07 20:23:52 -05:00
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
get_since_until(time_range='tomorrow : yesterday')
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2018-07-31 16:52:20 -04:00
|
|
|
def test_convert_legacy_filters_into_adhoc_where(self):
|
|
|
|
form_data = {
|
|
|
|
'where': 'a = 1',
|
|
|
|
}
|
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'expressionType': 'SQL',
|
|
|
|
'sqlExpression': 'a = 1',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
convert_legacy_filters_into_adhoc(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2018-07-31 16:52:20 -04:00
|
|
|
def test_convert_legacy_filters_into_adhoc_filters(self):
|
|
|
|
form_data = {
|
|
|
|
'filters': [{'col': 'a', 'op': 'in', 'val': 'someval'}],
|
|
|
|
}
|
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'comparator': 'someval',
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': 'in',
|
|
|
|
'subject': 'a',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
convert_legacy_filters_into_adhoc(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2018-07-31 16:52:20 -04:00
|
|
|
def test_convert_legacy_filters_into_adhoc_having(self):
|
|
|
|
form_data = {
|
|
|
|
'having': 'COUNT(1) = 1',
|
|
|
|
}
|
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'HAVING',
|
|
|
|
'expressionType': 'SQL',
|
|
|
|
'sqlExpression': 'COUNT(1) = 1',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
convert_legacy_filters_into_adhoc(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2018-07-31 16:52:20 -04:00
|
|
|
def test_convert_legacy_filters_into_adhoc_having_filters(self):
|
|
|
|
form_data = {
|
|
|
|
'having_filters': [{'col': 'COUNT(1)', 'op': '==', 'val': 1}],
|
|
|
|
}
|
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'HAVING',
|
|
|
|
'comparator': 1,
|
|
|
|
'expressionType': 'SIMPLE',
|
|
|
|
'operator': '==',
|
|
|
|
'subject': 'COUNT(1)',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
convert_legacy_filters_into_adhoc(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2018-08-10 13:22:06 -04:00
|
|
|
def test_convert_legacy_filters_into_adhoc_present_and_empty(self):
|
2018-07-31 16:52:20 -04:00
|
|
|
form_data = {
|
|
|
|
'adhoc_filters': [],
|
2018-08-10 13:22:06 -04:00
|
|
|
'where': 'a = 1',
|
|
|
|
}
|
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'expressionType': 'SQL',
|
|
|
|
'sqlExpression': 'a = 1',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
|
|
|
convert_legacy_filters_into_adhoc(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|
|
|
|
|
2018-10-16 20:59:34 -04:00
|
|
|
@patch('superset.utils.core.to_adhoc', mock_to_adhoc)
|
2018-08-10 13:22:06 -04:00
|
|
|
def test_convert_legacy_filters_into_adhoc_present_and_nonempty(self):
|
|
|
|
form_data = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'expressionType': 'SQL',
|
|
|
|
'sqlExpression': 'a = 1',
|
|
|
|
},
|
|
|
|
],
|
2018-07-31 16:52:20 -04:00
|
|
|
'filters': [{'col': 'a', 'op': 'in', 'val': 'someval'}],
|
|
|
|
'having': 'COUNT(1) = 1',
|
|
|
|
'having_filters': [{'col': 'COUNT(1)', 'op': '==', 'val': 1}],
|
|
|
|
}
|
2018-08-10 13:22:06 -04:00
|
|
|
expected = {
|
|
|
|
'adhoc_filters': [
|
|
|
|
{
|
|
|
|
'clause': 'WHERE',
|
|
|
|
'expressionType': 'SQL',
|
|
|
|
'sqlExpression': 'a = 1',
|
|
|
|
},
|
|
|
|
],
|
|
|
|
}
|
2018-07-31 16:52:20 -04:00
|
|
|
convert_legacy_filters_into_adhoc(form_data)
|
|
|
|
self.assertEquals(form_data, expected)
|