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