def convert_definitions(self, definitions: Definitions):
"""Convert a definitions instance to codegen classes."""
assert definitions.location is not None
key = definitions.location
classes = DefinitionsMapper.map(definitions)
self.class_map.setdefault(key, []).extend(classes)
def test_map(self, mock_map_port):
classes = [
ClassFactory.list(2),
ClassFactory.list(3),
]
mock_map_port.side_effect = classes
definitions = Definitions()
service = Service(ports=[ServicePort() for _ in range(2)])
definitions.services.append(service)
result = DefinitionsMapper.map(definitions)
self.assertEqual(classes[0] + classes[1], result)
mock_map_port.assert_has_calls(
[mock.call(definitions, port) for port in service.ports])
