edman124 commented on code in PR #29879:
URL: https://github.com/apache/beam/pull/29879#discussion_r1448156229
##########
runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/DataflowWorkUnitClientTest.java:
##########
@@ -227,10 +230,22 @@ public void testCloudServiceCallMultipleWorkItems()
throws Exception {
client.getWorkItem();
}
- // @Test
- // public void testWorkMessageCreation() throws Exception {
- // WorkUnitClient client = new DataflowWorkUnitClient(pipelineOptions,
LOG);
- // }
+ @Test
+ public void testReportWorkerMessageEmptyResponse() throws Exception {
+ MockLowLevelHttpResponse response = new MockLowLevelHttpResponse();
+ response.setContentType(Json.MEDIA_TYPE);
+ SendWorkerMessagesResponse workerMessage = new
SendWorkerMessagesResponse();
+ workerMessage.setFactory(Transport.getJsonFactory());
+ response.setContent(workerMessage.toPrettyString());
+ when(request.execute()).thenReturn(response);
+ StreamingScalingReport activeThreadsReport =
+ new StreamingScalingReport().setActiveThreadCount(1);
+ WorkUnitClient client = new DataflowWorkUnitClient(pipelineOptions, LOG);
+ WorkerMessageResponse expected = new WorkerMessageResponse();
+ WorkerMessageResponse clientResponse =
+ client.reportStreamingMetricsWorkerMessage(activeThreadsReport);
Review Comment:
Done.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
