def test_lookup_relations():
test_relation = relation.Relation(
database='SNOW_DATABASE', schema="TEST_SCHEMA", name="TEST_RELATION", materialization=TABLE, attributes=[])
test_relation2 = relation.Relation(
database='hamburger', schema="socks", name="TEST_RELATION", materialization=TABLE, attributes=[])
pattern1 = dict(database="(?i)snow_.*", schema=".*", name=".*")
pattern2 = dict(database="SNOW_DATABASE",
schema='TEST_SCHEMA', name='TEST_RELATION')
pattern3 = dict(database="NO", schema='banana', name='.*')
assert relation.lookup_relations(
pattern1, [test_relation, test_relation2]) == [test_relation]
assert not relation.lookup_relations(pattern2, [test_relation, test_relation2]) == [
test_relation, test_relation2]
assert relation.lookup_single_relation(
pattern3, [test_relation, test_relation2]) == None
def test_at_least_one_full_pattern_match():
test_relation = relation.Relation(
database='TEST_DATABASE', schema="TEST_SCHEMA", name="TEST_RELATION", materialization=TABLE, attributes=[])
pattern1 = dict(database=".*", schema=".*", name=".*")
pattern2 = dict(database="TEST", schema='banana', name='.*')
pattern3 = dict(database="NO", schema='banana', name='.*')
assert relation.at_least_one_full_pattern_match(
test_relation, [pattern1, pattern2, pattern3])
assert not relation.at_least_one_full_pattern_match(
test_relation, [pattern2, pattern3])
def test_single_full_pattern_match():
test_relation = relation.Relation(
database='TEST_DATABASE', schema="TEST_SCHEMA", name="TEST_RELATION", materialization=TABLE, attributes=[])
pattern = dict(database=".*", schema=".*", name=".*")
assert relation.single_full_pattern_match(test_relation, pattern)
pattern = dict(database="TEST_DATABASE", schema=".*", name=".*")
assert relation.single_full_pattern_match(test_relation, pattern)
pattern = dict(database="(?i)test_.*", schema=".*", name=".*")
assert relation.single_full_pattern_match(test_relation, pattern)
pattern = dict(database="(?i)test_.*", schema="TEST_SCHEMA", name=".*")
assert relation.single_full_pattern_match(test_relation, pattern)
pattern = dict(database="test_.*", schema="TEST_SCHEMA", name=".*")
assert not relation.single_full_pattern_match(test_relation, pattern)