diff --git a/src/test/java/com/firebolt/jdbc/client/gateway/FireboltGatewayUrlClientTest.java b/src/test/java/com/firebolt/jdbc/client/gateway/FireboltGatewayUrlClientTest.java index 83e6b016..360eac32 100644 --- a/src/test/java/com/firebolt/jdbc/client/gateway/FireboltGatewayUrlClientTest.java +++ b/src/test/java/com/firebolt/jdbc/client/gateway/FireboltGatewayUrlClientTest.java @@ -138,7 +138,6 @@ private FireboltAccountRetriever mockAccountRetriever(String path, Class< @CsvSource({ HTTP_BAD_REQUEST + "," + GENERIC_ERROR_MESSAGE, HTTP_PAYMENT_REQUIRED + "," + GENERIC_ERROR_MESSAGE, - HTTP_FORBIDDEN + "," + GENERIC_ERROR_MESSAGE, HTTP_BAD_METHOD + "," + GENERIC_ERROR_MESSAGE, HTTP_NOT_ACCEPTABLE + "," + GENERIC_ERROR_MESSAGE, HTTP_PROXY_AUTH + "," + GENERIC_ERROR_MESSAGE, @@ -157,8 +156,12 @@ private FireboltAccountRetriever mockAccountRetriever(String path, Class< HTTP_VERSION + "," + GENERIC_ERROR_MESSAGE, HTTP_NOT_FOUND + "," + "Account '%s' does not exist", - HTTP_UNAVAILABLE + "," + "Could not query Firebolt at https://test-firebolt.io/web/v3/account/%s/%s. The engine is not running.", - HTTP_UNAUTHORIZED + "," + "Could not query Firebolt at https://test-firebolt.io/web/v3/account/%s/%s. The operation is not authorized" + HTTP_UNAVAILABLE + "," + + "Could not query Firebolt at https://test-firebolt.io/web/v3/account/%s/%s. The engine is not running.", + HTTP_FORBIDDEN + "," + + "Could not query Firebolt at https://test-firebolt.io/web/v3/account/%s/%s. The operation is not authorized", + HTTP_UNAUTHORIZED + "," + + "Could not query Firebolt at https://test-firebolt.io/web/v3/account/%s/%s. The operation is not authorized" }) void testFailedAccountDataRetrieving(int statusCode, String errorMessageTemplate) throws IOException { injectMockedResponse(httpClient, statusCode, null);