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 ca9f4a3648..b1ef5dd262 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 @@ -1,6 +1,7 @@ package org.hibernate.tool.orm.jbt.api; import java.util.Iterator; +import java.util.List; import org.hibernate.mapping.Column; import org.hibernate.mapping.ForeignKey; @@ -14,6 +15,8 @@ public interface ForeignKeyWrapper extends Wrapper { default Iterator columnIterator() { return ((ForeignKey)getWrappedObject()).getColumns().iterator(); } default boolean isReferenceToPrimaryKey() { return ((ForeignKey)getWrappedObject()).isReferenceToPrimaryKey(); } + + default List getReferencedColumns() { return ((ForeignKey)getWrappedObject()).getReferencedColumns(); } } 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 05691e0cb2..1da46f52ef 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 @@ -1,5 +1,6 @@ package org.hibernate.tool.orm.jbt.api; +import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertFalse; import static org.junit.jupiter.api.Assertions.assertNotNull; import static org.junit.jupiter.api.Assertions.assertSame; @@ -7,6 +8,7 @@ import java.util.ArrayList; import java.util.Iterator; +import java.util.List; import org.hibernate.mapping.Column; import org.hibernate.mapping.ForeignKey; @@ -62,4 +64,17 @@ public void testIsReferenceToPrimaryKey() { assertFalse(foreignKeyWrapper.isReferenceToPrimaryKey()); } + @Test + public void testGetReferencedColumns() { + List list = foreignKeyWrapper.getReferencedColumns(); + assertTrue(list.isEmpty()); + Column column = new Column(); + ArrayList columns = new ArrayList(); + columns.add(column); + wrappedForeignKey.addReferencedColumns(columns); + list = foreignKeyWrapper.getReferencedColumns(); + assertEquals(1, list.size()); + assertSame(column, list.get(0)); + } + }