1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
// 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.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 org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
import java.io.File;
import java.util.List;
/**
* 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, long firstContactTime) throws Exception {
outErr.printOut(COMMAND_STDOUT);
outErr.printErr(COMMAND_STDERR);
return 42;
}
@Override
public boolean shutdown() {
return false;
}
};
@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());
}
}
|
