Examples with ThreatType com.google.webrisk.v1.ThreatType used on opensource projects

Example 1 with ThreatType

use of com.google.webrisk.v1.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.
    }
}

Example 2 with ThreatType

use of com.google.webrisk.v1.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.
    }
}

Example 3 with ThreatType

use of com.google.webrisk.v1.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()));
}

Example 4 with ThreatType

use of com.google.webrisk.v1.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.
    }
}

Example 5 with ThreatType

use of com.google.webrisk.v1.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()));
}