Skip to content

Commit 28337ca

Browse files
committed
HBX-2789: Review WrapperFactory class and create only new Wrapper intances
- Modify method 'org.hibernate.tool.orm.jbt.internal.factory.RevengStrategyWrapperFactory#createRevengStrategyWrapper(Object..)' to use a Wrapper argument - Adapt test 'org.hibernate.tool.orm.jbt.api.factory.WrapperFactoryTest#testCreateRevengStrategyWrapper()' Signed-off-by: Koen Aers <[email protected]>
1 parent f5a11b3 commit 28337ca

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

jbt/src/main/java/org/hibernate/tool/orm/jbt/internal/factory/RevengStrategyWrapperFactory.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import org.hibernate.tool.internal.reveng.strategy.DefaultStrategy;
88
import org.hibernate.tool.orm.jbt.api.wrp.RevengSettingsWrapper;
99
import org.hibernate.tool.orm.jbt.api.wrp.RevengStrategyWrapper;
10+
import org.hibernate.tool.orm.jbt.api.wrp.Wrapper;
1011
import org.hibernate.tool.orm.jbt.internal.wrp.AbstractWrapper;
1112
import org.hibernate.tool.orm.jbt.util.ReflectUtil;
1213

@@ -17,7 +18,7 @@ public static RevengStrategyWrapper createRevengStrategyWrapper(Object...objects
1718
if (objects.length == 0) {
1819
wrappedRevengStrategy = createDefaultStrategy();
1920
} else if (objects.length == 2) {
20-
wrappedRevengStrategy = createDelegatingStrategy((String)objects[0], (RevengStrategy)objects[1]);
21+
wrappedRevengStrategy = createDelegatingStrategy((String)objects[0], (RevengStrategy)((Wrapper)objects[1]).getWrappedObject());
2122
} else {
2223
throw new RuntimeException("RevengStrategyWrapperFactory#create has either 0 or 2 arguments");
2324
}

jbt/src/test/java/org/hibernate/tool/orm/jbt/api/factory/WrapperFactoryTest.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -135,17 +135,17 @@ public void testCreateRevengStrategyWrapper() throws Exception {
135135
assertNotNull(reverseEngineeringStrategyWrapper);
136136
assertTrue(reverseEngineeringStrategyWrapper instanceof Wrapper);
137137
assertTrue(((Wrapper)reverseEngineeringStrategyWrapper).getWrappedObject() instanceof DefaultStrategy);
138-
reverseEngineeringStrategyWrapper = null;
139-
assertNull(reverseEngineeringStrategyWrapper);
140-
RevengStrategy delegate = new TestRevengStrategy();
138+
RevengStrategyWrapper delegate = (RevengStrategyWrapper)reverseEngineeringStrategyWrapper;
141139
reverseEngineeringStrategyWrapper = WrapperFactory
142140
.createRevengStrategyWrapper(
143141
TestDelegatingStrategy.class.getName(),
144142
delegate);
145143
assertNotNull(reverseEngineeringStrategyWrapper);
146144
assertTrue(reverseEngineeringStrategyWrapper instanceof Wrapper);
147145
assertTrue(((Wrapper)reverseEngineeringStrategyWrapper).getWrappedObject() instanceof TestDelegatingStrategy);
148-
assertSame(delegateField.get(((Wrapper)reverseEngineeringStrategyWrapper).getWrappedObject()), delegate);
146+
assertSame(
147+
delegateField.get(((Wrapper)reverseEngineeringStrategyWrapper).getWrappedObject()),
148+
delegate.getWrappedObject());
149149
}
150150

151151
@Test

0 commit comments

Comments
 (0)