class TestSources(unittest.TestCase): def setUp(self): self.sources = GOBSources() def test_get_relations(self): # Assert we get a list of relations for a collection self.assertIsInstance(self.sources.get_relations('nap', 'peilmerken'), list)
class TestSources(unittest.TestCase): def setUp(self): self.sources = GOBSources() def test_get_relations(self): # Assert we get a list of relations for a collection self.assertIsInstance(self.sources.get_relations('nap', 'peilmerken'), list) def test_get_field_relations_keyerror(self): self.sources.get_relations = MagicMock(side_effect=KeyError) self.assertEqual([], self.sources.get_field_relations( 'catalog', 'collection', 'fieldname'))