Skip to content

Commit

Permalink
Added unit tests for endpoint
Browse files Browse the repository at this point in the history
  • Loading branch information
Mogball committed Sep 22, 2017
1 parent 2d63027 commit 977680b
Show file tree
Hide file tree
Showing 2 changed files with 38 additions and 1 deletion.
8 changes: 8 additions & 0 deletions tests/core_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -758,6 +758,14 @@ def test_slice_id_is_always_logged_correctly_on_ajax_request(self):
self.get_json_resp(slc_url)
self.assertEqual(1, qry.count())

def test_slice_query_endpoint(self):
# API endpoint for query string
self.login(username="admin")
slc = self.get_slice("Girls", db.session)
resp = self.get_resp('/superset/slice/{}/query/'.format(slc.id))
assert 'query' in resp
assert 'language' in resp
self.logout();

if __name__ == '__main__':
unittest.main()
31 changes: 30 additions & 1 deletion tests/utils_tests.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,16 @@
from datetime import datetime, date, timedelta, time
from decimal import Decimal
from superset.utils import (
json_int_dttm_ser, json_iso_dttm_ser, base_json_conv, parse_human_timedelta, zlib_compress, zlib_decompress_to_string
json_int_dttm_ser,
json_iso_dttm_ser,
base_json_conv,
parse_human_timedelta,
zlib_compress,
zlib_decompress_to_string,
datetime_f,
JSONEncodedDict,
validate_json,
SupersetException,
)
import unittest
import uuid
Expand Down Expand Up @@ -52,3 +61,23 @@ def test_zlib_compression(self):
got_str = zlib_decompress_to_string(blob)
self.assertEquals(json_str, got_str)

def test_datetime_f(self):
self.assertEquals(datetime_f(datetime(1990, 9, 21, 19, 11, 19, 626096)),
'<nobr>1990-09-21T19:11:19.626096</nobr>')
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]
self.assertEquals(datetime_f(datetime(a, b, c)), '<nobr>00:00:00</nobr>')

def test_json_encoded_obj(self):
obj = {'a': 5, 'b': ['a', 'g', 5]}
val = '{"a": 5, "b": ["a", "g", 5]}'
jsonObj = JSONEncodedDict()
self.assertEquals(jsonObj.process_bind_param(obj, 'dialect'), val)
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)

0 comments on commit 977680b

Please sign in to comment.