X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=pcep%2Fimpl%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fprotocol%2Fpcep%2Fimpl%2FServerSessionMock.java;h=7f7a6455b30d502c0e5b5c3e624a1649edfa81b5;hb=c807b0a218942fb394479bb09333257c740707d5;hp=15ed6582d4dfa586f50bde86ee1dd6572d486326;hpb=1a2629053f6839c370cfdddd60c4fa51054ed20d;p=bgpcep.git

diff --git a/pcep/impl/src/test/java/org/opendaylight/protocol/pcep/impl/ServerSessionMock.java b/pcep/impl/src/test/java/org/opendaylight/protocol/pcep/impl/ServerSessionMock.java
index 15ed6582d4..7f7a6455b3 100644
--- a/pcep/impl/src/test/java/org/opendaylight/protocol/pcep/impl/ServerSessionMock.java
+++ b/pcep/impl/src/test/java/org/opendaylight/protocol/pcep/impl/ServerSessionMock.java
@@ -9,32 +9,34 @@ package org.opendaylight.protocol.pcep.impl;
 
 import static org.mockito.Mockito.mock;
 import io.netty.channel.Channel;
-import io.netty.util.HashedWheelTimer;
+import io.netty.util.concurrent.Future;
+import io.netty.util.concurrent.GlobalEventExecutor;
 
 import org.opendaylight.protocol.pcep.PCEPCloseTermination;
 import org.opendaylight.protocol.pcep.PCEPSessionListener;
 import org.opendaylight.protocol.pcep.TerminationReason;
 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.open.message.open.message.OpenBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder;
 
 public class ServerSessionMock extends PCEPSessionImpl {
 
-	private final MockPCE client;
+    private final MockPCE client;
 
-	public ServerSessionMock(final PCEPSessionListener listener, final PCEPSessionListener client) {
-		super(new HashedWheelTimer(), listener, 5, mock(Channel.class), new OpenBuilder().setKeepalive((short) 4).setDeadTimer((short) 9).setSessionId(
-				(short) 2).build(), new OpenBuilder().setKeepalive((short) 4).setDeadTimer((short) 9).setSessionId((short) 2).build());
-		this.client = (MockPCE) client;
-	}
+    public ServerSessionMock(final PCEPSessionListener listener, final PCEPSessionListener client) {
+        super(listener, 5, mock(Channel.class), new OpenBuilder().setKeepalive((short) 4).setDeadTimer((short) 9).setSessionId(
+                (short) 2).build(), new OpenBuilder().setKeepalive((short) 4).setDeadTimer((short) 9).setSessionId((short) 2).build());
+        this.client = (MockPCE) client;
+    }
 
-	@Override
-	public void sendMessage(final Message msg) {
-		this.lastMessageSentAt = System.nanoTime();
-		this.client.onMessage(this, msg);
-	}
+    @Override
+    public Future<Void> sendMessage(final Message msg) {
+        this.lastMessageSentAt = System.nanoTime();
+        this.client.onMessage(this, msg);
+        return GlobalEventExecutor.INSTANCE.newSucceededFuture(null);
+    }
 
-	@Override
-	public void close() {
-		this.client.onSessionTerminated(this, new PCEPCloseTermination(TerminationReason.Unknown));
-	}
+    @Override
+    public void close() {
+        this.client.onSessionTerminated(this, new PCEPCloseTermination(TerminationReason.UNKNOWN));
+    }
 }