datahub/metadata-ingestion/tests/unit/test_parsing_util.py
Aseem Bansal c5f1d2c9bd
feat(ingestion): snowflake, bigquery - enhancements to log and bugfix (#4442)
feat(ingestion): add logging for snowflake, bigquery
2022-03-21 09:50:36 -07:00

18 lines
621 B
Python

from datahub.utilities.parsing_util import (
get_first_missing_key,
get_first_missing_key_any,
)
def test_get_missing_key():
assert get_first_missing_key({}, [""]) == ""
assert get_first_missing_key({"a": 1}, ["a"]) is None
assert get_first_missing_key({"a": {"b": 1}}, ["a", "b"]) is None
assert get_first_missing_key({"a": {"b": 1}}, ["a", "c"]) == "c"
assert get_first_missing_key({"a": ["b", "c", "d"]}, ["a", "c"]) == "c"
def test_get_missing_key_any():
assert get_first_missing_key_any({}, ["a"]) == "a"
assert get_first_missing_key_any({"a": 1, "b": 2}, ["a", "b"]) is None