Improve test coverage for coroutines (#526)

mockito · Sep 10, 2024 · 176fb3a · 176fb3a
1 parent 391bb3f
commit 176fb3a
Show file tree

Hide file tree

Showing 4 changed files with 54 additions and 0 deletions.
diff --git a/tests/build.gradle b/tests/build.gradle
@@ -19,6 +19,7 @@ dependencies {
 
     testImplementation 'junit:junit:4.13.2'
     testImplementation "com.nhaarman:expect.kt:1.0.1"
+    testImplementation "org.jetbrains.kotlinx:kotlinx-coroutines-test:1.9.0-RC"
 }
 
 tasks.withType(KotlinCompile).configureEach {

diff --git a/tests/src/test/kotlin/test/Classes.kt b/tests/src/test/kotlin/test/Classes.kt
@@ -45,6 +45,8 @@ class Closed
 interface Methods {
 
     fun closed(c: Closed)
+    fun classClosed(c: Class<Closed>)
+    suspend fun coroutinesClosed(c: Closed)
     fun closedArray(a: Array<Closed>)
     fun closedNullableArray(a: Array<Closed?>)
     fun closedCollection(c: Collection<Closed>)
@@ -77,6 +79,8 @@ interface Methods {
     fun nullableStringResult(): String?
     fun builderMethod(): Methods
     fun varargBooleanResult(vararg values: String): Boolean
+    suspend fun coroutinesClosedBooleanResult(c: Closed): Boolean
+    suspend fun coroutinesClassClosedBooleanResult(c: Class<Closed>): Boolean
     fun stringArray(a: Array<String>)
     fun argAndVararg(s: String, vararg a: String)
 

diff --git a/tests/src/test/kotlin/test/MatchersTest.kt b/tests/src/test/kotlin/test/MatchersTest.kt
@@ -2,6 +2,7 @@ package test
 
 import com.nhaarman.expect.expect
 import com.nhaarman.expect.expectErrorWithMessage
+import kotlinx.coroutines.test.runTest
 import org.junit.Test
 import org.mockito.ArgumentMatcher
 import org.mockito.invocation.InvocationOnMock
@@ -35,6 +36,24 @@ class MatchersTest : TestBase() {
         }
     }
 
+    @Test
+    fun anyClassClosedClass() {
+        mock<Methods>().apply {
+            classClosed(Closed::class.java)
+            verify(this).classClosed(any())
+        }
+    }
+
+    @Test
+    fun anyCoroutinesClosedClass() {
+        mock<Methods>().apply {
+            runTest {
+                coroutinesClosed(Closed())
+                verify(this@apply).coroutinesClosed(any())
+            }
+        }
+    }
+
     @Test
     fun anyIntArray() {
         mock<Methods>().apply {

diff --git a/tests/src/test/kotlin/test/MockingTest.kt b/tests/src/test/kotlin/test/MockingTest.kt
@@ -3,12 +3,14 @@ package test
 import com.nhaarman.expect.expect
 import com.nhaarman.expect.expectErrorWithMessage
 import com.nhaarman.expect.fail
+import kotlinx.coroutines.test.runTest
 import org.mockito.kotlin.UseConstructor.Companion.parameterless
 import org.mockito.kotlin.UseConstructor.Companion.withArguments
 import org.mockito.kotlin.doReturn
 import org.mockito.kotlin.mock
 import org.mockito.kotlin.verify
 import org.mockito.kotlin.whenever
+import org.mockito.kotlin.any
 import org.junit.Test
 import org.mockito.Mockito
 import org.mockito.exceptions.verification.WantedButNotInvoked
@@ -274,6 +276,34 @@ class MockingTest : TestBase() {
         expect(result).toBe("foo")
     }
 
+    @Test
+    fun mockCoroutines_withClosedBooleanReturn_name() = runTest {
+        /* Given */
+        val mock = mock<Methods>(name = "myName") {
+            onBlocking { coroutinesClosedBooleanResult(any()) } doReturn true
+        }
+
+        /* When */
+        val result = mock.coroutinesClosedBooleanResult(Closed())
+
+        /* Then */
+        expect(result).toBe(true)
+    }
+
+    @Test
+    fun mockCoroutines_withClassClosedBooleanReturn_name() = runTest {
+        /* Given */
+        val mock = mock<Methods>(name = "myName") {
+            onBlocking { coroutinesClassClosedBooleanResult(any()) } doReturn true
+        }
+
+        /* When */
+        val result = mock.coroutinesClassClosedBooleanResult(Closed::class.java)
+
+        /* Then */
+        expect(result).toBe(true)
+    }
+
     @Test
     fun mockStubbing_withSettingsAPI_defaultAnswer() {
         /* Given */