def test_reference_from_friend(self, write_row):
generate(StringIO(reference_from_friend), {}, None)
a_values = find_row("A", {}, write_row.mock_calls)
b_values = find_row("B", {}, write_row.mock_calls)
id_a = a_values["id"]
reference_a = b_values["A_ref"]
assert f"A({id_a})" == reference_a
def test_ancestor_reference(self, write_row):
generate(StringIO(ancestor_reference), {}, None)
a_values = find_row("A", {}, write_row.mock_calls)
c_values = find_row("C", {}, write_row.mock_calls)
id_a = a_values["id"]
reference_a = c_values["A_ref"]
assert f"A({id_a})" == reference_a
def test_simple_parent_list_child(self, write_row):
generate(StringIO(simple_parent_list), {}, None)
a_values = find_row("A", {}, write_row.mock_calls)
b_values = find_row("B", {}, write_row.mock_calls)
id_a = a_values["id"]
reference_b = a_values["B"]
id_b = b_values["id"]
reference_a = b_values["A_ref"]
assert f"A({id_a})" == reference_a
assert f"B({id_b})" == reference_b
def test_forward_reference__tablename(self, write_row):
yaml = """
- object: A
fields:
B:
reference:
B
- object: B
fields:
A:
reference:
A
"""
generate(StringIO(yaml), {}, None)
a_values = find_row("A", {}, write_row.mock_calls)
b_values = find_row("B", {}, write_row.mock_calls)
assert a_values["B"] == "B(1)"
assert b_values["A"] == "A(1)"
