HBX-2735: Create Interface for OverrideRepositoryWrapper

- Add new test case 'org.hibernate.tool.orm.jbt.api.OverrideRepositoryWrapperTest#testAddTableFilter()' - Add new default interface method 'org.hibernate.tool.orm.jbt.api.OverrideRepositoryWrapper#addTableFilter(TableFilter)' Signed-off-by: Koen Aers <koen.aers@gmail.com>
hibernate · Mar 21, 2024 · 23861f8 · 23861f8
1 parent 89720e3
commit 23861f8
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/jbt/src/main/java/org/hibernate/tool/orm/jbt/api/OverrideRepositoryWrapper.java b/jbt/src/main/java/org/hibernate/tool/orm/jbt/api/OverrideRepositoryWrapper.java
@@ -4,6 +4,7 @@
 
 import org.hibernate.tool.api.reveng.RevengStrategy;
 import org.hibernate.tool.internal.reveng.strategy.OverrideRepository;
+import org.hibernate.tool.internal.reveng.strategy.TableFilter;
 import org.hibernate.tool.orm.jbt.wrp.Wrapper;
 
 public interface OverrideRepositoryWrapper extends Wrapper {
@@ -15,5 +16,9 @@ default void addFile(File file) {
 	default RevengStrategy getReverseEngineeringStrategy(RevengStrategy res) {
 		return ((OverrideRepository)getWrappedObject()).getReverseEngineeringStrategy(res);
 	}
+
+	default void addTableFilter(TableFilter tf) {
+		((OverrideRepository)getWrappedObject()).addTableFilter(tf);
+	}
 
 }
diff --git a/jbt/src/test/java/org/hibernate/tool/orm/jbt/api/OverrideRepositoryWrapperTest.java b/jbt/src/test/java/org/hibernate/tool/orm/jbt/api/OverrideRepositoryWrapperTest.java
@@ -16,6 +16,7 @@
 import org.hibernate.tool.internal.reveng.strategy.DefaultStrategy;
 import org.hibernate.tool.internal.reveng.strategy.DelegatingStrategy;
 import org.hibernate.tool.internal.reveng.strategy.OverrideRepository;
+import org.hibernate.tool.internal.reveng.strategy.TableFilter;
 import org.hibernate.tool.orm.jbt.internal.factory.OverrideRepositoryWrapperFactory;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
@@ -79,4 +80,16 @@ public void testGetReverseEngineeringStrategy() throws Exception {
 		assertSame(rev, delegateField.get(delegatingStrategy));
 	}
 
+	@Test
+	public void testAddTableFilter() throws Exception {
+		TableFilter tableFilter = new TableFilter();
+		Field tableFiltersField = OverrideRepository.class.getDeclaredField("tableFilters");
+		tableFiltersField.setAccessible(true);
+		List<?> tableFilters = (List<?>)tableFiltersField.get(wrappedOverrideRepository);
+		assertTrue(tableFilters.isEmpty());
+		overrideRepositoryWrapper.addTableFilter(tableFilter);
+		tableFilters = (List<?>)tableFiltersField.get(wrappedOverrideRepository);
+		assertSame(tableFilter, tableFilters.get(0));		
+	}
+
 }