mirror of https://github.com/usememos/memos.git
fix(scripts): add file existence check in file_env (#5424)
Signed-off-by: majiayu000 <1835304752@qq.com>
This commit is contained in:
parent
cbbb585b2e
commit
86013d6c2a
|
|
@ -15,6 +15,10 @@ file_env() {
|
|||
if [ -n "$val_var" ]; then
|
||||
val="$val_var"
|
||||
elif [ -n "$val_fileVar" ]; then
|
||||
if [ ! -r "$val_fileVar" ]; then
|
||||
echo "error: file '$val_fileVar' does not exist or is not readable" >&2
|
||||
exit 1
|
||||
fi
|
||||
val="$(cat "$val_fileVar")"
|
||||
fi
|
||||
|
||||
|
|
|
|||
|
|
@ -0,0 +1,131 @@
|
|||
#!/usr/bin/env sh
|
||||
|
||||
# Test script for entrypoint.sh file_env function
|
||||
# Run: ./scripts/entrypoint_test.sh
|
||||
|
||||
set -e
|
||||
|
||||
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||||
TEMP_DIR=$(mktemp -d)
|
||||
trap "rm -rf $TEMP_DIR" EXIT
|
||||
|
||||
# Colors for output
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
pass_count=0
|
||||
fail_count=0
|
||||
|
||||
pass() {
|
||||
echo "${GREEN}PASS${NC}: $1"
|
||||
pass_count=$((pass_count + 1))
|
||||
}
|
||||
|
||||
fail() {
|
||||
echo "${RED}FAIL${NC}: $1"
|
||||
fail_count=$((fail_count + 1))
|
||||
}
|
||||
|
||||
# Test 1: Direct env var works
|
||||
test_direct_env_var() {
|
||||
unset MEMOS_DSN MEMOS_DSN_FILE
|
||||
export MEMOS_DSN="direct_value"
|
||||
|
||||
result=$("$SCRIPT_DIR/entrypoint.sh" sh -c 'echo $MEMOS_DSN' 2>&1)
|
||||
if [ "$result" = "direct_value" ]; then
|
||||
pass "Direct env var works"
|
||||
else
|
||||
fail "Direct env var: expected 'direct_value', got '$result'"
|
||||
fi
|
||||
unset MEMOS_DSN
|
||||
}
|
||||
|
||||
# Test 2: File env var works with readable file
|
||||
test_file_env_var_readable() {
|
||||
unset MEMOS_DSN MEMOS_DSN_FILE
|
||||
echo "file_value" > "$TEMP_DIR/dsn_file"
|
||||
export MEMOS_DSN_FILE="$TEMP_DIR/dsn_file"
|
||||
|
||||
result=$("$SCRIPT_DIR/entrypoint.sh" sh -c 'echo $MEMOS_DSN' 2>&1)
|
||||
if [ "$result" = "file_value" ]; then
|
||||
pass "File env var with readable file works"
|
||||
else
|
||||
fail "File env var readable: expected 'file_value', got '$result'"
|
||||
fi
|
||||
unset MEMOS_DSN_FILE
|
||||
}
|
||||
|
||||
# Test 3: Error when file doesn't exist
|
||||
test_file_env_var_missing() {
|
||||
unset MEMOS_DSN MEMOS_DSN_FILE
|
||||
export MEMOS_DSN_FILE="$TEMP_DIR/nonexistent_file"
|
||||
|
||||
if result=$("$SCRIPT_DIR/entrypoint.sh" sh -c 'echo $MEMOS_DSN' 2>&1); then
|
||||
fail "Missing file should fail, but succeeded with: $result"
|
||||
else
|
||||
if echo "$result" | grep -q "does not exist or is not readable"; then
|
||||
pass "Missing file returns error"
|
||||
else
|
||||
fail "Missing file error message unexpected: $result"
|
||||
fi
|
||||
fi
|
||||
unset MEMOS_DSN_FILE
|
||||
}
|
||||
|
||||
# Test 4: Error when file is not readable
|
||||
test_file_env_var_unreadable() {
|
||||
unset MEMOS_DSN MEMOS_DSN_FILE
|
||||
echo "secret" > "$TEMP_DIR/unreadable_file"
|
||||
chmod 000 "$TEMP_DIR/unreadable_file"
|
||||
export MEMOS_DSN_FILE="$TEMP_DIR/unreadable_file"
|
||||
|
||||
if result=$("$SCRIPT_DIR/entrypoint.sh" sh -c 'echo $MEMOS_DSN' 2>&1); then
|
||||
fail "Unreadable file should fail, but succeeded with: $result"
|
||||
else
|
||||
if echo "$result" | grep -q "does not exist or is not readable"; then
|
||||
pass "Unreadable file returns error"
|
||||
else
|
||||
fail "Unreadable file error message unexpected: $result"
|
||||
fi
|
||||
fi
|
||||
chmod 644 "$TEMP_DIR/unreadable_file" 2>/dev/null || true
|
||||
unset MEMOS_DSN_FILE
|
||||
}
|
||||
|
||||
# Test 5: Error when both var and file are set
|
||||
test_both_set_error() {
|
||||
unset MEMOS_DSN MEMOS_DSN_FILE
|
||||
echo "file_value" > "$TEMP_DIR/dsn_file"
|
||||
export MEMOS_DSN="direct_value"
|
||||
export MEMOS_DSN_FILE="$TEMP_DIR/dsn_file"
|
||||
|
||||
if result=$("$SCRIPT_DIR/entrypoint.sh" sh -c 'echo $MEMOS_DSN' 2>&1); then
|
||||
fail "Both set should fail, but succeeded with: $result"
|
||||
else
|
||||
if echo "$result" | grep -q "are set (but are exclusive)"; then
|
||||
pass "Both var and file set returns error"
|
||||
else
|
||||
fail "Both set error message unexpected: $result"
|
||||
fi
|
||||
fi
|
||||
unset MEMOS_DSN MEMOS_DSN_FILE
|
||||
}
|
||||
|
||||
# Run all tests
|
||||
echo "Running entrypoint.sh tests..."
|
||||
echo "================================"
|
||||
|
||||
test_direct_env_var
|
||||
test_file_env_var_readable
|
||||
test_file_env_var_missing
|
||||
test_file_env_var_unreadable
|
||||
test_both_set_error
|
||||
|
||||
echo "================================"
|
||||
echo "Tests completed: ${GREEN}$pass_count passed${NC}, ${RED}$fail_count failed${NC}"
|
||||
|
||||
if [ $fail_count -gt 0 ]; then
|
||||
exit 1
|
||||
fi
|
||||
exit 0
|
||||
Loading…
Reference in New Issue