Add corner case test (#11970)

rishav887 · web-flow · commit e5805a47f2b2 · 2023-05-02T10:49:42.000+02:00
Add some tests
diff --git a/java/test/com/thoughtworks/selenium/BaseSuite.java b/java/test/com/thoughtworks/selenium/BaseSuite.java
@@ -38,20 +38,26 @@ public static void setup() {
 
   }
 
+
   @AfterAll
   public static void teardown() {
     // browser
-    log.info("Stopping browser");
+    log.info("Stopping browser...");
     try {
-      InternalSelenseTestBase.destroyDriver();
-    } catch (SeleniumException ignored) {
-      // Nothing sane to do
+        InternalSelenseTestBase.destroyDriver();
+        log.info("Browser stopped successfully.");
+    } catch (SeleniumException e) {
+        log.error("Failed to stop browser: {}", e.getMessage());
     }
 
     // testEnvironment
-    log.info("Cleaning test environment");
-    GlobalTestEnvironment.stop();
-
-  }
+    log.info("Cleaning test environment...");
+    try {
+        GlobalTestEnvironment.stop();
+        log.info("Test environment cleaned successfully.");
+    } catch (Exception e) {
+        log.error("Failed to clean test environment: {}", e.getMessage());
+    }
+}
 
 }
diff --git a/java/test/com/thoughtworks/selenium/HttpCommandProcessorUnitTest.java b/java/test/com/thoughtworks/selenium/HttpCommandProcessorUnitTest.java
@@ -127,6 +127,20 @@ void testGetBooleanArray() {
     assertEquals(boolCmdResults[1], methodResults[1]);
   }
 
+  @Test
+ void testResourcesClosedWhenIoeOnGetOutputStream() {
+    IOEThrowingHttpCommandProcessor cmdProc = new IOEThrowingHttpCommandProcessor(
+            "localhost", 4444, "*chrome", "http://www.google.com");
+    cmdProc.throwIoeOnGetOutputStream = true;
+    try {
+        cmdProc.getCommandResponseAsString("testCommand");
+        fail();
+    } catch (IOException ioe) {
+        cmdProc.verifyClosedResources(false, true, true);
+    }
+}
+
+
   /**
    * Inner class to help mock out the network and pipe connections to verify that they are closed
    * regardless of where IOExceptions occur.
diff --git a/java/test/com/thoughtworks/selenium/condition/ConditionTest.java b/java/test/com/thoughtworks/selenium/condition/ConditionTest.java
@@ -188,6 +188,23 @@ public boolean isTrue(ConditionRunner.Context runner) {
     }
   }
 
+  @Test
+  void testAssertionFailureInConditionIsNotWrapped() {
+    Condition condition = new Condition() {
+        @Override
+        public boolean isTrue(ConditionRunner.Context runner) {
+            double result = 10.0 / 0.0;
+            return false;
+        }
+    };
+    try {
+        conditionRunner.waitFor(condition);
+        fail("should have thrown an assertion failed error");
+    } catch (AssertionError expected) {
+        assertTrue(expected.getMessage().contains("/ by zero"));
+    }
+}
+
   @Test
   void testMessageWithArgs() {
     final RuntimeException thrownException = new RuntimeException();
