use of com.ibm.watson.discovery.v2.model.CreateEnrichmentOptions in project java-sdk by watson-developer-cloud.
the class DiscoveryTest method testCreateEnrichmentWOptions.
@Test
public void testCreateEnrichmentWOptions() throws Throwable {
// Schedule some responses.
String mockResponseBody = "{\"enrichment_id\": \"enrichmentId\", \"name\": \"name\", \"description\": \"description\", \"type\": \"part_of_speech\", \"options\": {\"languages\": [\"languages\"], \"entity_type\": \"entityType\", \"regular_expression\": \"regularExpression\", \"result_field\": \"resultField\"}}";
String createEnrichmentPath = "/v2/projects/testString/enrichments";
server.enqueue(new MockResponse().setHeader("Content-type", "application/json").setResponseCode(201).setBody(mockResponseBody));
constructClientService();
// Construct an instance of the EnrichmentOptions model
EnrichmentOptions enrichmentOptionsModel = new EnrichmentOptions.Builder().languages(new java.util.ArrayList<String>(java.util.Arrays.asList("testString"))).entityType("testString").regularExpression("testString").resultField("testString").build();
// Construct an instance of the CreateEnrichment model
CreateEnrichment createEnrichmentModel = new CreateEnrichment.Builder().name("testString").description("testString").type("dictionary").options(enrichmentOptionsModel).build();
// Construct an instance of the CreateEnrichmentOptions model
CreateEnrichmentOptions createEnrichmentOptionsModel = new CreateEnrichmentOptions.Builder().projectId("testString").enrichment(createEnrichmentModel).file(TestUtilities.createMockStream("This is a mock file.")).build();
// Invoke operation with valid options model (positive test)
Response<Enrichment> response = discoveryService.createEnrichment(createEnrichmentOptionsModel).execute();
assertNotNull(response);
Enrichment responseObj = response.getResult();
assertNotNull(responseObj);
// Verify the contents of the request
RecordedRequest request = server.takeRequest();
assertNotNull(request);
assertEquals(request.getMethod(), "POST");
// Check query
Map<String, String> query = TestUtilities.parseQueryString(request);
assertNotNull(query);
// Get query params
assertEquals(query.get("version"), "testString");
// Check request path
String parsedPath = TestUtilities.parseReqPath(request);
assertEquals(parsedPath, createEnrichmentPath);
}
use of com.ibm.watson.discovery.v2.model.CreateEnrichmentOptions in project java-sdk by watson-developer-cloud.
the class Discovery method createEnrichment.
/**
* Create an enrichment.
*
* <p>Create an enrichment for use with the specified project.
*
* @param createEnrichmentOptions the {@link CreateEnrichmentOptions} containing the options for
* the call
* @return a {@link ServiceCall} with a result of type {@link Enrichment}
*/
public ServiceCall<Enrichment> createEnrichment(CreateEnrichmentOptions createEnrichmentOptions) {
com.ibm.cloud.sdk.core.util.Validator.notNull(createEnrichmentOptions, "createEnrichmentOptions cannot be null");
Map<String, String> pathParamsMap = new HashMap<String, String>();
pathParamsMap.put("project_id", createEnrichmentOptions.projectId());
RequestBuilder builder = RequestBuilder.post(RequestBuilder.resolveRequestUrl(getServiceUrl(), "/v2/projects/{project_id}/enrichments", pathParamsMap));
Map<String, String> sdkHeaders = SdkCommon.getSdkHeaders("discovery", "v2", "createEnrichment");
for (Entry<String, String> header : sdkHeaders.entrySet()) {
builder.header(header.getKey(), header.getValue());
}
builder.header("Accept", "application/json");
builder.query("version", String.valueOf(this.version));
MultipartBody.Builder multipartBuilder = new MultipartBody.Builder();
multipartBuilder.setType(MultipartBody.FORM);
multipartBuilder.addFormDataPart("enrichment", createEnrichmentOptions.enrichment().toString());
if (createEnrichmentOptions.file() != null) {
okhttp3.RequestBody fileBody = RequestUtils.inputStreamBody(createEnrichmentOptions.file(), "application/octet-stream");
multipartBuilder.addFormDataPart("file", "filename", fileBody);
}
builder.body(multipartBuilder.build());
ResponseConverter<Enrichment> responseConverter = ResponseConverterUtils.getValue(new com.google.gson.reflect.TypeToken<Enrichment>() {
}.getType());
return createServiceCall(builder.build(), responseConverter);
}
Aggregations