TEZ-4364: TestFaultTolerance timeout on master

apache · Dec 27, 2021 · 5929c6b · 5929c6b
1 parent c9b8e90
commit 5929c6b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 8 deletions.
diff --git a/tez-api/src/main/java/org/apache/tez/runtime/api/events/InputReadErrorEvent.java b/tez-api/src/main/java/org/apache/tez/runtime/api/events/InputReadErrorEvent.java
@@ -21,6 +21,8 @@
 import org.apache.hadoop.classification.InterfaceAudience.Public;
 import org.apache.tez.runtime.api.Event;
 
+import com.google.common.annotations.VisibleForTesting;
+
 import java.util.Objects;
 
 /**
@@ -79,11 +81,14 @@ private InputReadErrorEvent(final String diagnostics, final int index, final int
     this.destinationLocalhostName = destinationLocalhostName;
   }
 
-  public static InputReadErrorEvent create(String diagnostics, int index, int version,
-      boolean isLocalFetch, boolean isDiskErrorAtSource) {
-    return create(diagnostics, index, version, 1, isLocalFetch, isDiskErrorAtSource, null);
-  }
-
+  /**
+   * Creates an InputReadErrorEvent from less parameters, supposed to be used from tests only.
+   * @param diagnostics
+   * @param index
+   * @param version
+   * @return InputReadErrorEvent instance
+   */
+  @VisibleForTesting
   public static InputReadErrorEvent create(String diagnostics, int index, int version) {
     return create(diagnostics, index, version, 1, false, false, null);
   }

diff --git a/tez-tests/src/test/java/org/apache/tez/test/TestInput.java b/tez-tests/src/test/java/org/apache/tez/test/TestInput.java
@@ -175,7 +175,7 @@ public int doRead() {
               for (int i=0; i<getNumPhysicalInputs(); ++i) {
                 String msg = ("FailingInput: " + getContext().getUniqueIdentifier() + 
                     " index: " + i + " version: " + lastInputReadyValue);
-                events.add(InputReadErrorEvent.create(msg, i, lastInputReadyValue));
+                events.add(InputReadErrorEvent.create(msg, i, lastInputReadyValue, 1, false, false, "localhost"));
                 LOG.info("Failing input: " + msg);
               }
             } else {
@@ -189,7 +189,8 @@ public int doRead() {
                 }
                 String msg = ("FailingInput: " + getContext().getUniqueIdentifier() + 
                     " index: " + index.intValue() + " version: " + lastInputReadyValue);
-                events.add(InputReadErrorEvent.create(msg, index.intValue(), lastInputReadyValue));
+                events.add(InputReadErrorEvent.create(msg, index.intValue(), lastInputReadyValue, 1, false, false,
+                    "localhost"));
                 LOG.info("Failing input: " + msg);
               }
             }
@@ -263,7 +264,7 @@ public int doRead() {
                             getContext().getUniqueIdentifier() + 
                             " index: " + index + " version: " + sourceInputVersion;
               LOG.info(msg);
-              events.add(InputReadErrorEvent.create(msg, index, sourceInputVersion));
+              events.add(InputReadErrorEvent.create(msg, index, sourceInputVersion, 1, false, false, "localhost"));
             }
           }
         }