2 files changed, 223 insertions, 0 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/server/AfUnixServerTest.java b/src/test/java/com/google/devtools/build/lib/server/AfUnixServerTest.java
new file mode 100644
index 0000000000..58948e1720
--- /dev/null
+++ b/src/test/java/com/google/devtools/build/lib/server/AfUnixServerTest.java
@@ -0,0 +1,135 @@
+// Copyright 2015 The Bazel Authors. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//    http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.google.devtools.build.lib.server;
+
+import static com.google.common.truth.Truth.assertThat;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+import com.google.devtools.build.lib.runtime.BlazeCommandDispatcher.LockingMode;
+import com.google.devtools.build.lib.runtime.BlazeCommandDispatcher.ShutdownMethod;
+import com.google.devtools.build.lib.testutil.Suite;
+import com.google.devtools.build.lib.testutil.TestSpec;
+import com.google.devtools.build.lib.util.JavaClock;
+import com.google.devtools.build.lib.util.io.OutErr;
+import com.google.devtools.build.lib.util.io.RecordingOutErr;
+import com.google.devtools.build.lib.vfs.FileSystemUtils;
+import com.google.devtools.build.lib.vfs.Path;
+import com.google.devtools.build.lib.vfs.util.FsApparatus;
+import java.io.File;
+import java.util.List;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+/**
+ * Run a real RPC server on localhost, and talk to it using the testing
+ * client.
+ */
+@TestSpec(size = Suite.MEDIUM_TESTS)
+@RunWith(JUnit4.class)
+public class AfUnixServerTest {
+
+  private static final long MAX_IDLE_MILLIS = 10000;
+  private static final long HEALTH_CHECK_MILLIS = 1000 * 3;
+  private static final String COMMAND_STDOUT = "Heelllloo....";
+  private static final String COMMAND_STDERR = "...world!";
+
+  private AfUnixServer server;
+  private FsApparatus scratch = FsApparatus.newNative();
+  private RecordingOutErr outErr = new RecordingOutErr();
+  private Path serverDir;
+  private Path workspaceDir;
+  private RPCTestingClient client;
+  private Thread serverThread = new Thread(){
+    @Override
+    public void run() {
+      server.serve();
+    }
+  };
+
+  private static final ServerCommand helloWorldCommand = new ServerCommand() {
+    @Override
+    public int exec(List<String> args, OutErr outErr, LockingMode lockingMode,
+        String clientDescription, long firstContactTime) {
+      outErr.printOut(COMMAND_STDOUT);
+      outErr.printErr(COMMAND_STDERR);
+      return 42;
+    }
+
+    @Override
+    public ShutdownMethod shutdown() {
+      return ShutdownMethod.NONE;
+    }
+  };
+
+  @Before
+  public final void startServer() throws Exception  {
+    // Do not use `createUnixTempDir()` here since the file name that results is longer
+    // than 108 characters, so cannot be used as local socket address.
+    File file = File.createTempFile("scratch", ".tmp", new File("/tmp"));
+    file.delete();
+    file.mkdir();
+    serverDir = this.scratch.dir(file.getAbsolutePath());
+
+    workspaceDir = this.scratch.createUnixTempDir();
+    workspaceDir.createDirectory();
+    client = new RPCTestingClient(
+        outErr, serverDir.getRelative("server.socket"));
+    RPCService service = new RPCService(helloWorldCommand);
+    server = new AfUnixServer(new JavaClock(), service, MAX_IDLE_MILLIS, HEALTH_CHECK_MILLIS,
+        serverDir, workspaceDir);
+    serverThread.start();
+  }
+
+  @After
+  public final void stopServer() throws Exception {
+    serverThread.interrupt();
+    serverThread.join();
+
+    FileSystemUtils.deleteTree(serverDir);
+  }
+
+  private void runTestRequest(String request, int ret, String out, String err) throws Exception {
+    assertEquals(ret, client.sendRequest(request));
+    assertEquals(out, outErr.outAsLatin1());
+    assertThat(outErr.errAsLatin1()).contains(err);
+  }
+
+  @Test
+  public void testUnknownCommand() throws Exception {
+    runTestRequest("unknown", 2, "", "SERVER ERROR: Unknown command: unknown\n");
+  }
+
+  @Test
+  public void testEmptyBlazeCommand() throws Exception {
+    runTestRequest("unknown", 2, "", "SERVER ERROR: Unknown command: unknown\n");
+  }
+
+  @Test
+  public void testWorkspaceDies() throws Exception {
+    assertTrue(serverThread.isAlive());
+    runTestRequest("blaze", 42, COMMAND_STDOUT, COMMAND_STDERR);
+    Thread.sleep(HEALTH_CHECK_MILLIS * 2);
+    assertTrue(serverThread.isAlive());
+
+    assertTrue(workspaceDir.delete());
+    Thread.sleep(HEALTH_CHECK_MILLIS * 2);
+    assertFalse(serverThread.isAlive());
+  }
+}
diff --git a/src/test/java/com/google/devtools/build/lib/server/RPCTestingClient.java b/src/test/java/com/google/devtools/build/lib/server/RPCTestingClient.java
new file mode 100644
index 0000000000..3b1515d2e2
--- /dev/null
+++ b/src/test/java/com/google/devtools/build/lib/server/RPCTestingClient.java
@@ -0,0 +1,88 @@
+// Copyright 2015 The Bazel Authors. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//    http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.google.devtools.build.lib.server;
+
+import static java.nio.charset.StandardCharsets.UTF_8;
+
+import com.google.common.io.ByteStreams;
+import com.google.devtools.build.lib.unix.LocalClientSocket;
+import com.google.devtools.build.lib.unix.LocalSocketAddress;
+import com.google.devtools.build.lib.util.io.RecordingOutErr;
+import com.google.devtools.build.lib.util.io.StreamDemultiplexer;
+import com.google.devtools.build.lib.vfs.Path;
+import java.io.ByteArrayOutputStream;
+import java.io.OutputStream;
+
+/**
+ * A client to test RPCServer.
+ */
+public class RPCTestingClient {
+
+  private final RecordingOutErr outErr;
+  private final Path socketFile;
+
+  /**
+   * Create a client to RPCServer. {@code socketFile} must be a file
+   * on disk; this will not work with the in-memory file system.
+   */
+  public RPCTestingClient(RecordingOutErr outErr, Path socketFile) {
+    this.socketFile = socketFile;
+    this.outErr = outErr;
+  }
+
+  public int sendRequest(String command, String... params)
+      throws Exception {
+    String request = command;
+    for (String param : params) {
+      request += "\0" + param;
+    }
+    return sendRequest(request);
+  }
+
+  public int sendRequest(String request) throws Exception {
+    LocalClientSocket connection = new LocalClientSocket();
+    connection.connect(new LocalSocketAddress(socketFile.getPathFile()));
+    try {
+      OutputStream out = connection.getOutputStream();
+      byte[] requestBytes = request.getBytes(UTF_8);
+      byte[] requestLength = new byte[4];
+      requestLength[0] = (byte) (requestBytes.length << 24);
+      requestLength[1] = (byte) ((requestBytes.length << 16) & 0xff);
+      requestLength[2] = (byte) ((requestBytes.length << 8) & 0xff);
+      requestLength[3] = (byte) (requestBytes.length & 0xff);
+      out.write(requestLength);
+      out.write(requestBytes);
+      out.flush();
+      connection.shutdownOutput();
+
+      OutputStream stdout = outErr.getOutputStream();
+      OutputStream stderr = outErr.getErrorStream();
+      ByteArrayOutputStream control = new ByteArrayOutputStream();
+      StreamDemultiplexer demux = new StreamDemultiplexer((byte) 1,
+          stdout, stderr, control);
+      ByteStreams.copy(connection.getInputStream(), demux);
+      demux.flush();
+
+      byte[] controlBytes = control.toByteArray();
+      return (((int) controlBytes[0]) << 24)
+          + (((int) controlBytes[1]) << 16)
+          + (((int) controlBytes[2]) << 8)
+          + ((int) controlBytes[3]);
+    } finally {
+      connection.close();
+    }
+  }
+
+}