@@ -144,7 +144,7 @@ public void testShouldNotSetCauseIfResponseValueIsJustAString() {
144
144
handler .throwIfResponseFailed (createResponse (ErrorCodes .UNHANDLED_ERROR , "boom" ), 123 );
145
145
fail ("Should have thrown!" );
146
146
} catch (WebDriverException expected ) {
147
- assertEquals (InvalidArgumentException .class , expected .getClass ());
147
+ assertEquals (WebDriverException .class , expected .getClass ());
148
148
assertDoesNotHaveACause (expected );
149
149
assertThat (expected .getMessage (), containsString ("boom" ));
150
150
assertThat (expected .getMessage (), containsString (new WebDriverException ().getMessage ()));
@@ -282,7 +282,7 @@ public void testShouldDefaultToWebDriverExceptionIfClassIsNotSpecified()
282
282
283
283
Throwable cause = expected .getCause ();
284
284
assertNotNull (cause );
285
- assertEquals (InvalidArgumentException .class , cause .getClass ());
285
+ assertEquals (WebDriverException .class , cause .getClass ());
286
286
assertEquals (new WebDriverException (serverError .getMessage ()).getMessage (),
287
287
cause .getMessage ());
288
288
assertStackTracesEqual (serverError .getStackTrace (), cause .getStackTrace ());
@@ -315,7 +315,7 @@ public void testShouldStillTryToBuildWebDriverExceptionIfClassIsNotProvidedAndSt
315
315
316
316
Throwable cause = expected .getCause ();
317
317
assertNotNull (cause );
318
- assertEquals (InvalidArgumentException .class , cause .getClass ());
318
+ assertEquals (WebDriverException .class , cause .getClass ());
319
319
assertEquals (helper .getMessage (),
320
320
cause .getMessage ());
321
321
@@ -349,7 +349,7 @@ public void testToleratesNonNumericLineNumber() {
349
349
350
350
Throwable cause = expected .getCause ();
351
351
assertNotNull (cause );
352
- assertEquals (InvalidArgumentException .class , cause .getClass ());
352
+ assertEquals (WebDriverException .class , cause .getClass ());
353
353
assertEquals (helper .getMessage (),
354
354
cause .getMessage ());
355
355
@@ -383,7 +383,7 @@ public void testToleratesNumericLineNumberAsString() {
383
383
384
384
Throwable cause = expected .getCause ();
385
385
assertNotNull (cause );
386
- assertEquals (InvalidArgumentException .class , cause .getClass ());
386
+ assertEquals (WebDriverException .class , cause .getClass ());
387
387
assertEquals (helper .getMessage (),
388
388
cause .getMessage ());
389
389
0 commit comments