13
13
14
14
package com .ibm .watson .developer_cloud .discovery .v1 ;
15
15
16
+ import static org .junit .Assert .assertArrayEquals ;
17
+ import static org .junit .Assert .assertEquals ;
18
+ import static org .junit .Assert .assertFalse ;
19
+ import static org .junit .Assert .assertNull ;
20
+ import static org .junit .Assert .assertTrue ;
21
+
22
+ import java .io .ByteArrayInputStream ;
23
+ import java .io .FileNotFoundException ;
24
+ import java .io .FileReader ;
25
+ import java .io .InputStream ;
26
+ import java .util .ArrayList ;
27
+ import java .util .Arrays ;
28
+ import java .util .HashSet ;
29
+ import java .util .List ;
30
+ import java .util .Set ;
31
+ import java .util .UUID ;
32
+ import java .util .concurrent .TimeUnit ;
33
+
34
+ import org .junit .After ;
35
+ import org .junit .Assume ;
36
+ import org .junit .Before ;
37
+ import org .junit .BeforeClass ;
38
+ import org .junit .Ignore ;
39
+ import org .junit .Test ;
40
+
16
41
import com .google .gson .JsonObject ;
17
42
import com .google .gson .JsonPrimitive ;
18
43
import com .ibm .watson .developer_cloud .WatsonServiceTest ;
73
98
import com .ibm .watson .developer_cloud .service .exception .UnauthorizedException ;
74
99
import com .ibm .watson .developer_cloud .util .GsonSingleton ;
75
100
import com .ibm .watson .developer_cloud .util .WaitFor ;
76
- import org .junit .After ;
77
- import org .junit .Assume ;
78
- import org .junit .Before ;
79
- import org .junit .BeforeClass ;
80
- import org .junit .Ignore ;
81
- import org .junit .Test ;
82
-
83
- import static org .junit .Assert .assertEquals ;
84
- import static org .junit .Assert .assertFalse ;
85
- import static org .junit .Assert .assertNull ;
86
- import static org .junit .Assert .assertTrue ;
87
- import static org .junit .Assert .assertArrayEquals ;
88
-
89
- import java .io .ByteArrayInputStream ;
90
- import java .io .FileNotFoundException ;
91
- import java .io .FileReader ;
92
- import java .io .InputStream ;
93
- import java .util .ArrayList ;
94
- import java .util .Arrays ;
95
- import java .util .HashSet ;
96
- import java .util .List ;
97
- import java .util .Set ;
98
- import java .util .UUID ;
99
- import java .util .concurrent .TimeUnit ;
100
101
101
102
/**
102
103
* Integration tests for {@link Discovery}.
@@ -115,7 +116,7 @@ public class DiscoveryServiceIT extends WatsonServiceTest {
115
116
116
117
@ BeforeClass
117
118
public static void setupClass () {
118
- //get the properties
119
+ // get the properties
119
120
DiscoveryServiceIT dummyTest = new DiscoveryServiceIT ();
120
121
String username = dummyTest .getProperty ("discovery.username" );
121
122
String password = dummyTest .getProperty ("discovery.password" );
@@ -130,19 +131,18 @@ public static void setupClass() {
130
131
GetEnvironmentsRequest getRequest = new GetEnvironmentsRequest .Builder ().build ();
131
132
GetEnvironmentsResponse getResponse = discovery .getEnvironments (getRequest ).execute ();
132
133
for (Environment environment : getResponse .getEnvironments ()) {
133
- //look for an existing environment that isn't read only
134
+ // look for an existing environment that isn't read only
134
135
if (!environment .isReadOnly ()) {
135
136
environmentId = environment .getEnvironmentId ();
136
137
break ;
137
138
}
138
139
}
139
140
140
141
if (environmentId == null ) {
141
- //no environment found, create a new one (assuming we are a FREE plan)
142
+ // no environment found, create a new one (assuming we are a FREE plan)
142
143
String environmentName = "watson_developer_cloud_test_environment" ;
143
144
CreateEnvironmentRequest .Size size = CreateEnvironmentRequest .Size .FREE ;
144
- CreateEnvironmentRequest createRequest = new CreateEnvironmentRequest .Builder (environmentName , size )
145
- .build ();
145
+ CreateEnvironmentRequest createRequest = new CreateEnvironmentRequest .Builder (environmentName , size ).build ();
146
146
CreateEnvironmentResponse createResponse = discovery .createEnvironment (createRequest ).execute ();
147
147
environmentId = createResponse .getEnvironmentId ();
148
148
WaitFor .Condition environmentReady = new EnvironmentReady (discovery , environmentId );
@@ -166,14 +166,13 @@ public void setup() throws Exception {
166
166
@ After
167
167
public void cleanup () {
168
168
for (String collectionId : collectionIds ) {
169
- DeleteCollectionRequest deleteRequest = new DeleteCollectionRequest .Builder (environmentId , collectionId )
170
- .build ();
169
+ DeleteCollectionRequest deleteRequest = new DeleteCollectionRequest .Builder (environmentId , collectionId ).build ();
171
170
discovery .deleteCollection (deleteRequest ).execute ();
172
171
}
173
172
174
173
for (String configurationId : configurationIds ) {
175
- DeleteConfigurationRequest deleteRequest = new DeleteConfigurationRequest . Builder ( environmentId ,
176
- configurationId ).build ();
174
+ DeleteConfigurationRequest deleteRequest =
175
+ new DeleteConfigurationRequest . Builder ( environmentId , configurationId ).build ();
177
176
discovery .deleteConfiguration (deleteRequest ).execute ();
178
177
}
179
178
}
@@ -242,8 +241,8 @@ public void getEnvironmentsByNameIsSuccessful() {
242
241
@ Ignore ("Only 1 BYOD environment allowed per service instance, so we cannot create more" )
243
242
public void createEnvironmentIsSuccessful () {
244
243
String environmentName = uniqueName + "-environment" ;
245
- CreateEnvironmentRequest createRequest = new CreateEnvironmentRequest . Builder ( environmentName ,
246
- CreateEnvironmentRequest .Size .FREE ).build ();
244
+ CreateEnvironmentRequest createRequest =
245
+ new CreateEnvironmentRequest . Builder ( environmentName , CreateEnvironmentRequest .Size .FREE ).build ();
247
246
CreateEnvironmentResponse createResponse = createEnvironment (createRequest );
248
247
249
248
assertEquals (environmentName , createResponse .getName ());
@@ -253,12 +252,12 @@ public void createEnvironmentIsSuccessful() {
253
252
@ Ignore ("Only 1 BYOD environment allowed per service instance, so do not delete it" )
254
253
public void deleteEnvironmentIsSuccessful () {
255
254
String environmentName = uniqueName + "-environment" ;
256
- CreateEnvironmentRequest createRequest = new CreateEnvironmentRequest . Builder ( environmentName ,
257
- CreateEnvironmentRequest .Size .FREE ).build ();
255
+ CreateEnvironmentRequest createRequest =
256
+ new CreateEnvironmentRequest . Builder ( environmentName , CreateEnvironmentRequest .Size .FREE ).build ();
258
257
CreateEnvironmentResponse createResponse = createEnvironment (createRequest );
259
258
260
- DeleteEnvironmentRequest deleteRequest = new DeleteEnvironmentRequest . Builder ( createResponse . getEnvironmentId ())
261
- .build ();
259
+ DeleteEnvironmentRequest deleteRequest =
260
+ new DeleteEnvironmentRequest . Builder ( createResponse . getEnvironmentId ()) .build ();
262
261
DeleteEnvironmentResponse deleteRepsonse = deleteEnvironment (deleteRequest );
263
262
264
263
assertEquals (createResponse .getEnvironmentId (), deleteRepsonse .getEnvironmentId ());
@@ -268,13 +267,13 @@ public void deleteEnvironmentIsSuccessful() {
268
267
@ Ignore ("Only 1 BYOD environment allowed per service instance, so we cannot create more" )
269
268
public void updateEnvironmentIsSuccessful () {
270
269
String environmentName = uniqueName + "-environment" ;
271
- CreateEnvironmentRequest createRequest = new CreateEnvironmentRequest . Builder ( environmentName ,
272
- CreateEnvironmentRequest .Size .FREE ).build ();
270
+ CreateEnvironmentRequest createRequest =
271
+ new CreateEnvironmentRequest . Builder ( environmentName , CreateEnvironmentRequest .Size .FREE ).build ();
273
272
CreateEnvironmentResponse createResponse = createEnvironment (createRequest );
274
273
275
274
String randomDescription = UUID .randomUUID ().toString () + " appbuilder tests" ;
276
- UpdateEnvironmentRequest .Builder updateBuilder = new UpdateEnvironmentRequest . Builder (
277
- createResponse .getEnvironmentId (), environmentName );
275
+ UpdateEnvironmentRequest .Builder updateBuilder =
276
+ new UpdateEnvironmentRequest . Builder ( createResponse .getEnvironmentId (), environmentName );
278
277
updateBuilder .description (randomDescription );
279
278
UpdateEnvironmentResponse updateResponse = discovery .updateEnvironment (updateBuilder .build ()).execute ();
280
279
@@ -305,8 +304,8 @@ public void createConfigurationIsSuccessful() {
305
304
public void deleteConfigurationIsSuccessful () {
306
305
CreateConfigurationResponse createResponse = createTestConfig ();
307
306
308
- DeleteConfigurationRequest deleteRequest = new DeleteConfigurationRequest . Builder ( environmentId ,
309
- createResponse .getConfigurationId ()).build ();
307
+ DeleteConfigurationRequest deleteRequest =
308
+ new DeleteConfigurationRequest . Builder ( environmentId , createResponse .getConfigurationId ()).build ();
310
309
DeleteConfigurationResponse deleteResponse = deleteConfiguration (deleteRequest );
311
310
312
311
assertEquals (createResponse .getConfigurationId (), deleteResponse .getConfigurationId ());
@@ -316,8 +315,8 @@ public void deleteConfigurationIsSuccessful() {
316
315
public void getConfigurationIsSuccessful () {
317
316
CreateConfigurationResponse createResponse = createTestConfig ();
318
317
319
- GetConfigurationRequest getRequest = new GetConfigurationRequest . Builder ( environmentId ,
320
- createResponse .getConfigurationId ()).build ();
318
+ GetConfigurationRequest getRequest =
319
+ new GetConfigurationRequest . Builder ( environmentId , createResponse .getConfigurationId ()).build ();
321
320
GetConfigurationResponse getResponse = discovery .getConfiguration (getRequest ).execute ();
322
321
323
322
assertEquals (createResponse .getName (), getResponse .getName ());
@@ -340,8 +339,8 @@ public void updateConfigurationIsSuccessful() {
340
339
CreateConfigurationResponse createResponse = createTestConfig ();
341
340
342
341
String newUniqueName = createResponse .getName () + UUID .randomUUID ().toString ();
343
- UpdateConfigurationRequest .Builder updateBuilder = new UpdateConfigurationRequest . Builder ( environmentId ,
344
- createResponse .getConfigurationId ());
342
+ UpdateConfigurationRequest .Builder updateBuilder =
343
+ new UpdateConfigurationRequest . Builder ( environmentId , createResponse .getConfigurationId ());
345
344
createResponse .setName (newUniqueName );
346
345
updateBuilder .configuration (createResponse );
347
346
UpdateConfigurationResponse updateResponse = discovery .updateConfiguration (updateBuilder .build ()).execute ();
@@ -379,10 +378,10 @@ public void deleteCollectionIsSuccessful() {
379
378
createConfigResponse .getConfigurationId (), uniqueCollectionName );
380
379
CreateCollectionResponse createResponse = createCollection (createCollectionBuilder .build ());
381
380
382
- //need to wait for collection to be ready
381
+ // need to wait for collection to be ready
383
382
384
- DeleteCollectionRequest deleteRequest = new DeleteCollectionRequest . Builder ( environmentId ,
385
- createResponse .getCollectionId ()).build ();
383
+ DeleteCollectionRequest deleteRequest =
384
+ new DeleteCollectionRequest . Builder ( environmentId , createResponse .getCollectionId ()).build ();
386
385
DeleteCollectionResponse deleteResponse = deleteCollection (deleteRequest );
387
386
388
387
assertEquals (Status .DELETED , deleteResponse .getStatus ());
@@ -397,10 +396,10 @@ public void getCollectionIsSuccessful() {
397
396
createConfigResponse .getConfigurationId (), uniqueCollectionName );
398
397
CreateCollectionResponse createResponse = createCollection (createCollectionBuilder .build ());
399
398
400
- GetCollectionRequest getRequest = new GetCollectionRequest . Builder ( environmentId ,
401
- createResponse .getCollectionId ()).build ();
399
+ GetCollectionRequest getRequest =
400
+ new GetCollectionRequest . Builder ( environmentId , createResponse .getCollectionId ()).build ();
402
401
403
- //need to wait for collection to be ready
402
+ // need to wait for collection to be ready
404
403
405
404
GetCollectionResponse getResponse = discovery .getCollection (getRequest ).execute ();
406
405
@@ -424,22 +423,24 @@ public void getCollectionsByNameIsSuccessful() {
424
423
assertEquals (uniqueCollectionName , getResponse .getCollections ().get (0 ).getName ());
425
424
}
426
425
426
+ @ SuppressWarnings ("deprecation" )
427
427
@ Test
428
428
public void createDocumentIsSuccessful () {
429
429
CreateCollectionResponse createCollectionResponse = createTestCollection ();
430
430
431
431
String myDocumentJson = "{\" field\" :\" value\" }" ;
432
432
InputStream documentStream = new ByteArrayInputStream (myDocumentJson .getBytes ());
433
433
434
- CreateDocumentRequest .Builder builder = new CreateDocumentRequest . Builder ( environmentId ,
435
- createCollectionResponse .getCollectionId ());
434
+ CreateDocumentRequest .Builder builder =
435
+ new CreateDocumentRequest . Builder ( environmentId , createCollectionResponse .getCollectionId ());
436
436
builder .inputStream (documentStream , HttpMediaType .APPLICATION_JSON );
437
437
438
438
CreateDocumentResponse createResponse = discovery .createDocument (builder .build ()).execute ();
439
439
assertFalse (createResponse .getDocumentId ().isEmpty ());
440
440
assertNull (createResponse .getNotices ());
441
441
}
442
442
443
+ @ SuppressWarnings ("deprecation" )
443
444
@ Test
444
445
public void createDocumentWithMetadataIsSuccessful () {
445
446
CreateCollectionResponse createCollectionResponse = createTestCollection ();
@@ -473,8 +474,8 @@ public void deleteDocumentIsSuccessful() {
473
474
String collectionId = createCollectionResponse .getCollectionId ();
474
475
CreateDocumentResponse createDocumentResponse = createTestDocument (collectionId );
475
476
476
- DeleteDocumentRequest deleteRequest = new DeleteDocumentRequest . Builder ( environmentId , collectionId ,
477
- createDocumentResponse .getDocumentId ()).build ();
477
+ DeleteDocumentRequest deleteRequest =
478
+ new DeleteDocumentRequest . Builder ( environmentId , collectionId , createDocumentResponse .getDocumentId ()).build ();
478
479
DeleteDocumentResponse deleteResponse = discovery .deleteDocument (deleteRequest ).execute ();
479
480
assertEquals (Status .DELETED , deleteResponse .getStatus ());
480
481
}
@@ -485,8 +486,8 @@ public void getDocumentIsSuccessful() {
485
486
String collectionId = createCollectionResponse .getCollectionId ();
486
487
CreateDocumentResponse createDocumentResponse = createTestDocument (collectionId );
487
488
488
- GetDocumentRequest getRequest = new GetDocumentRequest . Builder ( environmentId , collectionId ,
489
- createDocumentResponse .getDocumentId ()).build ();
489
+ GetDocumentRequest getRequest =
490
+ new GetDocumentRequest . Builder ( environmentId , collectionId , createDocumentResponse .getDocumentId ()).build ();
490
491
GetDocumentResponse getResponse = discovery .getDocument (getRequest ).execute ();
491
492
492
493
assertEquals (Document .Status .AVAILABLE , getResponse .getStatus ());
@@ -498,15 +499,15 @@ public void updateDocumentIsSuccessful() {
498
499
String collectionId = createCollectionResponse .getCollectionId ();
499
500
CreateDocumentResponse createDocumentResponse = createTestDocument (collectionId );
500
501
501
- UpdateDocumentRequest .Builder updateBuilder = new UpdateDocumentRequest . Builder ( environmentId , collectionId ,
502
- createDocumentResponse .getDocumentId ());
502
+ UpdateDocumentRequest .Builder updateBuilder =
503
+ new UpdateDocumentRequest . Builder ( environmentId , collectionId , createDocumentResponse .getDocumentId ());
503
504
String myDocumentJson = "{\" field\" :\" value2\" }" ;
504
505
InputStream documentStream = new ByteArrayInputStream (myDocumentJson .getBytes ());
505
506
updateBuilder .inputStream (documentStream , HttpMediaType .APPLICATION_JSON );
506
507
UpdateDocumentResponse updateResponse = discovery .updateDocument (updateBuilder .build ()).execute ();
507
508
508
- GetDocumentRequest getRequest = new GetDocumentRequest . Builder ( environmentId , collectionId ,
509
- updateResponse .getDocumentId ()).build ();
509
+ GetDocumentRequest getRequest =
510
+ new GetDocumentRequest . Builder ( environmentId , collectionId , updateResponse .getDocumentId ()).build ();
510
511
GetDocumentResponse getResponse = discovery .getDocument (getRequest ).execute ();
511
512
512
513
assertEquals (Document .Status .AVAILABLE , getResponse .getStatus ());
@@ -525,12 +526,11 @@ public void updateDocumentWithMetadataIsSuccessful() {
525
526
JsonObject myMetadata = new JsonObject ();
526
527
myMetadata .add ("foo" , new JsonPrimitive ("bar" ));
527
528
528
- UpdateDocumentRequest .Builder updateBuilder = new UpdateDocumentRequest . Builder ( environmentId , collectionId ,
529
- createDocumentResponse .getDocumentId ());
529
+ UpdateDocumentRequest .Builder updateBuilder =
530
+ new UpdateDocumentRequest . Builder ( environmentId , collectionId , createDocumentResponse .getDocumentId ());
530
531
updateBuilder .inputStream (documentStream , HttpMediaType .APPLICATION_JSON );
531
532
updateBuilder .metadata (myMetadata );
532
- UpdateDocumentResponse updateResponse =
533
- discovery .updateDocument (updateBuilder .build ()).execute ();
533
+ UpdateDocumentResponse updateResponse = discovery .updateDocument (updateBuilder .build ()).execute ();
534
534
535
535
WaitFor .Condition documentAccepted =
536
536
new DocumentAccepted (environmentId , collectionId , updateResponse .getDocumentId ());
@@ -548,8 +548,7 @@ public void getCollectionFieldsIsSuccessful() {
548
548
String collectionId = createCollectionResponse .getCollectionId ();
549
549
createTestDocument (collectionId );
550
550
551
- GetCollectionFieldsRequest getRequest = new GetCollectionFieldsRequest .Builder (environmentId , collectionId )
552
- .build ();
551
+ GetCollectionFieldsRequest getRequest = new GetCollectionFieldsRequest .Builder (environmentId , collectionId ).build ();
553
552
GetCollectionFieldsResponse getResponse = discovery .getCollectionFields (getRequest ).execute ();
554
553
555
554
assertFalse (getResponse .getFields ().isEmpty ());
@@ -751,7 +750,7 @@ public void queryWithAggregationAverageIsSuccessful() throws InterruptedExceptio
751
750
public void issueNumber517 () {
752
751
String uniqueConfigName = uniqueName + "-config" ;
753
752
CreateConfigurationRequest .Builder createBuilder = new CreateConfigurationRequest .Builder (environmentId );
754
- Configuration configuration = getTestConfiguration (DISCOVERY1_TEST_CONFIG_FILE );
753
+ Configuration configuration = getTestConfiguration (DISCOVERY1_TEST_CONFIG_FILE );
755
754
756
755
configuration .setName (uniqueConfigName );
757
756
createBuilder .configuration (configuration );
@@ -771,7 +770,7 @@ public void issueNumber518() {
771
770
772
771
String uniqueConfigName = uniqueName + "-config" ;
773
772
CreateConfigurationRequest .Builder createBuilder = new CreateConfigurationRequest .Builder (environmentId );
774
- Configuration configuration = getTestConfiguration (DISCOVERY2_TEST_CONFIG_FILE );
773
+ Configuration configuration = getTestConfiguration (DISCOVERY2_TEST_CONFIG_FILE );
775
774
776
775
configuration .setName (uniqueConfigName );
777
776
createBuilder .configuration (configuration );
@@ -846,13 +845,14 @@ private CreateDocumentResponse createTestDocument(String collectionId) {
846
845
return createTestDocument (myDocumentJson , collectionId );
847
846
}
848
847
848
+ @ SuppressWarnings ("deprecation" )
849
849
private CreateDocumentResponse createTestDocument (String json , String collectionId ) {
850
850
InputStream documentStream = new ByteArrayInputStream (json .getBytes ());
851
851
CreateDocumentRequest .Builder builder = new CreateDocumentRequest .Builder (environmentId , collectionId );
852
852
builder .inputStream (documentStream , HttpMediaType .APPLICATION_JSON );
853
853
CreateDocumentResponse createResponse = discovery .createDocument (builder .build ()).execute ();
854
- WaitFor .Condition documentAccepted = new DocumentAccepted ( environmentId , collectionId ,
855
- createResponse .getDocumentId ());
854
+ WaitFor .Condition documentAccepted =
855
+ new DocumentAccepted ( environmentId , collectionId , createResponse .getDocumentId ());
856
856
WaitFor .waitFor (documentAccepted , 5 , TimeUnit .SECONDS , 500 );
857
857
return createResponse ;
858
858
}
@@ -887,7 +887,7 @@ private static class EnvironmentReady implements WaitFor.Condition {
887
887
private final Discovery discovery ;
888
888
private final String environmentId ;
889
889
890
- public EnvironmentReady (Discovery discovery , String environmentId ) {
890
+ private EnvironmentReady (Discovery discovery , String environmentId ) {
891
891
this .discovery = discovery ;
892
892
this .environmentId = environmentId ;
893
893
}
@@ -901,10 +901,6 @@ public boolean isSatisfied() {
901
901
}
902
902
903
903
private class DocumentAccepted implements WaitFor .Condition {
904
- private final String environmentId ;
905
- private final String collectionId ;
906
- private final String documentId ;
907
-
908
904
public DocumentAccepted (String environmentId , String collectionId , String documentId ) {
909
905
this .environmentId = environmentId ;
910
906
this .collectionId = collectionId ;
@@ -913,10 +909,15 @@ public DocumentAccepted(String environmentId, String collectionId, String docume
913
909
914
910
@ Override
915
911
public boolean isSatisfied () {
916
- GetDocumentRequest getRequest = new GetDocumentRequest .Builder (environmentId , collectionId , documentId )
917
- .build ();
912
+ GetDocumentRequest getRequest = new GetDocumentRequest .Builder (environmentId , collectionId , documentId ).build ();
918
913
Document .Status status = discovery .getDocument (getRequest ).execute ().getStatus ();
919
914
return status .equals (Document .Status .AVAILABLE );
920
915
}
916
+
917
+ private final String environmentId ;
918
+ private final String collectionId ;
919
+ private final String documentId ;
920
+
921
921
}
922
+
922
923
}
0 commit comments