Kotlin spice

atlassian-labs · Dec 2, 2024 · 016ba16 · 016ba16
1 parent b850c10
commit 016ba16
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 34 deletions.
diff --git a/lib/src/test/kotlin/graphql/nadel/archunit/ArchUnitEqualsClassesExactly.kt b/lib/src/test/kotlin/graphql/nadel/archunit/ArchUnitEqualsClassesExactly.kt
@@ -4,15 +4,19 @@ import com.tngtech.archunit.core.domain.JavaClass
 import com.tngtech.archunit.lang.ArchCondition
 import com.tngtech.archunit.lang.ConditionEvents
 import com.tngtech.archunit.lang.SimpleConditionEvent
+import com.tngtech.archunit.lang.syntax.elements.ClassesShouldConjunction
+import com.tngtech.archunit.lang.syntax.elements.GivenClassesConjunction
 import kotlin.reflect.KClass
 
+fun <T : Any> GivenClassesConjunction.equalsExactly(
+    vararg requiredClasses: KClass<out T>,
+): ClassesShouldConjunction {
+    return should(ArchUnitEqualsClassesExactly(requiredClasses.map { it.java }))
+}
+
 class ArchUnitEqualsClassesExactly<T : Any>(
     private val requiredClasses: List<Class<out T>>,
 ) : ArchCondition<JavaClass>("equal exactly the given classes") {
-    constructor(vararg requiredClasses: Class<out T>) : this(requiredClasses.toList())
-
-    constructor(vararg requiredClasses: KClass<out T>) : this(requiredClasses.map { it.java })
-
     private val actualClasses: MutableList<JavaClass> = mutableListOf()
 
     override fun check(item: JavaClass, events: ConditionEvents) {

diff --git a/lib/src/test/kotlin/graphql/nadel/archunit/NadelPseudoSealedTypeKtTest.kt b/lib/src/test/kotlin/graphql/nadel/archunit/NadelPseudoSealedTypeKtTest.kt
@@ -32,11 +32,9 @@ class NadelPseudoSealedTypeKtTest {
             .areAssignableTo(GraphQLFieldsContainer::class.java)
             .and()
             .areNotInterfaces()
-            .should(
-                ArchUnitEqualsClassesExactly(
-                    GraphQLInterfaceType::class,
-                    GraphQLObjectType::class,
-                ),
+            .equalsExactly(
+                GraphQLInterfaceType::class,
+                GraphQLObjectType::class,
             )
             .check(schemaClasses)
     }
@@ -48,17 +46,15 @@ class NadelPseudoSealedTypeKtTest {
             .areAssignableTo(GraphQLNamedType::class.java)
             .and()
             .areNotInterfaces()
-            .should(
-                ArchUnitEqualsClassesExactly(
-                    GraphQLEnumType::class,
-                    GraphQLInputObjectType::class,
-                    GraphQLInterfaceType::class,
-                    GraphQLObjectType::class,
-                    GraphQLScalarType::class,
-                    GraphQLUnionType::class,
-                    // Should almost never be used though
-                    GraphQLTypeReference::class,
-                ),
+            .equalsExactly(
+                GraphQLEnumType::class,
+                GraphQLInputObjectType::class,
+                GraphQLInterfaceType::class,
+                GraphQLObjectType::class,
+                GraphQLScalarType::class,
+                GraphQLUnionType::class,
+                // Should almost never be used though
+                GraphQLTypeReference::class,
             )
             .check(schemaClasses)
     }
@@ -72,14 +68,12 @@ class NadelPseudoSealedTypeKtTest {
             .areAssignableTo(GraphQLUnmodifiedType::class.java)
             .and()
             .areNotInterfaces()
-            .should(
-                ArchUnitEqualsClassesExactly(
-                    GraphQLEnumType::class,
-                    GraphQLInterfaceType::class,
-                    GraphQLObjectType::class,
-                    GraphQLScalarType::class,
-                    GraphQLUnionType::class,
-                ),
+            .equalsExactly(
+                GraphQLEnumType::class,
+                GraphQLInterfaceType::class,
+                GraphQLObjectType::class,
+                GraphQLScalarType::class,
+                GraphQLUnionType::class,
             )
             .check(schemaClasses)
     }
@@ -93,12 +87,10 @@ class NadelPseudoSealedTypeKtTest {
             .areAssignableTo(GraphQLUnmodifiedType::class.java)
             .and()
             .areNotInterfaces()
-            .should(
-                ArchUnitEqualsClassesExactly(
-                    GraphQLEnumType::class,
-                    GraphQLInputObjectType::class,
-                    GraphQLScalarType::class,
-                ),
+            .equalsExactly(
+                GraphQLEnumType::class,
+                GraphQLInputObjectType::class,
+                GraphQLScalarType::class,
             )
             .check(schemaClasses)
     }