QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-19 08:04:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

[format] Using Black (#7769)

Browse Source
This commit is contained in:
John Bodley
2019-06-25 13:34:48 -07:00
committed by GitHub
parent 0c9e6d0985
commit 5c58fd1802
270 changed files with 15592 additions and 14772 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
tests/datasource_tests.py
View File

@@ -22,25 +22,23 @@ from .fixtures.datasource import datasource_post
class DatasourceTests(SupersetTestCase):
def __init__(self, *args, **kwargs):
super(DatasourceTests, self).__init__(*args, **kwargs)
def test_external_metadata(self):
self.login(username='admin')
tbl = self.get_table_by_name('birth_names')
schema = tbl.schema or ''
self.login(username="admin")
tbl = self.get_table_by_name("birth_names")
schema = tbl.schema or ""
url = (
f'/datasource/external_metadata/table/{tbl.id}/?'
f'db_id={tbl.database.id}&'
f'table_name={tbl.table_name}&'
f'schema={schema}&'
f"/datasource/external_metadata/table/{tbl.id}/?"
f"db_id={tbl.database.id}&"
f"table_name={tbl.table_name}&"
f"schema={schema}&"
)
resp = self.get_json_resp(url)
col_names = {o.get('name') for o in resp}
col_names = {o.get("name") for o in resp}
self.assertEquals(
col_names,
{'sum_boys', 'num', 'gender', 'name', 'ds', 'state', 'sum_girls'},
col_names, {"sum_boys", "num", "gender", "name", "ds", "state", "sum_girls"}
)
def compare_lists(self, l1, l2, key):
@@ -48,38 +46,46 @@ class DatasourceTests(SupersetTestCase):
for obj1 in l1:
obj2 = l2_lookup.get(obj1.get(key))
for k in obj1:
if k not in 'id' and obj1.get(k):
if k not in "id" and obj1.get(k):
self.assertEquals(obj1.get(k), obj2.get(k))
def test_save(self):
self.login(username='admin')
tbl_id = self.get_table_by_name('birth_names').id
datasource_post['id'] = tbl_id
self.login(username="admin")
tbl_id = self.get_table_by_name("birth_names").id
datasource_post["id"] = tbl_id
data = dict(data=json.dumps(datasource_post))
resp = self.get_json_resp('/datasource/save/', data)
resp = self.get_json_resp("/datasource/save/", data)
for k in datasource_post:
if k == 'columns':
self.compare_lists(datasource_post[k], resp[k], 'column_name')
elif k == 'metrics':
self.compare_lists(datasource_post[k], resp[k], 'metric_name')
if k == "columns":
self.compare_lists(datasource_post[k], resp[k], "column_name")
elif k == "metrics":
self.compare_lists(datasource_post[k], resp[k], "metric_name")
else:
self.assertEquals(resp[k], datasource_post[k])
def test_get_datasource(self):
self.login(username='admin')
tbl = self.get_table_by_name('birth_names')
url = f'/datasource/get/{tbl.type}/{tbl.id}/'
self.login(username="admin")
tbl = self.get_table_by_name("birth_names")
url = f"/datasource/get/{tbl.type}/{tbl.id}/"
resp = self.get_json_resp(url)
self.assertEquals(resp.get('type'), 'table')
col_names = {o.get('column_name') for o in resp['columns']}
self.assertEquals(resp.get("type"), "table")
col_names = {o.get("column_name") for o in resp["columns"]}
self.assertEquals(
col_names,
{'sum_boys', 'num', 'gender', 'name', 'ds', 'state',
'sum_girls', 'num_california'},
{
"sum_boys",
"num",
"gender",
"name",
"ds",
"state",
"sum_girls",
"num_california",
},
)
def test_get_datasource_failed(self):
self.login(username='admin')
url = f'/datasource/get/druid/500000/'
self.login(username="admin")
url = f"/datasource/get/druid/500000/"
resp = self.get_json_resp(url)
self.assertEquals(resp.get('error'), 'This datasource does not exist')
self.assertEquals(resp.get("error"), "This datasource does not exist")