| #!/bin/bash |
| # |
| # Copyright 2016 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. |
| # |
| # Tests remote execution and caching. |
| # |
| |
| # Load the test setup defined in the parent directory |
| CURRENT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" |
| source "${CURRENT_DIR}/../integration_test_setup.sh" \ |
| || { echo "integration_test_setup.sh not found!" >&2; exit 1; } |
| |
| function set_up() { |
| work_path=$(mktemp -d "${TEST_TMPDIR}/remote.XXXXXXXX") |
| pid_file=$(mktemp -u "${TEST_TMPDIR}/remote.XXXXXXXX") |
| attempts=1 |
| while [ $attempts -le 5 ]; do |
| (( attempts++ )) |
| worker_port=$(pick_random_unused_tcp_port) || fail "no port found" |
| hazelcast_port=$(pick_random_unused_tcp_port) || fail "no port found" |
| "${bazel_data}/src/tools/remote_worker/remote_worker" \ |
| --work_path="${work_path}" \ |
| --listen_port=${worker_port} \ |
| --hazelcast_standalone_listen_port=${hazelcast_port} \ |
| --pid_file="${pid_file}" >& $TEST_log & |
| local wait_seconds=0 |
| until [ -s "${pid_file}" ] || [ "$wait_seconds" -eq 15 ]; do |
| sleep 1 |
| ((wait_seconds++)) || true |
| done |
| if [ -s "${pid_file}" ]; then |
| break |
| fi |
| done |
| if [ ! -s "${pid_file}" ]; then |
| fail "Timed out waiting for remote worker to start." |
| fi |
| } |
| |
| function tear_down() { |
| if [ -s "${pid_file}" ]; then |
| local pid=$(cat "${pid_file}") |
| kill "${pid}" || true |
| fi |
| rm -rf "${pid_file}" |
| rm -rf "${work_path}" |
| } |
| |
| function test_cc_binary() { |
| if [[ "$PLATFORM" == "darwin" ]]; then |
| # TODO(b/37355380): This test is disabled due to RemoteWorker not supporting |
| # setting SDKROOT and DEVELOPER_DIR appropriately, as is required of |
| # action executors in order to select the appropriate Xcode toolchain. |
| return 0 |
| fi |
| |
| mkdir -p a |
| cat > a/BUILD <<EOF |
| package(default_visibility = ["//visibility:public"]) |
| cc_binary( |
| name = 'test', |
| srcs = [ 'test.cc' ], |
| ) |
| EOF |
| cat > a/test.cc <<EOF |
| #include <iostream> |
| int main() { std::cout << "Hello world!" << std::endl; return 0; } |
| EOF |
| bazel build //a:test >& $TEST_log \ |
| || fail "Failed to build //a:test without remote execution" |
| cp -f bazel-bin/a/test ${TEST_TMPDIR}/test_expected |
| |
| bazel clean --expunge >& $TEST_log |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 build \ |
| --spawn_strategy=remote \ |
| --remote_executor=localhost:${worker_port} \ |
| --remote_cache=localhost:${worker_port} \ |
| //a:test >& $TEST_log \ |
| || fail "Failed to build //a:test with remote execution" |
| diff bazel-bin/a/test ${TEST_TMPDIR}/test_expected \ |
| || fail "Remote execution generated different result" |
| } |
| |
| function test_cc_test() { |
| mkdir -p a |
| cat > a/BUILD <<EOF |
| package(default_visibility = ["//visibility:public"]) |
| cc_test( |
| name = 'test', |
| srcs = [ 'test.cc' ], |
| ) |
| EOF |
| cat > a/test.cc <<EOF |
| #include <iostream> |
| int main() { std::cout << "Hello test!" << std::endl; return 0; } |
| EOF |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 test \ |
| --spawn_strategy=remote \ |
| --remote_executor=localhost:${worker_port} \ |
| --remote_cache=localhost:${worker_port} \ |
| --test_output=errors \ |
| //a:test >& $TEST_log \ |
| || fail "Failed to run //a:test with remote execution" |
| } |
| |
| function test_cc_binary_grpc_cache() { |
| mkdir -p a |
| cat > a/BUILD <<EOF |
| package(default_visibility = ["//visibility:public"]) |
| cc_binary( |
| name = 'test', |
| srcs = [ 'test.cc' ], |
| ) |
| EOF |
| cat > a/test.cc <<EOF |
| #include <iostream> |
| int main() { std::cout << "Hello world!" << std::endl; return 0; } |
| EOF |
| bazel build //a:test >& $TEST_log \ |
| || fail "Failed to build //a:test without remote cache" |
| cp -f bazel-bin/a/test ${TEST_TMPDIR}/test_expected |
| |
| bazel clean --expunge >& $TEST_log |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 build \ |
| --spawn_strategy=remote \ |
| --remote_cache=localhost:${worker_port} \ |
| //a:test >& $TEST_log \ |
| || fail "Failed to build //a:test with remote gRPC cache service" |
| diff bazel-bin/a/test ${TEST_TMPDIR}/test_expected \ |
| || fail "Remote cache generated different result" |
| } |
| |
| function test_failing_cc_test() { |
| mkdir -p a |
| cat > a/BUILD <<EOF |
| package(default_visibility = ["//visibility:public"]) |
| cc_test( |
| name = 'test', |
| srcs = [ 'test.cc' ], |
| ) |
| EOF |
| cat > a/test.cc <<EOF |
| #include <iostream> |
| int main() { std::cout << "Fail me!" << std::endl; return 1; } |
| EOF |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 test \ |
| --spawn_strategy=remote \ |
| --remote_executor=localhost:${worker_port} \ |
| --remote_cache=localhost:${worker_port} \ |
| --test_output=errors \ |
| //a:test >& $TEST_log \ |
| && fail "Expected test failure" || exitcode=$? |
| # TODO(ulfjack): Check that the test failure gets reported correctly. |
| } |
| |
| # Tests that the remote worker can return a 200MB blob that requires chunking. |
| # Blob has to be that large in order to exceed the grpc default max message size. |
| function test_genrule_large_output_chunking() { |
| mkdir -p a |
| cat > a/BUILD <<EOF |
| package(default_visibility = ["//visibility:public"]) |
| genrule( |
| name = "large_output", |
| srcs = ["small_blob.txt"], |
| outs = ["large_blob.txt"], |
| cmd = "cp \$(location small_blob.txt) tmp.txt; " + |
| "(for i in {1..22} ; do cat tmp.txt >> \$@; cp \$@ tmp.txt; done)", |
| ) |
| EOF |
| cat > a/small_blob.txt <<EOF |
| 0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890 |
| EOF |
| bazel build //a:large_output >& $TEST_log \ |
| || fail "Failed to build //a:large_output without remote execution" |
| cp -f bazel-genfiles/a/large_blob.txt ${TEST_TMPDIR}/large_blob_expected.txt |
| |
| bazel clean --expunge >& $TEST_log |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 build \ |
| --spawn_strategy=remote \ |
| --remote_executor=localhost:${worker_port} \ |
| --remote_cache=localhost:${worker_port} \ |
| //a:large_output >& $TEST_log \ |
| || fail "Failed to build //a:large_output with remote execution" |
| diff bazel-genfiles/a/large_blob.txt ${TEST_TMPDIR}/large_blob_expected.txt \ |
| || fail "Remote execution generated different result" |
| } |
| |
| function test_cc_binary_rest_cache() { |
| mkdir -p a |
| cat > a/BUILD <<EOF |
| package(default_visibility = ["//visibility:public"]) |
| cc_binary( |
| name = 'test', |
| srcs = [ 'test.cc' ], |
| ) |
| EOF |
| cat > a/test.cc <<EOF |
| #include <iostream> |
| int main() { std::cout << "Hello world!" << std::endl; return 0; } |
| EOF |
| bazel build //a:test >& $TEST_log \ |
| || fail "Failed to build //a:test without remote cache" |
| cp -f bazel-bin/a/test ${TEST_TMPDIR}/test_expected |
| |
| bazel clean --expunge >& $TEST_log |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 build \ |
| --spawn_strategy=remote \ |
| --remote_rest_cache=http://localhost:${hazelcast_port}/hazelcast/rest/maps \ |
| //a:test >& $TEST_log \ |
| || fail "Failed to build //a:test with remote REST cache service" |
| diff bazel-bin/a/test ${TEST_TMPDIR}/test_expected \ |
| || fail "Remote cache generated different result" |
| # Check that persistent connections are closed after the build. Is there a good cross-platform way |
| # to check this? |
| if [[ "$PLATFORM" = "linux" ]]; then |
| if netstat -tn | grep -qE ":${hazelcast_port}\\s+ESTABLISHED$"; then |
| fail "connections to to cache not closed" |
| fi |
| fi |
| } |
| |
| function test_py_test() { |
| mkdir -p a |
| cat > a/BUILD <<EOF |
| package(default_visibility = ["//visibility:public"]) |
| py_test( |
| name = 'test', |
| srcs = [ 'test.py' ], |
| ) |
| EOF |
| cat > a/test.py <<'EOF' |
| import sys |
| if __name__ == "__main__": |
| sys.exit(0) |
| EOF |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 test \ |
| --spawn_strategy=remote \ |
| --remote_executor=localhost:${worker_port} \ |
| --remote_cache=localhost:${worker_port} \ |
| --test_output=errors \ |
| //a:test >& $TEST_log \ |
| || fail "Failed to run //a:test with remote execution" |
| } |
| |
| function test_py_test_with_xml_output() { |
| mkdir -p a |
| cat > a/BUILD <<EOF |
| package(default_visibility = ["//visibility:public"]) |
| py_test( |
| name = 'test', |
| srcs = [ 'test.py' ], |
| ) |
| EOF |
| cat > a/test.py <<'EOF' |
| import sys |
| import os |
| if __name__ == "__main__": |
| f = open(os.environ['XML_OUTPUT_FILE'], "w") |
| f.write(''' |
| <?xml version="1.0" encoding="UTF-8"?> |
| <testsuites> |
| <testsuite name="test" tests="1" failures="1" errors="1"> |
| <testcase name="first" status="run"> |
| <failure>That did not work!</failure> |
| </testcase> |
| </testsuite> |
| </testsuites> |
| ''') |
| sys.exit(0) |
| EOF |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 test \ |
| --spawn_strategy=remote \ |
| --remote_executor=localhost:${worker_port} \ |
| --remote_cache=localhost:${worker_port} \ |
| --test_output=errors \ |
| //a:test >& $TEST_log \ |
| || fail "Failed to run //a:test with remote execution" |
| xml=bazel-testlogs/a/test/test.xml |
| [ -e $xml ] || fail "Expected to find XML output" |
| cat $xml > $TEST_log |
| expect_log 'That did not work!' |
| } |
| |
| function test_failing_py_test_with_xml_output() { |
| mkdir -p a |
| cat > a/BUILD <<EOF |
| package(default_visibility = ["//visibility:public"]) |
| py_test( |
| name = 'test', |
| srcs = [ 'test.py' ], |
| ) |
| EOF |
| cat > a/test.py <<'EOF' |
| import sys |
| import os |
| if __name__ == "__main__": |
| f = open(os.environ['XML_OUTPUT_FILE'], "w") |
| f.write(''' |
| <?xml version="1.0" encoding="UTF-8"?> |
| <testsuites> |
| <testsuite name="test" tests="1" failures="1" errors="1"> |
| <testcase name="first" status="run"> |
| <failure>That did not work!</failure> |
| </testcase> |
| </testsuite> |
| </testsuites> |
| ''') |
| sys.exit(1) |
| EOF |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 test \ |
| --spawn_strategy=remote \ |
| --remote_executor=localhost:${worker_port} \ |
| --remote_cache=localhost:${worker_port} \ |
| --test_output=errors \ |
| //a:test >& $TEST_log \ |
| && fail "Expected test failure" || exitcode=$? |
| xml=bazel-testlogs/a/test/test.xml |
| [ -e $xml ] || fail "Expected to find XML output" |
| cat $xml > $TEST_log |
| expect_log 'That did not work!' |
| } |
| |
| function test_noinput_action() { |
| mkdir -p a |
| cat > a/rule.bzl <<'EOF' |
| def _impl(ctx): |
| output = ctx.outputs.out |
| ctx.actions.run_shell( |
| outputs=[output], |
| command="echo 'Hello World' > %s" % (output.path)) |
| |
| empty = rule( |
| implementation=_impl, |
| outputs={"out": "%{name}.txt"}, |
| ) |
| EOF |
| cat > a/BUILD <<'EOF' |
| load("//a:rule.bzl", "empty") |
| package(default_visibility = ["//visibility:public"]) |
| empty(name = 'test') |
| EOF |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 build \ |
| --spawn_strategy=remote \ |
| --remote_cache=localhost:${worker_port} \ |
| --test_output=errors \ |
| //a:test >& $TEST_log \ |
| || fail "Failed to run //a:test with remote execution" |
| } |
| |
| function test_timeout() { |
| mkdir -p a |
| cat > a/BUILD <<'EOF' |
| sh_test( |
| name = "sleep", |
| timeout = "short", |
| srcs = ["sleep.sh"], |
| ) |
| EOF |
| |
| cat > a/sleep.sh <<'EOF' |
| #!/bin/sh |
| sleep 2 |
| EOF |
| chmod +x a/sleep.sh |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 test \ |
| --spawn_strategy=remote \ |
| --remote_executor=localhost:${worker_port} \ |
| --test_output=errors \ |
| --test_timeout=1,1,1,1 \ |
| //a:sleep >& $TEST_log \ |
| && fail "Test failure (timeout) expected" || true |
| expect_log "TIMEOUT" |
| } |
| |
| function test_passed_env_user() { |
| mkdir -p a |
| cat > a/BUILD <<'EOF' |
| sh_test( |
| name = "user_test", |
| timeout = "short", |
| srcs = ["user_test.sh"], |
| ) |
| EOF |
| |
| cat > a/user_test.sh <<'EOF' |
| #!/bin/sh |
| echo "user=$USER" |
| EOF |
| chmod +x a/user_test.sh |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 test \ |
| --spawn_strategy=remote \ |
| --remote_executor=localhost:${worker_port} \ |
| --test_output=all \ |
| --test_env=USER=boo \ |
| //a:user_test >& $TEST_log \ |
| || fail "Failed to run //a:user_test with remote execution" |
| expect_log "user=boo" |
| |
| # Rely on the test-setup script to set USER value to whoami. |
| export USER= |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 test \ |
| --spawn_strategy=remote \ |
| --remote_executor=localhost:${worker_port} \ |
| --test_output=all \ |
| //a:user_test >& $TEST_log \ |
| || fail "Failed to run //a:user_test with remote execution" |
| expect_log "user=$(whoami)" |
| } |
| |
| function test_exitcode() { |
| mkdir -p a |
| cat > a/BUILD <<'EOF' |
| genrule( |
| name = "foo", |
| srcs = [], |
| outs = ["foo.txt"], |
| cmd = "echo \"hello world\" > \"$@\"", |
| ) |
| EOF |
| |
| (set +e |
| bazel --host_jvm_args=-Dbazel.DigestFunction=SHA1 build \ |
| --genrule_strategy=remote \ |
| --remote_executor=bazel-test-does-not-exist \ |
| //a:foo >& $TEST_log |
| [ $? -eq 34 ]) || fail "Test failed due to wrong exit code" |
| } |
| |
| # TODO(alpha): Add a test that fails remote execution when remote worker |
| # supports sandbox. |
| |
| run_suite "Remote execution and remote cache tests" |