def test_get_foreign_key(self):
"""Test get foreign key"""
# Setup
primary_key = 'a_primary_key'
fields = {
'a_field': {
'ref': {
'field': 'a_primary_key'
},
'name': 'a_field'
},
'p_field': {
'ref': {
'field': 'another_key_field'
},
'name': 'p_field'
}
}
metadata = Mock(spec=Metadata)
metadata.get_primary_key.return_value = primary_key
metadata.get_fields.return_value = fields
# Run
result = Metadata.get_foreign_key(metadata, 'parent', 'child')
# Asserts
assert result == 'a_field'
metadata.get_primary_key.assert_called_once_with('parent')
metadata.get_fields.assert_called_once_with('child')
def test_get_foreign_key(self):
"""Test get foreign key"""
# Setup
primary_key = 'pk'
fields = {
'a_field': {
'ref': {
'field': 'pk'
},
'name': 'a_field'
},
'p_field': {
'ref': {
'field': 'kk'
},
'name': 'p_field'
}
}
# Run
metadata = Mock()
metadata.get_primary_key.return_value = primary_key
metadata.get_fields.return_value = fields
parent = 'parent_table'
child = 'child_table'
result = Metadata.get_foreign_key(metadata, parent, child)
# Asserts
expected = 'a_field'
metadata.get_primary_key.assert_called_once_with('parent_table')
metadata.get_fields.assert_called_once_with('child_table')
assert result == expected