add CoSecModule to AutoConfiguration

Ahoo-Wang · Dec 1, 2022 · 2562401 · 2562401
1 parent e6b74af
commit 2562401
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/...-boot-starter/src/main/kotlin/me/ahoo/cosec/spring/boot/starter/CoSecAutoConfiguration.kt b/...-boot-starter/src/main/kotlin/me/ahoo/cosec/spring/boot/starter/CoSecAutoConfiguration.kt
@@ -12,8 +12,11 @@
  */
 package me.ahoo.cosec.spring.boot.starter
 
+import me.ahoo.cosec.policy.serialization.CoSecModule
 import org.springframework.boot.autoconfigure.AutoConfiguration
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean
 import org.springframework.boot.context.properties.EnableConfigurationProperties
+import org.springframework.context.annotation.Bean
 
 /**
  * CoSec AutoConfiguration .
@@ -23,4 +26,10 @@ import org.springframework.boot.context.properties.EnableConfigurationProperties
 @AutoConfiguration
 @ConditionalOnCoSecEnabled
 @EnableConfigurationProperties(CoSecProperties::class)
-class CoSecAutoConfiguration
+class CoSecAutoConfiguration {
+    @Bean
+    @ConditionalOnMissingBean
+    fun coSecModule(): CoSecModule {
+        return CoSecModule()
+    }
+}
diff --git a/...t-starter/src/test/kotlin/me/ahoo/cosec/spring/boot/starter/CoSecAutoConfigurationTest.kt b/...t-starter/src/test/kotlin/me/ahoo/cosec/spring/boot/starter/CoSecAutoConfigurationTest.kt
@@ -13,6 +13,7 @@
 
 package me.ahoo.cosec.spring.boot.starter
 
+import me.ahoo.cosec.policy.serialization.CoSecModule
 import org.assertj.core.api.AssertionsForInterfaceTypes.assertThat
 import org.junit.jupiter.api.Test
 import org.springframework.boot.test.context.assertj.AssertableApplicationContext
@@ -29,6 +30,7 @@ internal class CoSecAutoConfigurationTest {
                 assertThat(context)
                     .hasSingleBean(CoSecProperties::class.java)
                     .hasSingleBean(CoSecAutoConfiguration::class.java)
+                    .hasSingleBean(CoSecModule::class.java)
             }
     }