HBX-2747: Create Interface for ForeignKeyWrapper

- Add new test case 'org.hibernate.tool.orm.jbt.api.ForeignKeyWrapperTest#testContainsColumn()' - Add new default interface method 'org.hibernate.tool.orm.jbt.api.ForeignKeyWrapper#containsColumn(Column)' Signed-off-by: Koen Aers <koen.aers@gmail.com>
hibernate · Mar 13, 2024 · 10431fc · 10431fc
1 parent 589a5bd
commit 10431fc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/jbt/src/main/java/org/hibernate/tool/orm/jbt/api/ForeignKeyWrapper.java b/jbt/src/main/java/org/hibernate/tool/orm/jbt/api/ForeignKeyWrapper.java
@@ -18,5 +18,6 @@ public interface ForeignKeyWrapper extends Wrapper {
 
 	default List<Column> getReferencedColumns() { return ((ForeignKey)getWrappedObject()).getReferencedColumns(); }
 
+	default boolean containsColumn(Column column) { return ((ForeignKey)getWrappedObject()).containsColumn(column); }
 
 }
diff --git a/jbt/src/test/java/org/hibernate/tool/orm/jbt/api/ForeignKeyWrapperTest.java b/jbt/src/test/java/org/hibernate/tool/orm/jbt/api/ForeignKeyWrapperTest.java
@@ -77,4 +77,12 @@ public void testGetReferencedColumns() {
 		assertSame(column, list.get(0));
 	}
 
+	@Test
+	public void testContainsColumn() throws Exception {
+		Column column = new Column("foo");
+		assertFalse(foreignKeyWrapper.containsColumn(column));
+		wrappedForeignKey.addColumn(column);
+		assertTrue(foreignKeyWrapper.containsColumn(column));
+	}
+
 }
Original file line number	Diff line number	Diff line change
Expand Up		@@ -18,5 +18,6 @@ public interface ForeignKeyWrapper extends Wrapper {

		default List<Column> getReferencedColumns() { return ((ForeignKey)getWrappedObject()).getReferencedColumns(); }

		default boolean containsColumn(Column column) { return ((ForeignKey)getWrappedObject()).containsColumn(column); }

		}