@java.lang.Override public Builder newBuilderForType() { return newBuilder(); }
private ListenResponse docDelete(String docPath) { ListenResponse.Builder response = ListenResponse.newBuilder(); response.setDocumentDelete( DocumentDelete.newBuilder() .addRemovedTargetIds(TARGET_ID) .setDocument(String.format("%s/documents/%s", DATABASE_NAME, docPath))); return response.build(); }
private ListenResponse docRemove(String docPath) { ListenResponse.Builder response = ListenResponse.newBuilder(); response.setDocumentRemove( DocumentRemove.newBuilder() .addRemovedTargetIds(TARGET_ID) .setDocument(String.format("%s/documents/%s", DATABASE_NAME, docPath))); return response.build(); }
private ListenResponse reset() { ListenResponse.Builder response = ListenResponse.newBuilder(); response.setTargetChange( TargetChange.newBuilder() .setTargetChangeType(TargetChangeType.RESET) .addTargetIds(TARGET_ID)); return response.build(); }
private ListenResponse filter(int documentCount) { ListenResponse.Builder response = ListenResponse.newBuilder(); response.setFilter(ExistenceFilter.newBuilder().setCount(documentCount).build()); return response.build(); }
private ListenResponse addTarget() { ListenResponse.Builder response = ListenResponse.newBuilder(); response.setTargetChange( TargetChange.newBuilder() .setTargetChangeType(TargetChangeType.ADD) .addTargetIds(TARGET_ID)); return response.build(); }
private ListenResponse current() { ListenResponse.Builder response = ListenResponse.newBuilder(); response.setTargetChange( TargetChange.newBuilder() .setTargetChangeType(TargetChangeType.CURRENT) .addTargetIds(TARGET_ID)); return response.build(); }
private ListenResponse removeTarget(@Nullable Code code) { TargetChange.Builder targetChange = TargetChange.newBuilder() .setTargetChangeType(TargetChangeType.REMOVE) .addTargetIds(TARGET_ID); if (code != null) { targetChange.setCause(Status.newBuilder().setCode(code.value())); } ListenResponse.Builder response = ListenResponse.newBuilder(); response.setTargetChange(targetChange); return response.build(); }
private ListenResponse snapshot() { ListenResponse.Builder response = ListenResponse.newBuilder(); response.setTargetChange( TargetChange.newBuilder() .setTargetChangeType(TargetChangeType.NO_CHANGE) .setReadTime(Timestamp.getDefaultInstance()) .setResumeToken(RESUME_TOKEN)); return response.build(); }
private ListenResponse doc(String docPath, Map<String, Value> singleFieldProto) { DocumentChange.Builder documentChange = DocumentChange.newBuilder(); documentChange.addTargetIds(TARGET_ID); documentChange.setDocument( Document.newBuilder() .setName(String.format("%s/documents/%s", DATABASE_NAME, docPath)) .putAllFields(singleFieldProto) .setUpdateTime(updateTime())); ListenResponse.Builder response = ListenResponse.newBuilder(); response.setDocumentChange(documentChange); return response.build(); }
@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)); }