add find system by type method

Adven27 · Oct 1, 2021 · fbb5e8f · fbb5e8f
1 parent 9f89f3b
commit fbb5e8f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/env-core/src/main/java/io/github/adven27/env/core/Environment.kt b/env-core/src/main/java/io/github/adven27/env/core/Environment.kt
@@ -103,6 +103,7 @@ open class Environment @JvmOverloads constructor(
 
     @Suppress("UNCHECKED_CAST")
     fun <T : ExternalSystem> find(name: String): T = (systems[name] ?: error("System $name not found")) as T
+    inline fun <reified T : ExternalSystem> find() = systems.values.filterIsInstance<T>().single()
 
     companion object : KLogging() {
         private fun start(

diff --git a/...bedded/src/test/kotlin/io/github/adven27/env/mq/kafka/embedded/EmbeddedKafkaSystemTest.kt b/...bedded/src/test/kotlin/io/github/adven27/env/mq/kafka/embedded/EmbeddedKafkaSystemTest.kt
@@ -25,5 +25,5 @@ class EmbeddedKafkaSystemTest {
 }
 
 class SomeEnvironment : Environment("EMBEDDED_KAFKA" to EmbeddedKafkaSystem("some-topic")) {
-    fun kafka() = find<EmbeddedKafkaSystem>("EMBEDDED_KAFKA")
+    fun kafka() = find<EmbeddedKafkaSystem>()
 }
diff --git a/example/src/test/kotlin/EnvTest.kt b/example/src/test/kotlin/EnvTest.kt
@@ -61,6 +61,6 @@ class SomeEnvironment : Environment(
     },
     "WIREMOCK" to WiremockSystem()
 ) {
-    fun rabbit() = find<RabbitContainerSystem>("RABBIT")
-    fun postgres() = systems["POSTGRES"] as PostgreSqlContainerSystem
+    fun rabbit() = find<RabbitContainerSystem>()
+    fun postgres() = find<PostgreSqlContainerSystem>()
 }