Skip to content

Commit

Permalink
formating tests.
Browse files Browse the repository at this point in the history
  • Loading branch information
zem360 committed Apr 22, 2024
1 parent 1444df2 commit 444ccd8
Showing 1 changed file with 10 additions and 8 deletions.
18 changes: 10 additions & 8 deletions tests/common/runtime/test_telemetry.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,11 +67,12 @@ def test_sentry_log_level() -> None:
)
def test_telemetry_endpoint(endpoint, write_key, expectation) -> None:
from dlt.common.runtime import segment

with expectation:
segment.init_segment(RunConfiguration(
dlthub_telemetry_endpoint=endpoint,
dlthub_telemetry_segment_write_key=write_key)
segment.init_segment(
RunConfiguration(
dlthub_telemetry_endpoint=endpoint, dlthub_telemetry_segment_write_key=write_key
)
)

assert segment._SEGMENT_ENDPOINT == endpoint
Expand All @@ -95,11 +96,12 @@ def test_telemetry_endpoint(endpoint, write_key, expectation) -> None:
)
def test_telemetry_endpoint_exceptions(endpoint, write_key, expectation) -> None:
from dlt.common.runtime import segment

with expectation:
segment.init_segment(RunConfiguration(
dlthub_telemetry_endpoint=endpoint,
dlthub_telemetry_segment_write_key=write_key)
segment.init_segment(
RunConfiguration(
dlthub_telemetry_endpoint=endpoint, dlthub_telemetry_segment_write_key=write_key
)
)


Expand Down

0 comments on commit 444ccd8

Please sign in to comment.