diff --git a/scripts/entrypoint.sh b/scripts/entrypoint.sh index a73f40136..ec62af83d 100755 --- a/scripts/entrypoint.sh +++ b/scripts/entrypoint.sh @@ -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 diff --git a/scripts/entrypoint_test.sh b/scripts/entrypoint_test.sh new file mode 100755 index 000000000..894638f2b --- /dev/null +++ b/scripts/entrypoint_test.sh @@ -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