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
        )