blob: 091a9b70386d9557ded5b811650324a033d5ed2f (
plain)
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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
#!/bin/bash
#
# Copyright 2015 Google Inc. 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.
case "${PLATFORM}" in
darwin)
function nc_l() {
nc -l $1
}
;;
*)
function nc_l() {
nc -l -p $1 -q 1
}
;;
esac
# Serves $1 as a file on localhost:$nc_port. Sets the following variables:
# * nc_port - the port nc is listening on.
# * nc_log - the path to nc's log.
# * nc_pid - the PID of nc.
# * http_response - the full response nc will provide to a request.
# This also creates the file $TEST_TMPDIR/http_response.
function serve_file() {
http_response=$TEST_TMPDIR/http_response
cat > $http_response <<EOF
HTTP/1.0 200 OK
EOF
cat $1 >> $http_response
# Assign random_port to nc_port if not already set.
echo ${nc_port:=$(pick_random_unused_tcp_port)} > /dev/null
nc_log=$TEST_TMPDIR/nc.log
nc_l $nc_port < $http_response >& $nc_log &
nc_pid=$!
}
# Creates a jar carnivore.Mongoose and serves it using serve_file.
function serve_jar() {
make_test_jar
serve_file $test_jar
cd ${WORKSPACE_DIR}
}
function make_test_jar() {
pkg_dir=$TEST_TMPDIR/carnivore
rm -fr $pkg_dir
mkdir -p $pkg_dir
cat > $pkg_dir/Mongoose.java <<EOF
package carnivore;
public class Mongoose {
public static void frolic() {
System.out.println("Tra-la!");
}
}
EOF
${bazel_javabase}/bin/javac $pkg_dir/Mongoose.java
test_jar=$TEST_TMPDIR/libcarnivore.jar
cd ${TEST_TMPDIR}
${bazel_javabase}/bin/jar cf $test_jar carnivore/Mongoose.class
sha256=$(sha256sum $test_jar | cut -f 1 -d ' ')
# OS X doesn't have sha1sum, so use openssl.
sha1=$(openssl sha1 $test_jar | cut -f 2 -d ' ')
cd -
}
# Serves a redirection from localhost:$redirect_port to $1. Sets the following variables:
# * redirect_port - the port nc is listening on.
# * redirect_log - the path to nc's log.
# * redirect_pid - the PID of nc.
function serve_redirect() {
# Assign random_port to nc_port if not already set.
echo ${redirect_port:=$(pick_random_unused_tcp_port)} > /dev/null
redirect_log=$TEST_TMPDIR/redirect.log
local response=$(cat <<EOF
HTTP/1.0 301 Moved Permanently
Location: $1
EOF
)
nc_l $redirect_port >& $redirect_log <<<"$response" &
redirect_pid=$!
}
# Waits for the SimpleHTTPServer to actually start up before the test is run.
# Otherwise the entire test can run before the server starts listening for
# connections, which causes flakes.
function wait_for_server_startup() {
touch some-file
while ! curl localhost:$fileserver_port/some-file; do
echo "waiting for server, exit code: $?"
sleep 1
done
echo "done waiting for server, exit code: $?"
rm some-file
}
function create_artifact() {
local group_id=$1
local artifact_id=$2
local version=$3
make_test_jar
maven_path=$PWD/$(echo $group_id | sed 's/\./\//g')/$artifact_id/$version
mkdir -p $maven_path
openssl sha1 $test_jar > $maven_path/$artifact_id-$version.jar.sha1
mv $test_jar $maven_path/$artifact_id-$version.jar
}
function serve_artifact() {
startup_server $PWD
create_artifact $1 $2 $3
}
function startup_server() {
fileserver_root=$1
cd $fileserver_root
fileserver_port=$(pick_random_unused_tcp_port) || exit 1
python $python_server --port=$fileserver_port &
fileserver_pid=$!
wait_for_server_startup
cd -
}
function startup_auth_server() {
fileserver_port=$(pick_random_unused_tcp_port) || exit 1
python $python_server --port=$fileserver_port --auth=basic &
fileserver_pid=$!
wait_for_server_startup
}
function shutdown_server() {
# Try to kill nc, otherwise the test will time out if Bazel has a bug and
# didn't make a request to it.
[ -z "${fileserver_pid:-}" ] || kill $fileserver_pid || true
[ -z "${redirect_pid:-}" ] || kill $redirect_pid || true
[ -z "${nc_log:-}" ] || cat $nc_log
[ -z "${redirect_log:-}" ] || cat $redirect_log
}
function kill_nc() {
shutdown_server
}
|
