X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=pcep%2Fimpl%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fprotocol%2Fpcep%2Fimpl%2FMockPCE.java;h=b3e3f5dfdbef019774552c03fe8cce0d4c84562d;hb=e0a5b2c8695603825cb242906479cd6bc05adb06;hp=ed9ffbe7fbf766b0a200712c2a7aef53d965b3c5;hpb=3dd28cfa15e490adf3ba1d46ee1ab193428da57b;p=bgpcep.git

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 ed9ffbe7fb..b3e3f5dfdb 100644
--- 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);
 	}