BUG-47 : switched subobjects to generated source code.

[bgpcep.git] / pcep / impl / src / test / java / org / opendaylight / protocol / pcep / impl / MockPCE.java

diff --git a/pcep/impl/src/test/java/org/opendaylight/protocol/pcep/impl/MockPCE.java b/pcep/impl/src/test/java/org/opendaylight/protocol/pcep/impl/MockPCE.java
index ed9ffbe7fbf766b0a200712c2a7aef53d965b3c5..b3e3f5dfdbef019774552c03fe8cce0d4c84562d 100644 (file)
--- a/pcep/impl/src/test/java/org/opendaylight/protocol/pcep/impl/MockPCE.java
+++ b/pcep/impl/src/test/java/org/opendaylight/protocol/pcep/impl/MockPCE.java
@@ -11,19 +11,17 @@ import java.util.ArrayList;
 import java.util.List;
 
 import org.opendaylight.protocol.pcep.PCEPErrors;
-import org.opendaylight.protocol.pcep.PCEPMessage;
 import org.opendaylight.protocol.pcep.PCEPSession;
 import org.opendaylight.protocol.pcep.PCEPSessionListener;
 import org.opendaylight.protocol.pcep.PCEPTerminationReason;
-import org.opendaylight.protocol.pcep.message.PCEPErrorMessage;
-import org.opendaylight.protocol.pcep.object.PCEPErrorObject;
-import org.opendaylight.protocol.pcep.object.PCEPOpenObject;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.OpenObject;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 public class MockPCE implements PCEPSessionListener {
 
-       private final List<PCEPMessage> listMsg = new ArrayList<PCEPMessage>();
+       private final List<Message> listMsg = new ArrayList<Message>();
 
        private PCEPSessionImpl session = null;
 
@@ -36,18 +34,15 @@ public class MockPCE implements PCEPSessionListener {
        public MockPCE() {
        }
 
-       public void sendMessage(final PCEPMessage msg) {
+       public void sendMessage(final Message msg) {
                this.session.handleMessage(msg);
        }
 
-       public void sendErrorMessage(final PCEPErrors value, final PCEPOpenObject open) {
-               final PCEPErrorObject error = new PCEPErrorObject(value);
-               final List<PCEPErrorObject> errors = new ArrayList<PCEPErrorObject>();
-               errors.add(error);
-               this.sendMessage(new PCEPErrorMessage(open, errors, null));
+       public void sendErrorMessage(final PCEPErrors value, final OpenObject open) {
+               this.sendMessage(Util.createErrorMessage(value, open));
        }
 
-       public List<PCEPMessage> getListMsg() {
+       public List<Message> getListMsg() {
                return this.listMsg;
        }
 
@@ -56,7 +51,7 @@ public class MockPCE implements PCEPSessionListener {
        }
 
        @Override
-       public void onMessage(final PCEPSession session, final PCEPMessage message) {
+       public void onMessage(final PCEPSession session, final Message message) {
                this.listMsg.add(message);
                logger.debug("Message received: {}", message);
        }