107 lines
2.3 KiB
Plaintext
107 lines
2.3 KiB
Plaintext
|
#!/bin/bash
|
||
|
|
|
||
|
|
## Helper functions
|
||
|
|
|
||
|
|
function wait_for_server {
|
||
|
|
port=$1
|
||
|
|
while ! nc -z -w 1 127.0.0.1 $port; do
|
||
|
|
sleep 0.1
|
||
|
|
done
|
||
|
|
sleep 1
|
||
|
|
}
|
||
|
|
|
||
|
|
function echo_info { printf "\033[1;36m~~ $1 ~~\033[0m\n"; }
|
||
|
|
function echo_success { printf "\033[1;32m~~ $1 ~~\033[0m\n\n"; }
|
||
|
|
function echo_failure { printf "\033[1;31m~~ $1 ~~\033[0m\n\n"; }
|
||
|
|
|
||
|
|
|
||
|
|
## Environment setup
|
||
|
|
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
|
||
|
|
cd "$DIR"
|
||
|
|
|
||
|
|
# Create a temporary directory for output files
|
||
|
|
tmpdir=/tmp/mg_client/output
|
||
|
|
if [ -d $tmpdir ]; then
|
||
|
|
rm -rf $tmpdir
|
||
|
|
fi
|
||
|
|
mkdir -p $tmpdir
|
||
|
|
cd $tmpdir
|
||
|
|
|
||
|
|
# Find memgraph binaries.
|
||
|
|
memgraph_dir="$DIR/../../build"
|
||
|
|
if [ ! -d $memgraph_dir ]; then
|
||
|
|
memgraph_dir="$DIR/../../build_release"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Find mg_client binaries.
|
||
|
|
client_dir="$memgraph_dir/tools/src"
|
||
|
|
if [ ! -d $client_dir ]; then
|
||
|
|
echo_failure "mg-client directory not found"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Find tests input files.
|
||
|
|
tests_dir="$DIR/client"
|
||
|
|
if [ ! -d $tests_dir ]; then
|
||
|
|
echo_failure "Directory with tests input not found"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
|
||
|
|
## Startup
|
||
|
|
|
||
|
|
# Start the memgraph process and wait for it to start.
|
||
|
|
echo_info "Starting memgraph"
|
||
|
|
$memgraph_dir/memgraph &> /dev/null &
|
||
|
|
pid=$!
|
||
|
|
wait_for_server 7687
|
||
|
|
echo_success "Started memgraph"
|
||
|
|
|
||
|
|
|
||
|
|
## Tests
|
||
|
|
|
||
|
|
echo_info "Running tests"
|
||
|
|
echo # Blank line
|
||
|
|
|
||
|
|
client_flags="--use-ssl=false"
|
||
|
|
test_code=0
|
||
|
|
for filename in $tests_dir/input_*.txt; do
|
||
|
|
test_name=$(basename $filename)
|
||
|
|
test_name=${test_name#*_}
|
||
|
|
test_name=${test_name%.*}
|
||
|
|
output_name="output_$test_name.txt"
|
||
|
|
|
||
|
|
echo_info "Running test $test_name"
|
||
|
|
$client_dir/mg_client $client_flags < $filename > $tmpdir/$test_name
|
||
|
|
diff -b $tmpdir/$test_name $tests_dir/$output_name
|
||
|
|
test_code=$?
|
||
|
|
if [ $test_code -ne 0 ]; then
|
||
|
|
echo_failure "Test $test_name failed"
|
||
|
|
break
|
||
|
|
else
|
||
|
|
echo_success "Test $test_name passed"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Clear database for each test.
|
||
|
|
$client_dir/mg_client $client_flags <<< "MATCH (n) DETACH DELETE n;" \
|
||
|
|
&> /dev/null || exit 1
|
||
|
|
done
|
||
|
|
|
||
|
|
|
||
|
|
## Cleanup
|
||
|
|
echo_info "Starting test cleanup"
|
||
|
|
|
||
|
|
# Shutdown the memgraph process.
|
||
|
|
kill $pid
|
||
|
|
wait -n
|
||
|
|
code_mg=$?
|
||
|
|
|
||
|
|
# Check memgraph exit code.
|
||
|
|
if [ $code_mg -ne 0 ]; then
|
||
|
|
echo_failure "The memgraph process didn't terminate properly!"
|
||
|
|
exit $code_mg
|
||
|
|
fi
|
||
|
|
echo_success "Test cleanup done"
|
||
|
|
|
||
|
|
exit $test_code
|