@@ -5,32 +5,27 @@ package software.aws.toolkits.jetbrains.core.credentials
55
66import com.intellij.openapi.project.Project
77import com.intellij.testFramework.ApplicationExtension
8- import io.mockk.clearAllMocks
9- import org.junit.jupiter.api.BeforeEach
10- import org.junit.jupiter.api.Test
11- import org.junit.jupiter.api.TestInstance
12- import org.mockito.kotlin.mock
13- import org.mockito.kotlin.never
14- import io.mockk.mockkStatic
15- import io.mockk.verify
168import io.mockk.every
179import io.mockk.just
1810import io.mockk.mockkObject
11+ import io.mockk.mockkStatic
1912import io.mockk.runs
20- import org.mockito.kotlin.verify
21- import org.mockito.kotlin.whenever
13+ import io.mockk.verify
2214import org.junit.jupiter.api.Assertions.*
15+ import org.junit.jupiter.api.BeforeEach
16+ import org.junit.jupiter.api.Test
17+ import org.junit.jupiter.api.TestInstance
2318import org.junit.jupiter.api.extension.ExtendWith
2419import org.mockito.kotlin.doThrow
20+ import org.mockito.kotlin.mock
2521import org.mockito.kotlin.reset
26- import software.amazon.awssdk.auth.token.credentials.SdkToken
22+ import org.mockito.kotlin.whenever
2723import software.aws.toolkits.jetbrains.core.credentials.sso.DeviceAuthorizationGrantToken
2824import software.aws.toolkits.jetbrains.core.credentials.sso.bearer.BearerTokenAuthState
2925import software.aws.toolkits.jetbrains.core.credentials.sso.bearer.BearerTokenProvider
3026import software.aws.toolkits.jetbrains.core.credentials.sso.bearer.BearerTokenProviderListener
3127import software.aws.toolkits.jetbrains.utils.notifyInfo
3228import software.aws.toolkits.resources.AwsCoreBundle.message
33- import java.net.UnknownHostException
3429import java.time.Instant
3530import java.time.temporal.ChronoUnit
3631
@@ -47,7 +42,7 @@ class ToolkitAuthManagerTest {
4742 tokenProvider = mock()
4843 reauthCallCount = 0
4944 val field = Class .forName(" software.aws.toolkits.jetbrains.core.credentials.ToolkitAuthManagerKt" )
50- .getDeclaredField(" hasNotifiedNetworkErrorOnce " )
45+ .getDeclaredField(" hasSeenFirstNetworkError " )
5146 field.isAccessible = true
5247 field.set(null , false )
5348
@@ -59,7 +54,6 @@ class ToolkitAuthManagerTest {
5954 every { BearerTokenProviderListener .notifyCredUpdate(any<String >()) } just runs
6055 }
6156
62-
6357 @Test
6458 fun `test NEEDS_REFRESH state with network error - first occurrence` () {
6559 whenever(tokenProvider.state()).thenReturn(BearerTokenAuthState .NEEDS_REFRESH )
@@ -75,7 +69,7 @@ class ToolkitAuthManagerTest {
7569
7670 assertFalse(result)
7771 assertEquals(0 , reauthCallCount)
78- verify(exactly = 1 ){
72+ verify(exactly = 1 ) {
7973 notifyInfo(
8074 message(" general.auth.network.error" ),
8175 message(" general.auth.network.error.message" ),
0 commit comments