return num_mock_dict[hn_id].postkanton else: return None def list_postkantons_by_gemeente(gem_id): if gem_id in gemeente_mock_dict: return gemeente_mock_dict[gem_id].postkantons else: return None crab_gateway_mock = Mock() crab_gateway_mock.get_gemeente_by_id = get_gemeente_by_id crab_gateway_mock.get_straat_by_id = get_straat_by_id crab_gateway_mock.get_huisnummer_by_id = get_huisnummer_by_id crab_gateway_mock.get_subadres_by_id = get_subadres_by_id crab_gateway_mock.get_postkanton_by_huisnummer = get_postkanton_by_huisnummer crab_gateway_mock.list_postkantons_by_gemeente = list_postkantons_by_gemeente crab_gateway_mock.list_gemeenten = list_gemeenten class AdressenSchemaTests(unittest.TestCase): def setUp(self): request = testing.DummyRequest() request.crab_gateway = Mock(return_value=crab_gateway_mock) adressen_schema = CrabAdresSchemaNode() self.schema = adressen_schema.bind( request=request