HBX-2390: Create a JBoss Tools adaptation layer in Hibernate Tools

- Add new test case 'org.hibernate.tool.orm.jbt.wrp.ExporterWrapperFactoryTest#testSetCustomProperties()' - Add new interface method 'org.hibernate.tool.orm.jbt.wrp.ExporterWrapperFactory#setCustomProperties(Properties)' Signed-off-by: Koen Aers <[email protected]>
hibernate · Jun 27, 2023 · 3262906 · 3262906
1 parent 312176e
commit 3262906
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/jbt/src/main/java/org/hibernate/tool/orm/jbt/wrp/ExporterWrapperFactory.java b/jbt/src/main/java/org/hibernate/tool/orm/jbt/wrp/ExporterWrapperFactory.java
@@ -102,6 +102,11 @@ default QueryExporterWrapper getQueryExporter() {
 				return null;
 			}
 		}
+		default void setCustomProperties(Properties properties) {
+			if (getWrappedObject() instanceof CfgExporter) {
+				((CfgExporter)getWrappedObject()).setCustomProperties(properties);
+			}
+		}
 	}
 
 	static class ExporterWrapperImpl implements ExporterWrapper {

diff --git a/jbt/src/test/java/org/hibernate/tool/orm/jbt/wrp/ExporterWrapperFactoryTest.java b/jbt/src/test/java/org/hibernate/tool/orm/jbt/wrp/ExporterWrapperFactoryTest.java
@@ -171,10 +171,28 @@ public void testGetQueryExporter() {
 		assertSame(exporterTarget, queryExporterTarget);
 	}
 
+	@Test
+	public void testSetCustomProperties() {
+		Properties properties = new Properties();
+		// 'setCustomProperties()' should not be called on other exporters than CfgExporter
+		TestExporter wrappedTestExporter = (TestExporter)exporterWrapper.getWrappedObject();
+		assertNull(wrappedTestExporter.props);
+		exporterWrapper.setCustomProperties(null);
+		assertNull(wrappedTestExporter.props);
+		// try now with CfgExporter 
+		exporterWrapper = ExporterWrapperFactory.create(CfgExporter.class.getName());
+		CfgExporter wrappedCfgExporter = (CfgExporter)exporterWrapper.getWrappedObject();
+		assertNotSame(properties, wrappedCfgExporter.getCustomProperties());
+		exporterWrapper.setCustomProperties(properties);
+		assertSame(properties, wrappedCfgExporter.getCustomProperties());
+	}
+
 	public static class TestExporter extends AbstractExporter {
 		private boolean started = false;
+		private Properties props = null;
 		@Override protected void doStart() {}
 		@Override public void start() { started = true; }		
+		void setCustomProperties(Properties p) { props = p; }
 	}
 
 }