Sometimes test start in one thread and finished in another

Author: pnatashap

jdi-light-html-tests/src/test/java/io/github/epam/testng/TestNGListener.java

@@ -5,7 +5,6 @@
  * Email: [email protected]; Skype: roman.iovlev
  */
 
-import com.jdiai.tools.Safe;
 import org.testng.IInvokedMethod;
 import org.testng.IInvokedMethodListener;
 import org.testng.ITestNGMethod;
@@ -22,7 +21,7 @@
 import static java.lang.System.currentTimeMillis;
 
 public class TestNGListener implements IInvokedMethodListener {
-    private Safe<Long> start = new Safe<>(0L);
+    private Long start = currentTimeMillis();
 
     @Override
     public void beforeInvocation(IInvokedMethod m, ITestResult tr) {
@@ -32,7 +31,7 @@ public void beforeInvocation(IInvokedMethod m, ITestResult tr) {
                 String testName = last(testMethod.getTestClass().getName().split("\\.")) +
                         "." + testMethod.getMethodName();
                 TEST_NAME.set(testName);
-                start.set(currentTimeMillis());
+                start = currentTimeMillis();
                 logger.step("== Test '%s' START ==", testName);
             }
         }
@@ -44,7 +43,7 @@ public void afterInvocation(IInvokedMethod method, ITestResult tr) {
             String result = getTestResult(tr);
             logger.step("=== Test '%s' %s [%s] ===", TEST_NAME.get(), result,
                     new SimpleDateFormat("mm:ss.SS")
-                            .format(new Date(currentTimeMillis() - start.get())));
+                            .format(new Date(currentTimeMillis() - start)));
             if ("FAILED".equals(result)) {
                 try {
                     screenshotStep("On Fail Screenshot");