forked from tableau/server-client-python
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_endpoint.py
57 lines (43 loc) · 2.1 KB
/
test_endpoint.py
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
from pathlib import Path
import unittest
import tableauserverclient as TSC
import requests_mock
ASSETS = Path(__file__).parent / "assets"
class TestEndpoint(unittest.TestCase):
def setUp(self) -> None:
self.server = TSC.Server("http://test/", use_server_version=False)
# Fake signin
self.server._site_id = "dad65087-b08b-4603-af4e-2887b8aafc67"
self.server._auth_token = "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM"
return super().setUp()
def test_get_request_stream(self) -> None:
url = "http://test/"
endpoint = TSC.server.Endpoint(self.server)
with requests_mock.mock() as m:
m.get(url, headers={"Content-Type": "application/octet-stream"})
response = endpoint.get_request(url, parameters={"stream": True})
self.assertFalse(response._content_consumed)
def test_binary_log_truncated(self):
class FakeResponse(object):
headers = {"Content-Type": "application/octet-stream"}
content = b"\x1337" * 1000
status_code = 200
endpoint = TSC.server.Endpoint(self.server)
server_response = FakeResponse()
log = endpoint.log_response_safely(server_response)
self.assertTrue(log.find("[Truncated File Contents]") > 0, log)
def test_set_user_agent_from_options_headers(self):
params = {"User-Agent": "1", "headers": {"User-Agent": "2"}}
result = TSC.server.Endpoint.set_user_agent(params)
# it should use the value under 'headers' if more than one is given
print(result)
print(result["headers"]["User-Agent"])
self.assertTrue(result["headers"]["User-Agent"] == "2")
def test_set_user_agent_from_options(self):
params = {"headers": {"User-Agent": "2"}}
result = TSC.server.Endpoint.set_user_agent(params)
self.assertTrue(result["headers"]["User-Agent"] == "2")
def test_set_user_agent_when_blank(self):
params = {"headers": {}}
result = TSC.server.Endpoint.set_user_agent(params)
self.assertTrue(result["headers"]["User-Agent"].startswith("Tableau Server Client"))