Implement clearStaticMockk for KFunction and KProperty

mockk · Apr 2, 2024 · db48a53 · db48a53
1 parent 143f6bb
commit db48a53
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/modules/mockk/src/commonTest/kotlin/io/mockk/it/StaticMockTest.kt b/modules/mockk/src/commonTest/kotlin/io/mockk/it/StaticMockTest.kt
@@ -13,7 +13,7 @@ class StaticMockTest {
 
     @Test
     fun simpleStaticMock() {
-        mockkStatic("io.mockk.it.StaticMockTestKt")
+        mockkStatic(::op)
 
         every { 5 op 6 } returns 2
 
@@ -24,15 +24,15 @@ class StaticMockTest {
 
     @Test
     fun clearStateStaticMock() {
-        mockkStatic("io.mockk.it.StaticMockTestKt")
+        mockkStatic(::op)
 
         every { 5 op 6 } returns 2
 
         assertEquals(2, 5 op 6)
 
         verify { 5 op 6 }
 
-        mockkStatic("io.mockk.it.StaticMockTestKt")
+        mockkStatic(::op)
 
         verify(exactly = 0) { 5 op 6 }
 

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.
  */