Add konsist test to ensure all view model test uses TestCoroutineRule

1 files changed, 18 insertions, 0 deletions

diff --git a/android/test/arch/src/test/kotlin/net/mullvad/mullvadvpn/test/arch/classes/ClassTest.kt b/android/test/arch/src/test/kotlin/net/mullvad/mullvadvpn/test/arch/classes/ClassTest.kt
index c7e3cc54ed..4e8de563fd 100644
--- a/android/test/arch/src/test/kotlin/net/mullvad/mullvadvpn/test/arch/classes/ClassTest.kt
+++ b/android/test/arch/src/test/kotlin/net/mullvad/mullvadvpn/test/arch/classes/ClassTest.kt
@@ -27,4 +27,22 @@ class ClassTest {
                 it.packagee?.name?.startsWith("net.mullvad.mullvadvpn") ?: false
             }
             .assertTrue { it.hasNameEndingWith("Test") }
+
+    @Test
+    fun `ensure that all view model test classes are annotated with TestCoroutineRule`() =
+        Konsist.scopeFromTest()
+            .classes(includeNested = false)
+            .filter {
+                // Only include classes that are view model tests
+                // We want to ignore the class "ViewModelTest" which contains konsist tests
+                it.name.contains(".+ViewModelTest".toRegex())
+            }
+            .assertTrue {
+                it.hasAnnotation { annotation ->
+                    annotation.name == "ExtendWith" &&
+                        annotation.arguments.any { argument ->
+                            argument.value == "TestCoroutineRule::class"
+                        }
+                }
+            }
 }