Add class test and fix issues

1 files changed, 21 insertions, 0 deletions

diff --git a/android/test/arch/src/test/kotlin/net/mullvad/mullvadvpn/test/arch/classes/ClassTests.kt b/android/test/arch/src/test/kotlin/net/mullvad/mullvadvpn/test/arch/classes/ClassTests.kt
new file mode 100644
index 0000000000..cc737250f6
--- /dev/null
+++ b/android/test/arch/src/test/kotlin/net/mullvad/mullvadvpn/test/arch/classes/ClassTests.kt
@@ -0,0 +1,21 @@
+package net.mullvad.mullvadvpn.test.arch.classes
+
+import com.lemonappdev.konsist.api.verify.assert
+import net.mullvad.mullvadvpn.test.arch.extensions.projectScope
+import org.junit.Test
+
+class ClassTests {
+    @Test
+    fun `companion object is last declaration in the class`() {
+        projectScope().classes(includeNested = true).assert {
+            val companionObject =
+                it.objects(includeNested = false).lastOrNull { obj -> obj.hasCompanionModifier }
+            if (companionObject != null) {
+                it.declarations(includeNested = false, includeLocal = false).last() ==
+                    companionObject
+            } else {
+                true
+            }
+        }
+    }
+}