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