forked from tableau/server-client-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_metadata.py
More file actions
98 lines (77 loc) · 4.03 KB
/
test_metadata.py
File metadata and controls
98 lines (77 loc) · 4.03 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
import unittest
import os.path
import json
import requests_mock
import tableauserverclient as TSC
from tableauserverclient.server.endpoint.exceptions import GraphQLError
TEST_ASSET_DIR = os.path.join(os.path.dirname(__file__), 'assets')
METADATA_QUERY_SUCCESS = os.path.join(TEST_ASSET_DIR, 'metadata_query_success.json')
METADATA_QUERY_ERROR = os.path.join(TEST_ASSET_DIR, 'metadata_query_error.json')
EXPECTED_PAGED_DICT = os.path.join(TEST_ASSET_DIR, 'metadata_query_expected_dict.dict')
METADATA_PAGE_1 = os.path.join(TEST_ASSET_DIR, 'metadata_paged_1.json')
METADATA_PAGE_2 = os.path.join(TEST_ASSET_DIR, 'metadata_paged_2.json')
METADATA_PAGE_3 = os.path.join(TEST_ASSET_DIR, 'metadata_paged_3.json')
EXPECTED_DICT = {'publishedDatasources':
[{'id': '01cf92b2-2d17-b656-fc48-5c25ef6d5352', 'name': 'Batters (TestV1)'},
{'id': '020ae1cd-c356-f1ad-a846-b0094850d22a', 'name': 'SharePoint_List_sharepoint2010.test.tsi.lan'},
{'id': '061493a0-c3b2-6f39-d08c-bc3f842b44af', 'name': 'Batters_mongodb'},
{'id': '089fe515-ad2f-89bc-94bd-69f55f69a9c2', 'name': 'Sample - Superstore'}]}
EXPECTED_DICT_ERROR = [
{
"message": "Reached time limit of PT5S for query execution.",
"path": None,
"extensions": None
}
]
class MetadataTests(unittest.TestCase):
def setUp(self):
self.server = TSC.Server('http://test')
self.baseurl = self.server.metadata.baseurl
self.server.version = "3.5"
self.server._site_id = 'dad65087-b08b-4603-af4e-2887b8aafc67'
self.server._auth_token = 'j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM'
def test_metadata_query(self):
with open(METADATA_QUERY_SUCCESS, 'rb') as f:
response_json = json.loads(f.read().decode())
with requests_mock.mock() as m:
m.post(self.baseurl, json=response_json)
actual = self.server.metadata.query('fake query')
datasources = actual['data']
self.assertDictEqual(EXPECTED_DICT, datasources)
def test_paged_metadata_query(self):
with open(EXPECTED_PAGED_DICT, 'rb') as f:
expected = eval(f.read())
# prepare the 3 pages of results
with open(METADATA_PAGE_1, 'rb') as f:
result_1 = f.read().decode()
with open(METADATA_PAGE_2, 'rb') as f:
result_2 = f.read().decode()
with open(METADATA_PAGE_3, 'rb') as f:
result_3 = f.read().decode()
with requests_mock.mock() as m:
m.post(self.baseurl, [{'text': result_1, 'status_code': 200},
{'text': result_2, 'status_code': 200},
{'text': result_3, 'status_code': 200}])
# validation checks for endCursor and hasNextPage,
# but the query text doesn't matter for the test
actual = self.server.metadata.paginated_query('fake query endCursor hasNextPage',
variables={'first': 1, 'afterToken': None})
self.assertDictEqual(expected, actual)
def test_metadata_query_ignore_error(self):
with open(METADATA_QUERY_ERROR, 'rb') as f:
response_json = json.loads(f.read().decode())
with requests_mock.mock() as m:
m.post(self.baseurl, json=response_json)
actual = self.server.metadata.query('fake query')
datasources = actual['data']
self.assertNotEqual(actual.get('errors', None), None)
self.assertListEqual(EXPECTED_DICT_ERROR, actual['errors'])
self.assertDictEqual(EXPECTED_DICT, datasources)
def test_metadata_query_abort_on_error(self):
with open(METADATA_QUERY_ERROR, 'rb') as f:
response_json = json.loads(f.read().decode())
with requests_mock.mock() as m:
m.post(self.baseurl, json=response_json)
with self.assertRaises(GraphQLError) as e:
self.server.metadata.query('fake query', abort_on_error=True)
self.assertListEqual(e.error, EXPECTED_DICT_ERROR)