mirror of
https://github.com/wassname/ray.git
synced 2026-07-07 21:50:14 +08:00
fangfengbin
70 lines
2.2 KiB
Bash
70 lines
2.2 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# Run all streaming c++ tests using streaming queue, instead of plasma queue
|
|
# This needs to be run in the root directory.
|
|
|
|
# Try to find an unused port for raylet to use.
|
|
PORTS="2000 2001 2002 2003 2004 2005 2006 2007 2008 2009"
|
|
RAYLET_PORT=0
|
|
for port in $PORTS; do
|
|
nc -z localhost $port
|
|
if [[ $? != 0 ]]; then
|
|
RAYLET_PORT=$port
|
|
break
|
|
fi
|
|
done
|
|
|
|
if [[ $RAYLET_PORT == 0 ]]; then
|
|
echo "WARNING: Could not find unused port for raylet to use. Exiting without running tests."
|
|
exit
|
|
fi
|
|
|
|
# Cause the script to exit if a single command fails.
|
|
set -e
|
|
set -x
|
|
|
|
# Get the directory in which this script is executing.
|
|
SCRIPT_DIR="`dirname \"$0\"`"
|
|
|
|
# Get the directory in which this script is executing.
|
|
SCRIPT_DIR="`dirname \"$0\"`"
|
|
RAY_ROOT="$SCRIPT_DIR/../../.."
|
|
# Makes $RAY_ROOT an absolute path.
|
|
RAY_ROOT="`( cd \"$RAY_ROOT\" && pwd )`"
|
|
if [ -z "$RAY_ROOT" ] ; then
|
|
exit 1
|
|
fi
|
|
|
|
bazel build "//:core_worker_test" "//:mock_worker" "//:raylet" "//:gcs_server" "//:libray_redis_module.so" "@plasma//:plasma_store_server"
|
|
bazel build //streaming:streaming_test_worker
|
|
bazel build //streaming:streaming_queue_tests
|
|
|
|
# Ensure we're in the right directory.
|
|
if [ ! -d "$RAY_ROOT/python" ]; then
|
|
echo "Unable to find root Ray directory. Has this script moved?"
|
|
exit 1
|
|
fi
|
|
|
|
REDIS_MODULE="./bazel-bin/libray_redis_module.so"
|
|
LOAD_MODULE_ARGS="--loadmodule ${REDIS_MODULE}"
|
|
STORE_EXEC="./bazel-bin/external/plasma/plasma_store_server"
|
|
RAYLET_EXEC="./bazel-bin/raylet"
|
|
STREAMING_TEST_WORKER_EXEC="./bazel-bin/streaming/streaming_test_worker"
|
|
GCS_SERVER_EXEC="./bazel-bin/gcs_server"
|
|
|
|
# Allow cleanup commands to fail.
|
|
bazel run //:redis-cli -- -p 6379 shutdown || true
|
|
sleep 1s
|
|
bazel run //:redis-cli -- -p 6380 shutdown || true
|
|
sleep 1s
|
|
bazel run //:redis-server -- --loglevel warning ${LOAD_MODULE_ARGS} --port 6379 &
|
|
sleep 2s
|
|
bazel run //:redis-server -- --loglevel warning ${LOAD_MODULE_ARGS} --port 6380 &
|
|
sleep 2s
|
|
# Run tests.
|
|
./bazel-bin/streaming/streaming_queue_tests $STORE_EXEC $RAYLET_EXEC $RAYLET_PORT $STREAMING_TEST_WORKER_EXEC $GCS_SERVER_EXEC
|
|
sleep 1s
|
|
bazel run //:redis-cli -- -p 6379 shutdown
|
|
bazel run //:redis-cli -- -p 6380 shutdown
|
|
sleep 1s
|