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 index b1ef5dd262..bf525f37f0 100644 --- 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 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 index 1da46f52ef..4169f6b011 100644 --- 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)); + } + }