mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-19 23:49:25 +00:00
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from datahub.emitter.rest_emitter import DatahubRestEmitter
|
|
|
|
MOCK_GMS_ENDPOINT = "http://fakegmshost:8080"
|
|
|
|
|
|
def test_datahub_rest_emitter_construction():
|
|
emitter = DatahubRestEmitter(MOCK_GMS_ENDPOINT)
|
|
assert emitter._connect_timeout_sec == emitter.DEFAULT_CONNECT_TIMEOUT_SEC
|
|
assert emitter._read_timeout_sec == emitter.DEFAULT_READ_TIMEOUT_SEC
|
|
assert emitter._retry_status_codes == emitter.DEFAULT_RETRY_STATUS_CODES
|
|
assert emitter._retry_max_times == emitter.DEFAULT_RETRY_MAX_TIMES
|
|
|
|
|
|
def test_datahub_rest_emitter_timeout_construction():
|
|
emitter = DatahubRestEmitter(
|
|
MOCK_GMS_ENDPOINT, connect_timeout_sec=2, read_timeout_sec=4
|
|
)
|
|
assert emitter._connect_timeout_sec == 2
|
|
assert emitter._read_timeout_sec == 4
|
|
|
|
|
|
def test_datahub_rest_emitter_retry_construction():
|
|
emitter = DatahubRestEmitter(
|
|
MOCK_GMS_ENDPOINT,
|
|
retry_status_codes=[418],
|
|
retry_max_times=42,
|
|
)
|
|
assert emitter._retry_status_codes == [418]
|
|
assert emitter._retry_max_times == 42
|
|
|
|
|
|
def test_datahub_rest_emitter_extra_params():
|
|
emitter = DatahubRestEmitter(
|
|
MOCK_GMS_ENDPOINT, extra_headers={"key1": "value1", "key2": "value2"}
|
|
)
|
|
assert emitter._session.headers.get("key1") == "value1"
|
|
assert emitter._session.headers.get("key2") == "value2"
|