mirror of
https://github.com/datahub-project/datahub.git
synced 2025-08-27 18:45:50 +00:00
26 lines
897 B
Python
26 lines
897 B
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
|
|
|
|
|
|
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_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"
|