use of com.spectralogic.ds3autogen.python.model.response.BaseResponse in project ds3_autogen by SpectraLogic.
the class PythonCodeGenerator method generateCommands.
/**
* Generates the ds3.py python code which includes: request handlers, response handlers,
* response payload descriptors, and the client
*/
@Override
public void generateCommands(final ImmutableList<Ds3Request> ds3Requests, final ImmutableMap<String, Ds3Type> typeMap, final Ds3DocSpec docSpec) throws IOException, TemplateException {
if (isEmpty(ds3Requests)) {
LOG.info("There are no requests to generate");
return;
}
final ImmutableList<BaseRequest> baseRequests = toRequestModelList(ds3Requests, docSpec);
final ImmutableList<BaseResponse> baseResponses = toResponseModelList(ds3Requests);
final ImmutableList<TypeDescriptor> baseTypes = toTypeDescriptorList(typeMap);
final ImmutableList<BaseClient> clientCommands = toClientCommands(ds3Requests, docSpec);
final CommandSet commandSet = new CommandSet(baseRequests, baseResponses, baseTypes, clientCommands);
final Template tmpl = getCommandTemplate(config);
final Path path = toBaseProjectPath("ds3.py");
LOG.info("Getting OutputStream for file: " + path.toString());
try (final OutputStream outStream = fileUtils.getOutputFile(path);
final Writer writer = new OutputStreamWriter(outStream)) {
tmpl.process(commandSet, writer);
}
}
use of com.spectralogic.ds3autogen.python.model.response.BaseResponse in project ds3_autogen by SpectraLogic.
the class PythonCodeGenerator_Test method toResponseModel_Test.
@Test
public void toResponseModel_Test() {
final Ds3Request request = getBucketRequest();
final BaseResponse result = toResponseModel(request);
assertThat(result.getName(), is("GetBucketResponseHandler"));
assertThat(result.getCodes().size(), is(1));
final String expectedParseResponse = "if self.response.status == 200:\n" + " self.result = parseModel(xmldom.fromstring(response.read()), ListBucketResult())";
assertThat(result.getParseResponseCode(), is(expectedParseResponse));
}
Aggregations