mirror of
https://github.com/rustfs/rustfs.git
synced 2026-08-11 15:46:53 +00:00
houseme
52 lines
1.5 KiB
Bash
Executable File
52 lines
1.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
repo_root=$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)
|
|
tmp_dir=$(mktemp -d)
|
|
trap 'rm -rf "$tmp_dir"' EXIT
|
|
|
|
mock_ssh="$tmp_dir/ssh"
|
|
cat >"$mock_ssh" <<'EOF'
|
|
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
if read -r consumed; then
|
|
printf 'ssh-consumed-stdin:%s\n' "$consumed" >>"$CALL_LOG"
|
|
exit 24
|
|
fi
|
|
printf 'ssh:%s\n' "$*" >>"$CALL_LOG"
|
|
echo "123 12.5 456789 00:01:02"
|
|
EOF
|
|
chmod +x "$mock_ssh"
|
|
|
|
export CALL_LOG="$tmp_dir/calls.log"
|
|
output=$("$repo_root/scripts/sample_remote_rustfs_rss.sh" \
|
|
--nodes vm004,vm005 \
|
|
--duration-secs 1 \
|
|
--interval-secs 1 \
|
|
--out "$tmp_dir/rss.tsv" \
|
|
--ssh-bin "$mock_ssh")
|
|
|
|
[[ "$output" == "rss_samples=2" ]]
|
|
[[ "$(wc -l <"$tmp_dir/rss.tsv")" -eq 3 ]]
|
|
grep -q $'^ts_utc\tnode\tpid\tpcpu\trss_kib\tetime$' "$tmp_dir/rss.tsv"
|
|
grep -q $'\tvm004\t123\t12.5\t456789\t00:01:02$' "$tmp_dir/rss.tsv"
|
|
grep -q $'\tvm005\t123\t12.5\t456789\t00:01:02$' "$tmp_dir/rss.tsv"
|
|
[[ "$(grep -c '^ssh:' "$CALL_LOG")" -eq 2 ]]
|
|
if grep -q '^ssh-consumed-stdin:' "$CALL_LOG"; then
|
|
echo "ssh consumed the sampling loop stdin" >&2
|
|
exit 1
|
|
fi
|
|
grep -q 'sudo su - root -c' "$CALL_LOG"
|
|
|
|
if "$repo_root/scripts/sample_remote_rustfs_rss.sh" \
|
|
--nodes 'vm004;rm' \
|
|
--duration-secs 1 \
|
|
--out "$tmp_dir/bad.tsv" \
|
|
--ssh-bin "$mock_ssh" >"$tmp_dir/bad.stdout" 2>"$tmp_dir/bad.stderr"; then
|
|
echo "unsafe node name was accepted" >&2
|
|
exit 1
|
|
fi
|
|
grep -q 'node contains unsafe characters' "$tmp_dir/bad.stderr"
|
|
|
|
echo "test_sample_remote_rustfs_rss: ok"
|