use of com.google.webrisk.v1beta1.ThreatType in project java-webrisk by googleapis.
the class WebRiskServiceClientTest method searchUrisExceptionTest.
@Test
public void searchUrisExceptionTest() throws Exception {
StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
mockWebRiskService.addException(exception);
try {
String uri = "uri116076";
List<ThreatType> threatTypes = new ArrayList<>();
client.searchUris(uri, threatTypes);
Assert.fail("No exception raised");
} catch (InvalidArgumentException e) {
// Expected exception.
}
}
use of com.google.webrisk.v1beta1.ThreatType in project java-webrisk by googleapis.
the class WebRiskServiceV1Beta1ClientTest method searchHashesExceptionTest.
@Test
public void searchHashesExceptionTest() throws Exception {
StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
mockWebRiskServiceV1Beta1.addException(exception);
try {
ByteString hashPrefix = ByteString.EMPTY;
List<ThreatType> threatTypes = new ArrayList<>();
client.searchHashes(hashPrefix, threatTypes);
Assert.fail("No exception raised");
} catch (InvalidArgumentException e) {
// Expected exception.
}
}
use of com.google.webrisk.v1beta1.ThreatType in project java-webrisk by googleapis.
the class WebRiskServiceV1Beta1ClientTest method searchHashesTest.
@Test
public void searchHashesTest() throws Exception {
SearchHashesResponse expectedResponse = SearchHashesResponse.newBuilder().addAllThreats(new ArrayList<SearchHashesResponse.ThreatHash>()).setNegativeExpireTime(Timestamp.newBuilder().build()).build();
mockWebRiskServiceV1Beta1.addResponse(expectedResponse);
ByteString hashPrefix = ByteString.EMPTY;
List<ThreatType> threatTypes = new ArrayList<>();
SearchHashesResponse actualResponse = client.searchHashes(hashPrefix, threatTypes);
Assert.assertEquals(expectedResponse, actualResponse);
List<AbstractMessage> actualRequests = mockWebRiskServiceV1Beta1.getRequests();
Assert.assertEquals(1, actualRequests.size());
SearchHashesRequest actualRequest = ((SearchHashesRequest) actualRequests.get(0));
Assert.assertEquals(hashPrefix, actualRequest.getHashPrefix());
Assert.assertEquals(threatTypes, actualRequest.getThreatTypesList());
Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
use of com.google.webrisk.v1beta1.ThreatType in project java-webrisk by googleapis.
the class WebRiskServiceV1Beta1ClientTest method computeThreatListDiffExceptionTest.
@Test
public void computeThreatListDiffExceptionTest() throws Exception {
StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
mockWebRiskServiceV1Beta1.addException(exception);
try {
ThreatType threatType = ThreatType.forNumber(0);
ByteString versionToken = ByteString.EMPTY;
ComputeThreatListDiffRequest.Constraints constraints = ComputeThreatListDiffRequest.Constraints.newBuilder().build();
client.computeThreatListDiff(threatType, versionToken, constraints);
Assert.fail("No exception raised");
} catch (InvalidArgumentException e) {
// Expected exception.
}
}
use of com.google.webrisk.v1beta1.ThreatType in project java-webrisk by googleapis.
the class WebRiskServiceV1Beta1ClientTest method searchUrisTest.
@Test
public void searchUrisTest() throws Exception {
SearchUrisResponse expectedResponse = SearchUrisResponse.newBuilder().setThreat(SearchUrisResponse.ThreatUri.newBuilder().build()).build();
mockWebRiskServiceV1Beta1.addResponse(expectedResponse);
String uri = "uri116076";
List<ThreatType> threatTypes = new ArrayList<>();
SearchUrisResponse actualResponse = client.searchUris(uri, threatTypes);
Assert.assertEquals(expectedResponse, actualResponse);
List<AbstractMessage> actualRequests = mockWebRiskServiceV1Beta1.getRequests();
Assert.assertEquals(1, actualRequests.size());
SearchUrisRequest actualRequest = ((SearchUrisRequest) actualRequests.get(0));
Assert.assertEquals(uri, actualRequest.getUri());
Assert.assertEquals(threatTypes, actualRequest.getThreatTypesList());
Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
Aggregations