Bump xtendbeans version for AssertBeans to support YANG union types

[mdsal.git] / binding / mdsal-binding-test-utils / src / main / java / org / opendaylight / mdsal / binding / testutils / XtendYangBeanGenerator.java

diff --git a/binding/mdsal-binding-test-utils/src/main/java/org/opendaylight/mdsal/binding/testutils/XtendYangBeanGenerator.java b/binding/mdsal-binding-test-utils/src/main/java/org/opendaylight/mdsal/binding/testutils/XtendYangBeanGenerator.java
index 2f7f7aab67522d7a65848935d34aad796e87190c..fb12ea3e7da50d7ebaa59fcd3180dd13572429f0 100644 (file)
--- a/binding/mdsal-binding-test-utils/src/main/java/org/opendaylight/mdsal/binding/testutils/XtendYangBeanGenerator.java
+++ b/binding/mdsal-binding-test-utils/src/main/java/org/opendaylight/mdsal/binding/testutils/XtendYangBeanGenerator.java
@@ -76,9 +76,13 @@ class XtendYangBeanGenerator extends XtendBeanGenerator {
     protected CharSequence getNewBeanExpression(Object bean) {
         if (bean instanceof DataContainer) {
             DataContainer dataContainerBean = (DataContainer) bean;
-            Class<?> builderClass = getBuilderClassByAppendingBuilderToClassName(
+            Optional<Class<?>> optBuilderClass = getOptionalBuilderClassByAppendingBuilderToClassName(
                     dataContainerBean.getImplementedInterface());
-            return super.getNewBeanExpression(dataContainerBean, builderClass);
+            if (optBuilderClass.isPresent()) {
+                return super.getNewBeanExpression(dataContainerBean, optBuilderClass.get());
+            } else {
+                throw new IllegalArgumentException("DataContainer has no *Builder class: " + bean.getClass());
+            }
         } else {
             return super.getNewBeanExpression(bean);
         }