mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-13 04:02:34 +00:00
26 lines
817 B
Python
26 lines
817 B
Python
|
import json
|
||
|
|
|
||
|
|
from datahub.ingestion.api.source import SourceReport
|
||
|
|
|
||
|
|
|
||
|
|
def test_report_to_string_unsampled():
|
||
|
|
source_report: SourceReport = SourceReport()
|
||
|
|
source_report.report_warning("key1", "problem 1")
|
||
|
|
source_report.report_failure("key2", "reason 2")
|
||
|
|
str = source_report.as_string()
|
||
|
|
assert "'warnings': {'key1': ['problem 1']}" in str
|
||
|
|
assert "'failures': {'key2': ['reason 2']}" in str
|
||
|
|
|
||
|
|
|
||
|
|
def test_report_to_string_sampled():
|
||
|
|
source_report = SourceReport()
|
||
|
|
for i in range(0, 100):
|
||
|
|
source_report.report_warning(f"key{i}", "Test message")
|
||
|
|
|
||
|
|
str = source_report.as_string()
|
||
|
|
assert "'sampled': '10 sampled of at most 100 entries.'" in str
|
||
|
|
str = source_report.as_json()
|
||
|
|
print(str)
|
||
|
|
report_as_dict = json.loads(str)
|
||
|
|
assert len(report_as_dict["warnings"]) == 11
|