@Test @SuppressWarnings("all") public void streamingRecognizeTest() throws Exception { StreamingRecognizeResponse expectedResponse = StreamingRecognizeResponse.newBuilder().build(); mockSpeech.addResponse(expectedResponse); StreamingRecognizeRequest request = StreamingRecognizeRequest.newBuilder().build(); MockStreamObserver<StreamingRecognizeResponse> responseObserver = new MockStreamObserver<>(); BidiStreamingCallable<StreamingRecognizeRequest, StreamingRecognizeResponse> callable = client.streamingRecognizeCallable(); ApiStreamObserver<StreamingRecognizeRequest> requestObserver = callable.bidiStreamingCall(responseObserver); requestObserver.onNext(request); requestObserver.onCompleted(); List<StreamingRecognizeResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void readRowsTest() throws Exception { ReadRowsResponse expectedResponse = ReadRowsResponse.newBuilder().build(); mockBigQueryStorage.addResponse(expectedResponse); StreamPosition readPosition = StreamPosition.newBuilder().build(); ReadRowsRequest request = ReadRowsRequest.newBuilder().setReadPosition(readPosition).build(); MockStreamObserver<ReadRowsResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<ReadRowsRequest, ReadRowsResponse> callable = client.readRowsCallable(); callable.serverStreamingCall(request, responseObserver); List<ReadRowsResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void readRowsTest() throws Exception { ByteString lastScannedRowKey = ByteString.copyFromUtf8("-126"); ReadRowsResponse expectedResponse = ReadRowsResponse.newBuilder().setLastScannedRowKey(lastScannedRowKey).build(); mockBigtable.addResponse(expectedResponse); TableName tableName = TableName.of("[PROJECT]", "[INSTANCE]", "[TABLE]"); ReadRowsRequest request = ReadRowsRequest.newBuilder().setTableName(tableName.toString()).build(); MockStreamObserver<ReadRowsResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<ReadRowsRequest, ReadRowsResponse> callable = client.readRowsCallable(); callable.serverStreamingCall(request, responseObserver); List<ReadRowsResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void readRowsTest() throws Exception { ReadRowsResponse expectedResponse = ReadRowsResponse.newBuilder().build(); mockBigQueryStorage.addResponse(expectedResponse); StreamPosition readPosition = StreamPosition.newBuilder().build(); ReadRowsRequest request = ReadRowsRequest.newBuilder().setReadPosition(readPosition).build(); MockStreamObserver<ReadRowsResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<ReadRowsRequest, ReadRowsResponse> callable = client.readRowsCallable(); callable.serverStreamingCall(request, responseObserver); List<ReadRowsResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void streamingRecognizeTest() throws Exception { StreamingRecognizeResponse expectedResponse = StreamingRecognizeResponse.newBuilder().build(); mockSpeech.addResponse(expectedResponse); StreamingRecognizeRequest request = StreamingRecognizeRequest.newBuilder().build(); MockStreamObserver<StreamingRecognizeResponse> responseObserver = new MockStreamObserver<>(); BidiStreamingCallable<StreamingRecognizeRequest, StreamingRecognizeResponse> callable = client.streamingRecognizeCallable(); ApiStreamObserver<StreamingRecognizeRequest> requestObserver = callable.bidiStreamingCall(responseObserver); requestObserver.onNext(request); requestObserver.onCompleted(); List<StreamingRecognizeResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void listenTest() throws Exception { ListenResponse expectedResponse = ListenResponse.newBuilder().build(); mockFirestore.addResponse(expectedResponse); String formattedDatabase = FirestoreClient.formatDatabaseRootName("[PROJECT]", "[DATABASE]"); ListenRequest request = ListenRequest.newBuilder().setDatabase(formattedDatabase).build(); MockStreamObserver<ListenResponse> responseObserver = new MockStreamObserver<>(); BidiStreamingCallable<ListenRequest, ListenResponse> callable = client.listenCallable(); ApiStreamObserver<ListenRequest> requestObserver = callable.bidiStreamingCall(responseObserver); requestObserver.onNext(request); requestObserver.onCompleted(); List<ListenResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void listenTest() throws Exception { ListenResponse expectedResponse = ListenResponse.newBuilder().build(); mockFirestore.addResponse(expectedResponse); String formattedDatabase = FirestoreClient.formatDatabaseRootName("[PROJECT]", "[DATABASE]"); ListenRequest request = ListenRequest.newBuilder().setDatabase(formattedDatabase).build(); MockStreamObserver<ListenResponse> responseObserver = new MockStreamObserver<>(); BidiStreamingCallable<ListenRequest, ListenResponse> callable = client.listenCallable(); ApiStreamObserver<ListenRequest> requestObserver = callable.bidiStreamingCall(responseObserver); requestObserver.onNext(request); requestObserver.onCompleted(); List<ListenResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void mutateRowsTest() throws Exception { MutateRowsResponse expectedResponse = MutateRowsResponse.newBuilder().build(); mockBigtable.addResponse(expectedResponse); TableName tableName = TableName.of("[PROJECT]", "[INSTANCE]", "[TABLE]"); List<MutateRowsRequest.Entry> entries = new ArrayList<>(); MutateRowsRequest request = MutateRowsRequest.newBuilder() .setTableName(tableName.toString()) .addAllEntries(entries) .build(); MockStreamObserver<MutateRowsResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<MutateRowsRequest, MutateRowsResponse> callable = client.mutateRowsCallable(); callable.serverStreamingCall(request, responseObserver); List<MutateRowsResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void runQueryTest() throws Exception { ByteString transaction = ByteString.copyFromUtf8("-34"); int skippedResults = 880286183; RunQueryResponse expectedResponse = RunQueryResponse.newBuilder() .setTransaction(transaction) .setSkippedResults(skippedResults) .build(); mockFirestore.addResponse(expectedResponse); String formattedParent = FirestoreClient.formatAnyPathName("[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]"); RunQueryRequest request = RunQueryRequest.newBuilder().setParent(formattedParent).build(); MockStreamObserver<RunQueryResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<RunQueryRequest, RunQueryResponse> callable = client.runQueryCallable(); callable.serverStreamingCall(request, responseObserver); List<RunQueryResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void writeTest() throws Exception { String streamId = "streamId-315624902"; ByteString streamToken = ByteString.copyFromUtf8("122"); WriteResponse expectedResponse = WriteResponse.newBuilder().setStreamId(streamId).setStreamToken(streamToken).build(); mockFirestore.addResponse(expectedResponse); String formattedDatabase = FirestoreClient.formatDatabaseRootName("[PROJECT]", "[DATABASE]"); WriteRequest request = WriteRequest.newBuilder().setDatabase(formattedDatabase).build(); MockStreamObserver<WriteResponse> responseObserver = new MockStreamObserver<>(); BidiStreamingCallable<WriteRequest, WriteResponse> callable = client.writeCallable(); ApiStreamObserver<WriteRequest> requestObserver = callable.bidiStreamingCall(responseObserver); requestObserver.onNext(request); requestObserver.onCompleted(); List<WriteResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void runQueryTest() throws Exception { ByteString transaction = ByteString.copyFromUtf8("-34"); int skippedResults = 880286183; RunQueryResponse expectedResponse = RunQueryResponse.newBuilder() .setTransaction(transaction) .setSkippedResults(skippedResults) .build(); mockFirestore.addResponse(expectedResponse); String formattedParent = FirestoreClient.formatAnyPathName("[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]"); RunQueryRequest request = RunQueryRequest.newBuilder().setParent(formattedParent).build(); MockStreamObserver<RunQueryResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<RunQueryRequest, RunQueryResponse> callable = client.runQueryCallable(); callable.serverStreamingCall(request, responseObserver); List<RunQueryResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void streamingRecognizeExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockSpeech.addException(exception); StreamingRecognizeRequest request = StreamingRecognizeRequest.newBuilder().build(); MockStreamObserver<StreamingRecognizeResponse> responseObserver = new MockStreamObserver<>(); BidiStreamingCallable<StreamingRecognizeRequest, StreamingRecognizeResponse> callable = client.streamingRecognizeCallable(); ApiStreamObserver<StreamingRecognizeRequest> requestObserver = callable.bidiStreamingCall(responseObserver); requestObserver.onNext(request); try { List<StreamingRecognizeResponse> actualResponses = responseObserver.future().get(); Assert.fail("No exception thrown"); } catch (ExecutionException e) { Assert.assertTrue(e.getCause() instanceof InvalidArgumentException); InvalidArgumentException apiException = (InvalidArgumentException) e.getCause(); Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode()); } } }
@Test @SuppressWarnings("all") public void sampleRowKeysTest() throws Exception { ByteString rowKey = ByteString.copyFromUtf8("122"); long offsetBytes = 889884095L; SampleRowKeysResponse expectedResponse = SampleRowKeysResponse.newBuilder().setRowKey(rowKey).setOffsetBytes(offsetBytes).build(); mockBigtable.addResponse(expectedResponse); TableName tableName = TableName.of("[PROJECT]", "[INSTANCE]", "[TABLE]"); SampleRowKeysRequest request = SampleRowKeysRequest.newBuilder().setTableName(tableName.toString()).build(); MockStreamObserver<SampleRowKeysResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<SampleRowKeysRequest, SampleRowKeysResponse> callable = client.sampleRowKeysCallable(); callable.serverStreamingCall(request, responseObserver); List<SampleRowKeysResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void batchGetDocumentsTest() throws Exception { String missing = "missing1069449574"; ByteString transaction = ByteString.copyFromUtf8("-34"); BatchGetDocumentsResponse expectedResponse = BatchGetDocumentsResponse.newBuilder() .setMissing(missing) .setTransaction(transaction) .build(); mockFirestore.addResponse(expectedResponse); String formattedDatabase = FirestoreClient.formatDatabaseRootName("[PROJECT]", "[DATABASE]"); List<String> documents = new ArrayList<>(); BatchGetDocumentsRequest request = BatchGetDocumentsRequest.newBuilder() .setDatabase(formattedDatabase) .addAllDocuments(documents) .build(); MockStreamObserver<BatchGetDocumentsResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<BatchGetDocumentsRequest, BatchGetDocumentsResponse> callable = client.batchGetDocumentsCallable(); callable.serverStreamingCall(request, responseObserver); List<BatchGetDocumentsResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void batchGetDocumentsTest() throws Exception { String missing = "missing1069449574"; ByteString transaction = ByteString.copyFromUtf8("-34"); BatchGetDocumentsResponse expectedResponse = BatchGetDocumentsResponse.newBuilder() .setMissing(missing) .setTransaction(transaction) .build(); mockFirestore.addResponse(expectedResponse); String formattedDatabase = FirestoreClient.formatDatabaseRootName("[PROJECT]", "[DATABASE]"); List<String> documents = new ArrayList<>(); BatchGetDocumentsRequest request = BatchGetDocumentsRequest.newBuilder() .setDatabase(formattedDatabase) .addAllDocuments(documents) .build(); MockStreamObserver<BatchGetDocumentsResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<BatchGetDocumentsRequest, BatchGetDocumentsResponse> callable = client.batchGetDocumentsCallable(); callable.serverStreamingCall(request, responseObserver); List<BatchGetDocumentsResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void executeStreamingSqlTest() throws Exception { boolean chunkedValue = true; ByteString resumeToken = ByteString.copyFromUtf8("103"); PartialResultSet expectedResponse = PartialResultSet.newBuilder() .setChunkedValue(chunkedValue) .setResumeToken(resumeToken) .build(); mockSpanner.addResponse(expectedResponse); SessionName session = SessionName.of("[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]"); String sql = "sql114126"; ExecuteSqlRequest request = ExecuteSqlRequest.newBuilder().setSession(session.toString()).setSql(sql).build(); MockStreamObserver<PartialResultSet> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<ExecuteSqlRequest, PartialResultSet> callable = client.executeStreamingSqlCallable(); callable.serverStreamingCall(request, responseObserver); List<PartialResultSet> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void writeTest() throws Exception { String streamId = "streamId-315624902"; ByteString streamToken = ByteString.copyFromUtf8("122"); WriteResponse expectedResponse = WriteResponse.newBuilder().setStreamId(streamId).setStreamToken(streamToken).build(); mockFirestore.addResponse(expectedResponse); String formattedDatabase = FirestoreClient.formatDatabaseRootName("[PROJECT]", "[DATABASE]"); WriteRequest request = WriteRequest.newBuilder().setDatabase(formattedDatabase).build(); MockStreamObserver<WriteResponse> responseObserver = new MockStreamObserver<>(); BidiStreamingCallable<WriteRequest, WriteResponse> callable = client.writeCallable(); ApiStreamObserver<WriteRequest> requestObserver = callable.bidiStreamingCall(responseObserver); requestObserver.onNext(request); requestObserver.onCompleted(); List<WriteResponse> actualResponses = responseObserver.future().get(); Assert.assertEquals(1, actualResponses.size()); Assert.assertEquals(expectedResponse, actualResponses.get(0)); }
@Test @SuppressWarnings("all") public void readRowsExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockBigQueryStorage.addException(exception); StreamPosition readPosition = StreamPosition.newBuilder().build(); ReadRowsRequest request = ReadRowsRequest.newBuilder().setReadPosition(readPosition).build(); MockStreamObserver<ReadRowsResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<ReadRowsRequest, ReadRowsResponse> callable = client.readRowsCallable(); callable.serverStreamingCall(request, responseObserver); try { List<ReadRowsResponse> actualResponses = responseObserver.future().get(); Assert.fail("No exception thrown"); } catch (ExecutionException e) { Assert.assertTrue(e.getCause() instanceof InvalidArgumentException); InvalidArgumentException apiException = (InvalidArgumentException) e.getCause(); Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode()); } }
@Test @SuppressWarnings("all") public void streamingRecognizeExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockSpeech.addException(exception); StreamingRecognizeRequest request = StreamingRecognizeRequest.newBuilder().build(); MockStreamObserver<StreamingRecognizeResponse> responseObserver = new MockStreamObserver<>(); BidiStreamingCallable<StreamingRecognizeRequest, StreamingRecognizeResponse> callable = client.streamingRecognizeCallable(); ApiStreamObserver<StreamingRecognizeRequest> requestObserver = callable.bidiStreamingCall(responseObserver); requestObserver.onNext(request); try { List<StreamingRecognizeResponse> actualResponses = responseObserver.future().get(); Assert.fail("No exception thrown"); } catch (ExecutionException e) { Assert.assertTrue(e.getCause() instanceof InvalidArgumentException); InvalidArgumentException apiException = (InvalidArgumentException) e.getCause(); Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode()); } } }
@Test @SuppressWarnings("all") public void runQueryExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockFirestore.addException(exception); String formattedParent = FirestoreClient.formatAnyPathName("[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]"); RunQueryRequest request = RunQueryRequest.newBuilder().setParent(formattedParent).build(); MockStreamObserver<RunQueryResponse> responseObserver = new MockStreamObserver<>(); ServerStreamingCallable<RunQueryRequest, RunQueryResponse> callable = client.runQueryCallable(); callable.serverStreamingCall(request, responseObserver); try { List<RunQueryResponse> actualResponses = responseObserver.future().get(); Assert.fail("No exception thrown"); } catch (ExecutionException e) { Assert.assertTrue(e.getCause() instanceof InvalidArgumentException); InvalidArgumentException apiException = (InvalidArgumentException) e.getCause(); Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode()); } }