Fix thee retryAnalyzer Annotations

RetryAnalyzer is set to Class.class which masks the actual
Test failure errror.
Mask optional exit code from the LocalCliProcess. Presto when
query fails exits with 1, but there are tests to verify failure.

Author: Arunachalam Thirupathi

tempto-core/src/main/java/io/prestodb/tempto/internal/convention/ConventionBasedTestProxyGenerator.java

@@ -25,8 +25,10 @@
 import net.bytebuddy.dynamic.loading.ClassLoadingStrategy;
 import net.bytebuddy.implementation.MethodCall;
 import org.slf4j.Logger;
+import org.testng.IRetryAnalyzer;
 import org.testng.annotations.CustomAttribute;
 import org.testng.annotations.Test;
+import org.testng.internal.annotations.DisabledRetryAnalyzer;
 
 import java.lang.annotation.Annotation;
 import java.lang.reflect.InvocationTargetException;
@@ -267,9 +269,9 @@ public boolean singleThreaded()
         }
 
         @Override
-        public Class retryAnalyzer()
+        public Class<? extends IRetryAnalyzer> retryAnalyzer()
         {
-            return Class.class;
+            return DisabledRetryAnalyzer.class;
         }
 
         @Override

tempto-core/src/main/java/io/prestodb/tempto/process/LocalCliProcess.java

@@ -16,7 +16,9 @@
 import io.prestodb.tempto.internal.process.CliProcessBase;
 
 import java.time.Duration;
+import java.util.OptionalInt;
 
+import static java.util.Objects.requireNonNull;
 import static java.util.concurrent.TimeUnit.MILLISECONDS;
 
 /**
@@ -26,11 +28,18 @@ public class LocalCliProcess
         extends CliProcessBase
 {
     private final Process process;
+    private final OptionalInt expectedExitValue;
 
-    public LocalCliProcess(Process process)
+    public LocalCliProcess(Process process, OptionalInt expectedExitValue)
     {
         super(process.getInputStream(), process.getErrorStream(), process.getOutputStream());
         this.process = process;
+        this.expectedExitValue = requireNonNull(expectedExitValue, "expectedExitValue is null");
+    }
+
+    public LocalCliProcess(Process process)
+    {
+        this(process, OptionalInt.empty());
     }
 
     @Override
@@ -43,7 +52,7 @@ public void waitForWithTimeoutAndKill(Duration timeout)
         }
 
         int exitValue = process.exitValue();
-        if (exitValue != 0) {
+        if (exitValue != 0 && exitValue != expectedExitValue.orElse(0)) {
             throw new RuntimeException("Child process exited with non-zero code: " + exitValue);
         }
     }