Refactor GeneratedTOBuilderImpl
[mdsal.git] / binding / mdsal-binding-generator-util / src / test / java / org / opendaylight / mdsal / binding / model / util / generated / type / builder / GeneratedTypeBuilderTest.java
index e12819c5fec9d58fbc7a6c5fe1f9999bce8e6b11..0bd589d75bf24f1c0cf6f26a6716430b67ec744f 100644 (file)
@@ -264,7 +264,7 @@ public class GeneratedTypeBuilderTest {
 
         GeneratedTOBuilder enclosingTransferObject = generatedTypeBuilder.addEnclosingTransferObject("myTOName");
         GeneratedTOBuilder enclosingTransferObject2 = generatedTypeBuilder.addEnclosingTransferObject("myTOName2");
-        assertEquals(generatedTypeBuilder, generatedTypeBuilder.addEnclosingTransferObject(new GeneratedTOBuilderImpl(
+        assertEquals(generatedTypeBuilder, generatedTypeBuilder.addEnclosingTransferObject(new CodegenGeneratedTOBuilder(
                 generatedTypeBuilder.getFullyQualifiedName(), "myTOName3")));
 
         assertNotNull(enclosingTransferObject);
@@ -277,9 +277,9 @@ public class GeneratedTypeBuilderTest {
 
         assertTrue(enclosedTypes.contains(enclosingTransferObject.toInstance()));
         assertTrue(enclosedTypes.contains(enclosingTransferObject2.toInstance()));
-        assertTrue(enclosedTypes.contains(new GeneratedTOBuilderImpl(generatedTypeBuilder.getFullyQualifiedName(),
+        assertTrue(enclosedTypes.contains(new CodegenGeneratedTOBuilder(generatedTypeBuilder.getFullyQualifiedName(),
                 "myTOName3").toInstance()));
-        assertFalse(enclosedTypes.contains(new GeneratedTOBuilderImpl(generatedTypeBuilder.getFullyQualifiedName(),
+        assertFalse(enclosedTypes.contains(new CodegenGeneratedTOBuilder(generatedTypeBuilder.getFullyQualifiedName(),
                 "myTOName4").toInstance()));
 
     }