1 files changed, 224 insertions, 0 deletions
diff --git a/container-search/src/test/java/com/yahoo/fs4/mplex/BackendTestCase.java b/container-search/src/test/java/com/yahoo/fs4/mplex/BackendTestCase.java
new file mode 100644
index 00000000000..8b9f7a345b1
--- /dev/null
+++ b/container-search/src/test/java/com/yahoo/fs4/mplex/BackendTestCase.java
@@ -0,0 +1,224 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.fs4.mplex;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+
+import java.io.IOException;
+import java.net.InetSocketAddress;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.nio.ByteBuffer;
+import java.util.logging.Logger;
+
+import com.yahoo.container.search.Fs4Config;
+import com.yahoo.prelude.fastsearch.FS4ResourcePool;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import com.yahoo.fs4.BasicPacket;
+import com.yahoo.fs4.ChannelTimeoutException;
+import com.yahoo.fs4.PacketListener;
+import com.yahoo.fs4.PingPacket;
+import com.yahoo.fs4.QueryPacket;
+import com.yahoo.fs4.mplex.Backend.BackendStatistics;
+import com.yahoo.search.Query;
+
+/**
+ * Test networking code for talking to dispatch.
+ *
+ * @author <a href="mailto:steinar@yahoo-inc.com">Steinar Knutsen</a>
+ */
+public class BackendTestCase {
+
+    public static class MockDispatch implements Runnable {
+        public ServerSocket socket;
+        public volatile Socket connection;
+        volatile int channelId;
+
+        public byte[] packetData = new byte[] { 0, 0, 0, 76, 0, 0, 0, 202 - 256, 0, 0,
+                0, 1, 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0, 0, 0, 0, 0, 5, 0x40,
+                0x39, 0, 0, 0, 0, 0, 0, 0, 0, 0, 111, 1, 1, 1, 1, 1, 1, 1, 1,
+                1, 1, 1, 1, 0x40, 0x37, 0, 0, 0, 0, 0, 0, 2, 2, 2, 2, 2, 2, 2,
+                2, 2, 2, 2, 2, 0x40, 0x35, 0, 0, 0, 0, 0, 0 };
+
+        public MockDispatch(final ServerSocket socket) {
+            this.socket = socket;
+        }
+
+        @Override
+        public void run() {
+            try {
+                connection = socket.accept();
+            } catch (final IOException e) {
+                e.printStackTrace();
+                return;
+            }
+            requestRespond();
+        }
+
+        void requestRespond() {
+            final byte[] length = new byte[4];
+            try {
+                connection.getInputStream().read(length);
+            } catch (final IOException e) {
+                e.printStackTrace();
+                return;
+            }
+            final int actual = ByteBuffer.wrap(length).getInt();
+
+            int read = 0;
+            int i = 0;
+            while (read != -1 && i < actual) {
+                try {
+                    read = connection.getInputStream().read();
+                    ++i;
+                } catch (final IOException e) {
+                    e.printStackTrace();
+                    return;
+                }
+            }
+            final ByteBuffer reply = ByteBuffer.wrap(packetData);
+            if (channelId != -1) {
+                reply.putInt(8, channelId);
+            }
+            try {
+                connection.getOutputStream().write(packetData);
+            } catch (final IOException e) {
+                e.printStackTrace();
+            }
+        }
+
+    }
+
+    public static class MockPacketListener implements PacketListener {
+
+        @Override
+        public void packetSent(final FS4Channel channel,
+                final BasicPacket packet, final ByteBuffer serializedForm) {
+
+        }
+
+        @Override
+        public void packetReceived(final FS4Channel channel,
+                final BasicPacket packet, final ByteBuffer serializedForm) {
+
+        }
+
+    }
+
+    public static class MockServer {
+        public InetSocketAddress host;
+        public Thread worker;
+        public MockDispatch dispatch;
+
+        public MockServer() throws IOException {
+            final ServerSocket socket = new ServerSocket(0);
+            host = (InetSocketAddress) socket.getLocalSocketAddress();
+            dispatch = new MockDispatch(socket);
+            worker = new Thread(dispatch);
+            worker.start();
+        }
+
+    }
+
+    Backend backend;
+    MockServer server;
+    private Logger logger;
+    private boolean initUseParent;
+    FS4ResourcePool listeners;
+
+    public static final byte[] PONG = new byte[] { 0, 0, 0, 28, 0, 0, 0, 210 - 256,
+            0, 0, 0, 42, 0, 0, 0, 127, 0, 0, 0, 2, 0, 0, 0, 2, 0, 0, 0, 1, 0,
+            0, 0, 1 };
+
+    @Before
+    public void setUp() throws Exception {
+        logger = Logger.getLogger(Backend.class.getName());
+        initUseParent = logger.getUseParentHandlers();
+        logger.setUseParentHandlers(false);
+        listeners = new FS4ResourcePool(new Fs4Config());
+
+        server = new MockServer();
+        backend = listeners.getBackend(server.host.getHostString(), server.host.getPort());
+    }
+
+    @After
+    public void tearDown() throws Exception {
+        listeners.deconstruct();
+        if (server.dispatch.socket != null) server.dispatch.socket.close();
+        if (server.dispatch.connection !=null) server.dispatch.connection.close();
+        if (server.worker!=null) server.worker.join();
+        if (logger !=null) logger.setUseParentHandlers(initUseParent);
+    }
+
+    @Test
+    public final void testBackend() throws IOException, InvalidChannelException {
+        try {
+            final FS4Channel channel = backend.openChannel();
+            final Query q = new Query("/?query=a");
+            BasicPacket[] b = null;
+            final int channelId = channel.getChannelId();
+            server.dispatch.channelId = channelId;
+
+            assertTrue(backend.sendPacket(QueryPacket.create(q), channelId));
+            try {
+                b = channel.receivePackets(1000, 1);
+            } catch (final ChannelTimeoutException e) {
+                fail("Could not get packets from simulated backend.");
+            }
+            assertEquals(1, b.length);
+            assertEquals(202, b[0].getCode());
+            channel.close();
+        }
+        catch (java.net.UnknownHostException e) {
+            // We are on vpn, or have no network
+        }
+    }
+
+    @Test
+    public final void testPinging() throws IOException, InvalidChannelException {
+        try {
+            final FS4Channel channel = backend.openPingChannel();
+            BasicPacket[] b = null;
+            server.dispatch.channelId = -1;
+            server.dispatch.packetData = PONG;
+
+            assertTrue(channel.sendPacket(new PingPacket()));
+            try {
+                b = channel.receivePackets(1000, 1);
+            } catch (final ChannelTimeoutException e) {
+                fail("Could not get packets from simulated backend.");
+            }
+            assertEquals(1, b.length);
+            assertEquals(210, b[0].getCode());
+            channel.close();
+        }
+        catch (java.net.UnknownHostException e) {
+            // We are on vpn, or have no network
+        }
+    }
+
+    @Test
+    public final void requireStatistics() throws IOException, InvalidChannelException {
+        try {
+            final FS4Channel channel = backend.openPingChannel();
+            server.dispatch.channelId = -1;
+            server.dispatch.packetData = PONG;
+
+            assertTrue(channel.sendPacket(new PingPacket()));
+            try {
+                channel.receivePackets(1000, 1);
+            } catch (final ChannelTimeoutException e) {
+                fail("Could not get packets from simulated backend.");
+            }
+            final BackendStatistics stats = backend.getStatistics();
+            assertEquals(1, stats.totalConnections());
+        }
+        catch (java.net.UnknownHostException e) {
+            // We are on vpn, or have no network
+        }
+    }
+}