Search in sources :

Example 1 with Collection

use of com.ibm.watson.visual_recognition.v4.model.Collection in project java-sdk by watson-developer-cloud.

the class DiscoveryServiceIT method issueNumber659.

/* Issue 659: creating a collection does not use the configuration id */
@Test
public void issueNumber659() {
    String uniqueConfigName = UUID.randomUUID().toString() + "-config";
    Configuration testConfiguration = getTestConfiguration(DISCOVERY_TEST_CONFIG_FILE);
    CreateConfigurationOptions configOptions = new CreateConfigurationOptions.Builder(environmentId).configuration(testConfiguration).name(uniqueConfigName).build();
    Configuration configuration = discovery.createConfiguration(configOptions).execute();
    configurationIds.add(configuration.getConfigurationId());
    String uniqueCollectionName = UUID.randomUUID().toString() + "-collection";
    CreateCollectionOptions collectionOptions = new CreateCollectionOptions.Builder(environmentId, uniqueCollectionName).configurationId(configuration.getConfigurationId()).build();
    Collection collection = discovery.createCollection(collectionOptions).execute();
    collectionIds.add(collection.getCollectionId());
    assertEquals(collection.getConfigurationId(), configuration.getConfigurationId());
}
Also used : CreateConfigurationOptions(com.ibm.watson.developer_cloud.discovery.v1.model.CreateConfigurationOptions) Configuration(com.ibm.watson.developer_cloud.discovery.v1.model.Configuration) CreateCollectionOptions(com.ibm.watson.developer_cloud.discovery.v1.model.CreateCollectionOptions) Collection(com.ibm.watson.developer_cloud.discovery.v1.model.Collection) Test(org.junit.Test) WatsonServiceTest(com.ibm.watson.developer_cloud.WatsonServiceTest)

Example 2 with Collection

use of com.ibm.watson.visual_recognition.v4.model.Collection in project java-sdk by watson-developer-cloud.

the class DiscoveryServiceIT method createCollectionWithMinimalParametersIsSuccessful.

@Test
public void createCollectionWithMinimalParametersIsSuccessful() {
    String uniqueCollectionName = uniqueName + "-collection";
    CreateCollectionOptions createOptions = new CreateCollectionOptions.Builder(environmentId, uniqueCollectionName).build();
    Collection createResponse = createCollection(createOptions);
    assertNotNull(createResponse.getCollectionId());
}
Also used : CreateCollectionOptions(com.ibm.watson.developer_cloud.discovery.v1.model.CreateCollectionOptions) Collection(com.ibm.watson.developer_cloud.discovery.v1.model.Collection) Test(org.junit.Test) WatsonServiceTest(com.ibm.watson.developer_cloud.WatsonServiceTest)

Example 3 with Collection

use of com.ibm.watson.visual_recognition.v4.model.Collection in project java-sdk by watson-developer-cloud.

the class DiscoveryServiceIT method createTestCollection.

private Collection createTestCollection() {
    Configuration createConfigResponse = createTestConfig();
    String uniqueCollectionName = uniqueName + "-collection";
    CreateCollectionOptions.Builder createCollectionBuilder = new CreateCollectionOptions.Builder(environmentId, uniqueCollectionName).configurationId(createConfigResponse.getConfigurationId());
    Collection createResponse = createCollection(createCollectionBuilder.build());
    return createResponse;
}
Also used : Configuration(com.ibm.watson.developer_cloud.discovery.v1.model.Configuration) CreateCollectionOptions(com.ibm.watson.developer_cloud.discovery.v1.model.CreateCollectionOptions) Collection(com.ibm.watson.developer_cloud.discovery.v1.model.Collection)

Example 4 with Collection

use of com.ibm.watson.visual_recognition.v4.model.Collection in project java-sdk by watson-developer-cloud.

the class DiscoveryServiceIT method createCollection.

private Collection createCollection(CreateCollectionOptions createOptions) {
    Collection createResponse = discovery.createCollection(createOptions).execute();
    collectionIds.add(createResponse.getCollectionId());
    return createResponse;
}
Also used : Collection(com.ibm.watson.developer_cloud.discovery.v1.model.Collection)

Example 5 with Collection

use of com.ibm.watson.visual_recognition.v4.model.Collection in project java-sdk by watson-developer-cloud.

the class DiscoveryServiceIT method addDocumentIsSuccessful.

@SuppressWarnings("deprecation")
@Test
public void addDocumentIsSuccessful() {
    Collection collection = createTestCollection();
    String myDocumentJson = "{\"field\":\"value\"}";
    InputStream documentStream = new ByteArrayInputStream(myDocumentJson.getBytes());
    AddDocumentOptions.Builder builder = new AddDocumentOptions.Builder();
    builder.environmentId(environmentId);
    builder.collectionId(collection.getCollectionId());
    builder.file(documentStream).fileContentType(HttpMediaType.APPLICATION_JSON);
    builder.filename("test_file");
    DocumentAccepted createResponse = discovery.addDocument(builder.build()).execute();
    assertFalse(createResponse.getDocumentId().isEmpty());
    assertNull(createResponse.getNotices());
}
Also used : DocumentAccepted(com.ibm.watson.developer_cloud.discovery.v1.model.DocumentAccepted) ByteArrayInputStream(java.io.ByteArrayInputStream) AddDocumentOptions(com.ibm.watson.developer_cloud.discovery.v1.model.AddDocumentOptions) ByteArrayInputStream(java.io.ByteArrayInputStream) FileInputStream(java.io.FileInputStream) InputStream(java.io.InputStream) Collection(com.ibm.watson.developer_cloud.discovery.v1.model.Collection) Test(org.junit.Test) WatsonServiceTest(com.ibm.watson.developer_cloud.WatsonServiceTest)

Aggregations

Collection (com.ibm.watson.developer_cloud.discovery.v1.model.Collection)26 Test (org.junit.Test)23 WatsonServiceTest (com.ibm.watson.developer_cloud.WatsonServiceTest)18 RequestBuilder (com.ibm.cloud.sdk.core.http.RequestBuilder)13 Collection (com.ibm.watson.visual_recognition.v4.model.Collection)11 HashMap (java.util.HashMap)11 RecordedRequest (okhttp3.mockwebserver.RecordedRequest)11 DocumentAccepted (com.ibm.watson.developer_cloud.discovery.v1.model.DocumentAccepted)10 JsonObject (com.google.gson.JsonObject)9 CreateCollectionOptions (com.ibm.watson.developer_cloud.discovery.v1.model.CreateCollectionOptions)9 Collection (com.ibm.watson.discovery.v1.model.Collection)9 InputStream (java.io.InputStream)9 Configuration (com.ibm.watson.developer_cloud.discovery.v1.model.Configuration)8 ByteArrayInputStream (java.io.ByteArrayInputStream)8 FileInputStream (java.io.FileInputStream)8 MockResponse (okhttp3.mockwebserver.MockResponse)7 Test (org.testng.annotations.Test)7 AddDocumentOptions (com.ibm.watson.developer_cloud.discovery.v1.model.AddDocumentOptions)5 QueryOptions (com.ibm.watson.developer_cloud.discovery.v1.model.QueryOptions)5 QueryResponse (com.ibm.watson.developer_cloud.discovery.v1.model.QueryResponse)5