You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tez.apache.org by je...@apache.org on 2020/04/30 09:52:55 UTC
[tez] branch master updated: TEZ-4163. Speed up TestTaskReporter
This is an automated email from the ASF dual-hosted git repository.
jeagles pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/tez.git
The following commit(s) were added to refs/heads/master by this push:
new 5373ca8 TEZ-4163. Speed up TestTaskReporter
5373ca8 is described below
commit 5373ca8ff06cc315c63ac66448daaa31c9b5882e
Author: Jonathan Eagles <je...@apache.org>
AuthorDate: Thu Apr 30 04:52:36 2020 -0500
TEZ-4163. Speed up TestTaskReporter
---
.../test/java/org/apache/tez/runtime/task/TestTaskReporter.java | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/tez-runtime-internals/src/test/java/org/apache/tez/runtime/task/TestTaskReporter.java b/tez-runtime-internals/src/test/java/org/apache/tez/runtime/task/TestTaskReporter.java
index 04c467a..9f514f4 100644
--- a/tez-runtime-internals/src/test/java/org/apache/tez/runtime/task/TestTaskReporter.java
+++ b/tez-runtime-internals/src/test/java/org/apache/tez/runtime/task/TestTaskReporter.java
@@ -60,6 +60,7 @@ public class TestTaskReporter {
final Object lock = new Object();
final AtomicBoolean hb2Done = new AtomicBoolean(false);
+ final int maxEvents = 5;
TezTaskUmbilicalProtocol mockUmbilical = mock(TezTaskUmbilicalProtocol.class);
doAnswer(new Answer() {
@@ -68,7 +69,7 @@ public class TestTaskReporter {
Object[] args = invocation.getArguments();
TezHeartbeatRequest request = (TezHeartbeatRequest) args[0];
if (request.getRequestId() == 1 || request.getRequestId() == 2) {
- TezHeartbeatResponse response = new TezHeartbeatResponse(createEvents(5));
+ TezHeartbeatResponse response = new TezHeartbeatResponse(createEvents(maxEvents));
response.setLastRequestId(request.getRequestId());
return response;
} else if (request.getRequestId() == 3) {
@@ -92,7 +93,7 @@ public class TestTaskReporter {
// Setup the sleep time to be way higher than the test timeout
TaskReporter.HeartbeatCallable heartbeatCallable =
- new TaskReporter.HeartbeatCallable(mockTask, mockUmbilical, 100000, 100000, 5,
+ new TaskReporter.HeartbeatCallable(mockTask, mockUmbilical, 100000, 100000, maxEvents,
new AtomicLong(0),
"containerIdStr");
@@ -105,8 +106,8 @@ public class TestTaskReporter {
}
}
verify(mockUmbilical, times(3)).heartbeat(any(TezHeartbeatRequest.class));
- Thread.sleep(2000l);
- // Sleep for 2 seconds, less than the callable sleep time. No more invocations.
+ Thread.sleep(200l);
+ // Sleep for less than the callable sleep time. No more invocations.
verify(mockUmbilical, times(3)).heartbeat(any(TezHeartbeatRequest.class));
} finally {
executor.shutdownNow();