Search in sources :

Example 11 with ThreatType

use of com.google.webrisk.v1beta1.ThreatType in project java-webrisk by googleapis.

the class WebRiskServiceClientTest method computeThreatListDiffTest.

@Test
public void computeThreatListDiffTest() throws Exception {
    ComputeThreatListDiffResponse expectedResponse = ComputeThreatListDiffResponse.newBuilder().setAdditions(ThreatEntryAdditions.newBuilder().build()).setRemovals(ThreatEntryRemovals.newBuilder().build()).setNewVersionToken(ByteString.EMPTY).setChecksum(ComputeThreatListDiffResponse.Checksum.newBuilder().build()).setRecommendedNextDiff(Timestamp.newBuilder().build()).build();
    mockWebRiskService.addResponse(expectedResponse);
    ThreatType threatType = ThreatType.forNumber(0);
    ByteString versionToken = ByteString.EMPTY;
    ComputeThreatListDiffRequest.Constraints constraints = ComputeThreatListDiffRequest.Constraints.newBuilder().build();
    ComputeThreatListDiffResponse actualResponse = client.computeThreatListDiff(threatType, versionToken, constraints);
    Assert.assertEquals(expectedResponse, actualResponse);
    List<AbstractMessage> actualRequests = mockWebRiskService.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    ComputeThreatListDiffRequest actualRequest = ((ComputeThreatListDiffRequest) actualRequests.get(0));
    Assert.assertEquals(threatType, actualRequest.getThreatType());
    Assert.assertEquals(versionToken, actualRequest.getVersionToken());
    Assert.assertEquals(constraints, actualRequest.getConstraints());
    Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
Also used : AbstractMessage(com.google.protobuf.AbstractMessage) ThreatType(com.google.webrisk.v1.ThreatType) ByteString(com.google.protobuf.ByteString) ComputeThreatListDiffRequest(com.google.webrisk.v1.ComputeThreatListDiffRequest) ComputeThreatListDiffResponse(com.google.webrisk.v1.ComputeThreatListDiffResponse) Test(org.junit.Test)

Example 12 with ThreatType

use of com.google.webrisk.v1beta1.ThreatType in project java-webrisk by googleapis.

the class WebRiskServiceClientTest method searchUrisTest.

@Test
public void searchUrisTest() throws Exception {
    SearchUrisResponse expectedResponse = SearchUrisResponse.newBuilder().setThreat(SearchUrisResponse.ThreatUri.newBuilder().build()).build();
    mockWebRiskService.addResponse(expectedResponse);
    String uri = "uri116076";
    List<ThreatType> threatTypes = new ArrayList<>();
    SearchUrisResponse actualResponse = client.searchUris(uri, threatTypes);
    Assert.assertEquals(expectedResponse, actualResponse);
    List<AbstractMessage> actualRequests = mockWebRiskService.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()));
}
Also used : SearchUrisRequest(com.google.webrisk.v1.SearchUrisRequest) AbstractMessage(com.google.protobuf.AbstractMessage) ThreatType(com.google.webrisk.v1.ThreatType) ArrayList(java.util.ArrayList) ByteString(com.google.protobuf.ByteString) SearchUrisResponse(com.google.webrisk.v1.SearchUrisResponse) Test(org.junit.Test)

Example 13 with ThreatType

use of com.google.webrisk.v1beta1.ThreatType in project java-webrisk by googleapis.

the class WebRiskServiceV1Beta1ClientTest method searchUrisExceptionTest.

@Test
public void searchUrisExceptionTest() throws Exception {
    StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
    mockWebRiskServiceV1Beta1.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.
    }
}
Also used : InvalidArgumentException(com.google.api.gax.rpc.InvalidArgumentException) ThreatType(com.google.webrisk.v1beta1.ThreatType) StatusRuntimeException(io.grpc.StatusRuntimeException) ArrayList(java.util.ArrayList) ByteString(com.google.protobuf.ByteString) Test(org.junit.Test)

Example 14 with ThreatType

use of com.google.webrisk.v1beta1.ThreatType in project java-webrisk by googleapis.

the class SearchUriExampleTest method testSearchWithThreat.

@Test
public void testSearchWithThreat() throws IOException {
    // The URL to be searched
    String uri = "http://testsafebrowsing.appspot.com/s/malware.html";
    SearchUrisResponse actualResponse = SearchUriExample.searchUriExample(uri);
    List<ThreatType> type = actualResponse.getThreat().getThreatTypesList();
    Truth.assertThat(type).contains(ThreatType.MALWARE);
}
Also used : ThreatType(com.google.webrisk.v1.ThreatType) SearchUrisResponse(com.google.webrisk.v1.SearchUrisResponse) Test(org.junit.Test)

Aggregations

Test (org.junit.Test)14 ByteString (com.google.protobuf.ByteString)12 ThreatType (com.google.webrisk.v1.ThreatType)8 ArrayList (java.util.ArrayList)8 InvalidArgumentException (com.google.api.gax.rpc.InvalidArgumentException)6 AbstractMessage (com.google.protobuf.AbstractMessage)6 ThreatType (com.google.webrisk.v1beta1.ThreatType)6 StatusRuntimeException (io.grpc.StatusRuntimeException)6 SearchUrisResponse (com.google.webrisk.v1.SearchUrisResponse)3 ComputeThreatListDiffRequest (com.google.webrisk.v1.ComputeThreatListDiffRequest)2 ComputeThreatListDiffRequest (com.google.webrisk.v1beta1.ComputeThreatListDiffRequest)2 ComputeThreatListDiffResponse (com.google.webrisk.v1.ComputeThreatListDiffResponse)1 SearchHashesRequest (com.google.webrisk.v1.SearchHashesRequest)1 SearchHashesResponse (com.google.webrisk.v1.SearchHashesResponse)1 SearchUrisRequest (com.google.webrisk.v1.SearchUrisRequest)1 ComputeThreatListDiffResponse (com.google.webrisk.v1beta1.ComputeThreatListDiffResponse)1 SearchHashesRequest (com.google.webrisk.v1beta1.SearchHashesRequest)1 SearchHashesResponse (com.google.webrisk.v1beta1.SearchHashesResponse)1 SearchUrisRequest (com.google.webrisk.v1beta1.SearchUrisRequest)1 SearchUrisResponse (com.google.webrisk.v1beta1.SearchUrisResponse)1