Search in sources :

Example 1 with AnalyzeEntitiesRequest

use of com.google.cloud.language.v1.AnalyzeEntitiesRequest in project google-cloud-java by GoogleCloudPlatform.

the class LanguageServiceClientTest method analyzeEntitiesTest.

@Test
@SuppressWarnings("all")
public void analyzeEntitiesTest() {
    String language = "language-1613589672";
    AnalyzeEntitiesResponse expectedResponse = AnalyzeEntitiesResponse.newBuilder().setLanguage(language).build();
    mockLanguageService.addResponse(expectedResponse);
    Document document = Document.newBuilder().build();
    EncodingType encodingType = EncodingType.NONE;
    AnalyzeEntitiesResponse actualResponse = client.analyzeEntities(document, encodingType);
    Assert.assertEquals(expectedResponse, actualResponse);
    List<GeneratedMessageV3> actualRequests = mockLanguageService.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    AnalyzeEntitiesRequest actualRequest = (AnalyzeEntitiesRequest) actualRequests.get(0);
    Assert.assertEquals(document, actualRequest.getDocument());
    Assert.assertEquals(encodingType, actualRequest.getEncodingType());
}
Also used : AnalyzeEntitiesRequest(com.google.cloud.language.v1beta2.AnalyzeEntitiesRequest) AnalyzeEntitiesResponse(com.google.cloud.language.v1beta2.AnalyzeEntitiesResponse) EncodingType(com.google.cloud.language.v1beta2.EncodingType) Document(com.google.cloud.language.v1beta2.Document) GeneratedMessageV3(com.google.protobuf.GeneratedMessageV3) Test(org.junit.Test)

Example 2 with AnalyzeEntitiesRequest

use of com.google.cloud.language.v1.AnalyzeEntitiesRequest in project google-cloud-java by GoogleCloudPlatform.

the class LanguageServiceClientTest method analyzeEntitiesTest.

@Test
@SuppressWarnings("all")
public void analyzeEntitiesTest() {
    String language = "language-1613589672";
    AnalyzeEntitiesResponse expectedResponse = AnalyzeEntitiesResponse.newBuilder().setLanguage(language).build();
    mockLanguageService.addResponse(expectedResponse);
    Document document = Document.newBuilder().build();
    EncodingType encodingType = EncodingType.NONE;
    AnalyzeEntitiesResponse actualResponse = client.analyzeEntities(document, encodingType);
    Assert.assertEquals(expectedResponse, actualResponse);
    List<GeneratedMessageV3> actualRequests = mockLanguageService.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    AnalyzeEntitiesRequest actualRequest = (AnalyzeEntitiesRequest) actualRequests.get(0);
    Assert.assertEquals(document, actualRequest.getDocument());
    Assert.assertEquals(encodingType, actualRequest.getEncodingType());
}
Also used : AnalyzeEntitiesRequest(com.google.cloud.language.v1.AnalyzeEntitiesRequest) AnalyzeEntitiesResponse(com.google.cloud.language.v1.AnalyzeEntitiesResponse) EncodingType(com.google.cloud.language.v1.EncodingType) Document(com.google.cloud.language.v1.Document) GeneratedMessageV3(com.google.protobuf.GeneratedMessageV3) Test(org.junit.Test)

Aggregations

GeneratedMessageV3 (com.google.protobuf.GeneratedMessageV3)2 Test (org.junit.Test)2 AnalyzeEntitiesRequest (com.google.cloud.language.v1.AnalyzeEntitiesRequest)1 AnalyzeEntitiesResponse (com.google.cloud.language.v1.AnalyzeEntitiesResponse)1 Document (com.google.cloud.language.v1.Document)1 EncodingType (com.google.cloud.language.v1.EncodingType)1 AnalyzeEntitiesRequest (com.google.cloud.language.v1beta2.AnalyzeEntitiesRequest)1 AnalyzeEntitiesResponse (com.google.cloud.language.v1beta2.AnalyzeEntitiesResponse)1 Document (com.google.cloud.language.v1beta2.Document)1 EncodingType (com.google.cloud.language.v1beta2.EncodingType)1