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)"