test

Author: rli

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/services/telemetry/PluginResolver.kt

@@ -45,7 +45,7 @@ class PluginResolver private constructor(callerStackTrace: Array<StackTraceEleme
          * Sets a PluginResolver instance to a thread-local for the current thread.
          * This value will be retrieved by subsequent calls to fromCurrentThread.
          */
-        fun setThreadLocal(value: PluginResolver) {
+        fun setThreadLocal(value: PluginResolver?) {
             threadLocalResolver.set(value)
         }
     }

plugins/core/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/telemetry/PluginResolverTest.kt

@@ -6,28 +6,25 @@ package software.aws.toolkits.jetbrains.services.telemetry
 import com.intellij.ide.plugins.IdeaPluginDescriptor
 import com.intellij.ide.plugins.PluginManagerCore
 import io.mockk.called
-import io.mockk.clearAllMocks
 import io.mockk.every
+import io.mockk.junit5.MockKExtension
 import io.mockk.mockk
 import io.mockk.mockkStatic
 import io.mockk.verify
 import junit.framework.TestCase.assertEquals
-import org.junit.jupiter.api.AfterEach
 import org.junit.jupiter.api.BeforeEach
 import org.junit.jupiter.api.Test
+import org.junit.jupiter.api.extension.ExtendWith
 import software.amazon.awssdk.services.toolkittelemetry.model.AWSProduct
 
+@ExtendWith(MockKExtension::class)
 class PluginResolverTest {
     @BeforeEach
     fun setup() {
+        PluginResolver.setThreadLocal(null)
         mockkStatic(PluginManagerCore::class)
     }
 
-    @AfterEach
-    fun tearDown() {
-        clearAllMocks()
-    }
-
     @Test
     fun getsProductForAmazonQPlugin() {
         val pluginDescriptor = mockk<IdeaPluginDescriptor> {