1- package net .snowflake .client .authentication ;
1+ package net .snowflake .client .jdbc ;
22
33import static org .hamcrest .CoreMatchers .is ;
44import static org .hamcrest .MatcherAssert .assertThat ;
1111import java .util .stream .Stream ;
1212import net .snowflake .client .AbstractDriverIT ;
1313import net .snowflake .client .category .TestTags ;
14+ import net .snowflake .client .core .HttpUtil ;
1415import net .snowflake .client .core .SecurityUtil ;
15- import net .snowflake .client .jdbc .BaseJDBCTest ;
16- import net .snowflake .client .jdbc .ErrorCode ;
17- import net .snowflake .client .jdbc .SnowflakeSQLException ;
16+ import org .junit .jupiter .api .AfterEach ;
1817import org .junit .jupiter .api .Tag ;
1918import org .junit .jupiter .params .ParameterizedTest ;
2019import org .junit .jupiter .params .provider .MethodSource ;
2120
22- @ Tag (TestTags .AUTHENTICATION )
23- public class PrivateKeyAuthenticationExceptionHandlingTest extends BaseJDBCTest {
21+ @ Tag (TestTags .CONNECTION )
22+ public class PrivateKeyAuthenticationExceptionHandlingLatestIT {
2423
2524 static Stream <String > jwtTimeoutProvider () {
2625 return Stream .of ("10" , "100" , null );
@@ -30,9 +29,14 @@ static Stream<String> timeOutSettings() {
3029 return Stream .of ("HTTP_CLIENT_CONNECTION_TIMEOUT" , "HTTP_CLIENT_SOCKET_TIMEOUT" );
3130 }
3231
32+ @ AfterEach
33+ void cleanup () {
34+ HttpUtil .reset ();
35+ }
36+
3337 /**
3438 * Tests the authentication exception and retry JWT renew functionality when retrying login
35- * requests. To run, update environment variables to use connect with JWT authentication.
39+ * requests
3640 */
3741 @ ParameterizedTest
3842 @ MethodSource ("jwtTimeoutProvider" )
0 commit comments