Implement clearStaticMockk for KFunction and KProperty

krocard · krocard · commit 4917ecce69b6 · 2024-04-02T18:49:29.000+02:00
diff --git a/modules/mockk/src/jvmMain/kotlin/io/mockk/MockK.kt b/modules/mockk/src/jvmMain/kotlin/io/mockk/MockK.kt
@@ -44,6 +44,18 @@ fun unmockkStatic(vararg functions: KFunction<*>) =
 fun unmockkStatic(vararg functions: KProperty<*>) =
     unmockkStatic(*functions.map(KProperty<*>::getter).toTypedArray())
 
+/**
+ * Clear static mocks.
+ */
+fun clearStaticMockk(vararg functions: KFunction<*>) =
+    clearStaticMockk(*functions.map { it.declaringKotlinFile }.toTypedArray())
+
+/**
+ * Clear static mocks.
+ */
+fun clearStaticMockk(vararg functions: KProperty<*>) =
+    clearStaticMockk(*functions.map(KProperty<*>::getter).toTypedArray())
+
 /**
  * Builds a static mock and unmocks it after the block has been executed.
  */
diff --git a/modules/mockk/src/jvmTest/kotlin/io/mockk/it/StaticMockkTest.kt b/modules/mockk/src/jvmTest/kotlin/io/mockk/it/StaticMockkTest.kt
@@ -1,5 +1,6 @@
 package io.mockk.it
 
+import io.mockk.clearStaticMockk
 import io.mockk.every
 import io.mockk.mockkStatic
 import io.mockk.unmockkStatic
@@ -73,6 +74,19 @@ class StaticMockkTest {
         verify { 5 op 6 }
     }
 
+    @Test
+    fun extensionFunctionClearStaticMock() {
+        mockkStatic(Int::op)
+
+        every { 5 op 6 } returns 2
+
+        assertEquals(2, 5 op 6)
+
+        clearStaticMockk(Int::op)
+
+        verify(exactly = 0) { 5 op 6 }
+    }
+
     @Test
     fun extensionPropertyStaticMock() {
         mockkStatic(Int::selfOp)
@@ -106,4 +120,17 @@ class StaticMockkTest {
 
         verify { 5.selfOp }
     }
+
+    @Test
+    fun extensionPropertyClearStaticMock() {
+        mockkStatic(Int::selfOp)
+
+        every { 5.selfOp } returns 2
+
+        assertEquals(2, 5.selfOp)
+
+        clearStaticMockk(Int::selfOp)
+
+        verify(exactly = 0) { 5.selfOp }
+    }
 }
