refactor: Add some more testing and polishment around config customiz…

…ation.
neo4j · Aug 27, 2024 · a541821 · a541821
1 parent fe86876
commit a541821
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/deployment/src/test/java/org/neo4j/ogm/quarkus/test/ConfigurationTest.java b/deployment/src/test/java/org/neo4j/ogm/quarkus/test/ConfigurationTest.java
@@ -15,6 +15,7 @@
  */
 package org.neo4j.ogm.quarkus.test;
 
+import static org.junit.jupiter.api.Assertions.assertArrayEquals;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertNull;
@@ -65,5 +66,6 @@ void configShouldBeApplied() {
 		assertTrue(configuration.getUseNativeTypes());
 		assertTrue(configuration.getUseStrictQuerying());
 		assertEquals("aDatabase", configuration.getDatabase());
+		assertArrayEquals(new String[] {"org.neo4j.ogm.quarkus.test.domain"}, configuration.getBasePackages());
 	}
 }
diff --git a/runtime/src/main/java/org/neo4j/ogm/quarkus/runtime/Neo4jOgmConfigCustomizer.java b/runtime/src/main/java/org/neo4j/ogm/quarkus/runtime/Neo4jOgmConfigCustomizer.java
@@ -15,6 +15,7 @@
  */
 package org.neo4j.ogm.quarkus.runtime;
 
+import io.quarkus.runtime.annotations.StaticInitSafe;
 import io.smallrye.config.SmallRyeConfigBuilder;
 import io.smallrye.config.SmallRyeConfigBuilderCustomizer;
 
@@ -26,6 +27,7 @@
  * @see Neo4jOgmBuiltTimeProperties
  * @see Neo4jOgmProperties
  */
+@StaticInitSafe
 public final class Neo4jOgmConfigCustomizer implements SmallRyeConfigBuilderCustomizer {
 
 	@Override

diff --git a/runtime/src/main/java/org/neo4j/ogm/quarkus/runtime/Neo4jOgmRecorder.java b/runtime/src/main/java/org/neo4j/ogm/quarkus/runtime/Neo4jOgmRecorder.java
@@ -36,6 +36,7 @@ public class Neo4jOgmRecorder {
 	 *
 	 * @param driverRuntimeValue The required java driver
 	 * @param shutdownContext    Needed to close it
+	 * @param neo4jConfiguration The applicable configuration
 	 * @param ogmProperties      Runtime properties that can be configured
 	 * @param allPackages        the list of packages already discovered
 	 * @return A session factory