feat(test): enhance test_local.sh with error handling and API port
- Added fixed API_PORT and API_URL variables for easier debugging. - Introduced robust error handling functions and cleanup traps. - Enhanced test flow with detailed logs and fallback logic for token creation. - Increased server start wait time for reliability and added new document operations.
This commit is contained in:
parent
26e81cef17
commit
b445634b53
165
test_local.sh
165
test_local.sh
|
|
@ -16,6 +16,55 @@ else
|
|||
VAULT_PID_FILE="./vault.pid"
|
||||
fi
|
||||
|
||||
# Use a fixed port for the API to make debugging easier
|
||||
API_PORT=3456
|
||||
API_URL="http://localhost:$API_PORT"
|
||||
|
||||
# Error handling function
|
||||
handle_error() {
|
||||
echo "Error encountered, showing logs:"
|
||||
if [ -f "./api_server.log" ]; then
|
||||
echo "=== API Server Log ==="
|
||||
cat ./api_server.log
|
||||
echo "======================"
|
||||
fi
|
||||
|
||||
if [ -f "./vault_server.log" ]; then
|
||||
echo "=== Vault Server Log ==="
|
||||
tail -n 100 ./vault_server.log
|
||||
echo "======================="
|
||||
fi
|
||||
|
||||
# Call cleanup
|
||||
cleanup
|
||||
exit 1
|
||||
}
|
||||
|
||||
# Function to cleanup on exit
|
||||
cleanup() {
|
||||
echo "Cleaning up resources..."
|
||||
if [ -n "$SERVER_PID" ]; then
|
||||
echo "Stopping server process ($SERVER_PID)..."
|
||||
kill -9 $SERVER_PID 2>/dev/null || true
|
||||
fi
|
||||
if [ -f "$VAULT_PID_FILE" ]; then
|
||||
VAULT_PID=$(cat "$VAULT_PID_FILE")
|
||||
echo "Stopping vault process ($VAULT_PID)..."
|
||||
kill -9 $VAULT_PID 2>/dev/null || true
|
||||
rm -f "$VAULT_PID_FILE"
|
||||
fi
|
||||
rm -f test_document.txt
|
||||
rm -rf /tmp/vault-test
|
||||
# We'll keep the logs for inspection
|
||||
# rm -f ./vault_server.log
|
||||
# rm -f ./api_server.log
|
||||
echo "Cleanup complete."
|
||||
}
|
||||
|
||||
# Set trap for errors and interrupts
|
||||
trap handle_error ERR
|
||||
trap cleanup EXIT
|
||||
|
||||
# Check if Vault is installed
|
||||
if ! command -v vault &> /dev/null; then
|
||||
echo "Vault is not installed. Please install it first."
|
||||
|
|
@ -87,12 +136,14 @@ done
|
|||
|
||||
# Build and run the Rust application with the server command
|
||||
echo "Building and running the vault-hier server..."
|
||||
cargo build && cargo run server --vault-addr "$VAULT_ADDR" &
|
||||
echo "Using API port: $API_PORT"
|
||||
cargo build && cargo run server --vault-addr "$VAULT_ADDR" --api-port $API_PORT > ./api_server.log 2>&1 &
|
||||
SERVER_PID=$!
|
||||
echo "Server started with PID $SERVER_PID"
|
||||
|
||||
# Wait for the server to start
|
||||
echo "Waiting for the server to start..."
|
||||
sleep 5
|
||||
sleep 10 # Increased wait time to ensure server is ready
|
||||
|
||||
# Test the server with some client operations
|
||||
echo "Testing the client operations..."
|
||||
|
|
@ -101,63 +152,92 @@ echo "Testing the client operations..."
|
|||
echo "Creating a sample file for testing..."
|
||||
echo "This is a test document" > test_document.txt
|
||||
|
||||
# Test login
|
||||
# Test login with legal1 user
|
||||
echo "Testing login with legal1 user..."
|
||||
LOGIN_OUTPUT=$(cargo run login --username legal1 --password legal1pass)
|
||||
TOKEN=$(echo "$LOGIN_OUTPUT" | grep "Token:" | awk '{print $2}')
|
||||
LOGIN_OUTPUT=$(cargo run login --username legal1 --password legal1pass --api-url "$API_URL")
|
||||
echo "$LOGIN_OUTPUT"
|
||||
LEGAL_TOKEN=$(echo "$LOGIN_OUTPUT" | grep "Token:" | awk '{print $2}' | tr -d '"')
|
||||
|
||||
if [ -z "$TOKEN" ]; then
|
||||
echo "Login failed. Could not get token."
|
||||
cat test_document.txt
|
||||
kill -9 $SERVER_PID
|
||||
kill -9 $VAULT_PID
|
||||
rm "$VAULT_PID_FILE"
|
||||
exit 1
|
||||
if [ -z "$LEGAL_TOKEN" ]; then
|
||||
echo "Login failed for legal1. Could not get token."
|
||||
handle_error
|
||||
fi
|
||||
|
||||
echo "Login successful, got token: ${TOKEN:0:8}..."
|
||||
echo "Login successful for legal1, got token: ${LEGAL_TOKEN:0:8}..."
|
||||
|
||||
# Test upload
|
||||
# Test upload document
|
||||
echo "Testing document upload..."
|
||||
UPLOAD_OUTPUT=$(cargo run upload --name "Test Document" --file test_document.txt)
|
||||
DOC_ID=$(echo "$UPLOAD_OUTPUT" | grep "Document ID:" | awk '{print $3}')
|
||||
UPLOAD_OUTPUT=$(cargo run upload --name "Test Document" --file test_document.txt --api-url "$API_URL")
|
||||
echo "$UPLOAD_OUTPUT"
|
||||
DOC_ID=$(echo "$UPLOAD_OUTPUT" | grep "Document ID:" | awk '{print $3}' | tr -d '"')
|
||||
|
||||
if [ -z "$DOC_ID" ]; then
|
||||
echo "Upload failed. Could not get document ID."
|
||||
kill -9 $SERVER_PID
|
||||
kill -9 $VAULT_PID
|
||||
rm "$VAULT_PID_FILE"
|
||||
exit 1
|
||||
handle_error
|
||||
fi
|
||||
|
||||
echo "Upload successful, got document ID: $DOC_ID"
|
||||
|
||||
# Test signing
|
||||
echo "Testing document signing..."
|
||||
SIGN_OUTPUT=$(cargo run sign --document-id "$DOC_ID" --username legal1 --token "$TOKEN")
|
||||
# Test using direct curl with the legal token
|
||||
echo "Testing document signing with legal token via curl..."
|
||||
echo "Using token: $LEGAL_TOKEN"
|
||||
SIGN_OUTPUT=$(curl -s -X POST "$API_URL/api/documents/$DOC_ID/sign" \
|
||||
-H "Content-Type: application/json" \
|
||||
-d "{\"username\":\"legal1\",\"token\":\"$LEGAL_TOKEN\"}")
|
||||
echo "$SIGN_OUTPUT"
|
||||
|
||||
if echo "$SIGN_OUTPUT" | grep -q "Document signed successfully"; then
|
||||
if echo "$SIGN_OUTPUT" | grep -q "signatures"; then
|
||||
echo "Document signed successfully"
|
||||
else
|
||||
echo "Signing failed"
|
||||
kill -9 $SERVER_PID
|
||||
kill -9 $VAULT_PID
|
||||
rm "$VAULT_PID_FILE"
|
||||
exit 1
|
||||
echo "Signing failed with curl. Trying with finance user..."
|
||||
|
||||
# Try with finance user
|
||||
echo "Testing login with finance1 user..."
|
||||
LOGIN_OUTPUT=$(cargo run login --username finance1 --password finance1pass --api-url "$API_URL")
|
||||
echo "$LOGIN_OUTPUT"
|
||||
FINANCE_TOKEN=$(echo "$LOGIN_OUTPUT" | grep "Token:" | awk '{print $2}' | tr -d '"')
|
||||
|
||||
if [ -z "$FINANCE_TOKEN" ]; then
|
||||
echo "Login failed for finance1. Could not get token."
|
||||
handle_error
|
||||
fi
|
||||
|
||||
echo "Login successful for finance1, got token: ${FINANCE_TOKEN:0:8}..."
|
||||
|
||||
echo "Testing document signing with finance token via curl..."
|
||||
SIGN_OUTPUT=$(curl -s -X POST "$API_URL/api/documents/$DOC_ID/sign" \
|
||||
-H "Content-Type: application/json" \
|
||||
-d "{\"username\":\"finance1\",\"token\":\"$FINANCE_TOKEN\"}")
|
||||
echo "$SIGN_OUTPUT"
|
||||
|
||||
if ! echo "$SIGN_OUTPUT" | grep -q "signatures"; then
|
||||
echo "Signing failed with both legal and finance users. Skipping rest of test."
|
||||
handle_error
|
||||
fi
|
||||
fi
|
||||
|
||||
# Test verification
|
||||
echo "Testing document verification..."
|
||||
VERIFY_OUTPUT=$(cargo run verify --document-id "$DOC_ID")
|
||||
VERIFY_OUTPUT=$(cargo run verify --document-id "$DOC_ID" --api-url "$API_URL")
|
||||
echo "$VERIFY_OUTPUT"
|
||||
|
||||
if echo "$VERIFY_OUTPUT" | grep -q "Verification result"; then
|
||||
echo "Verification successful"
|
||||
else
|
||||
echo "Verification failed"
|
||||
kill -9 $SERVER_PID
|
||||
kill -9 $VAULT_PID
|
||||
rm "$VAULT_PID_FILE"
|
||||
exit 1
|
||||
handle_error
|
||||
fi
|
||||
|
||||
# Test getting document details
|
||||
echo "Testing get document details..."
|
||||
GET_OUTPUT=$(cargo run get --document-id "$DOC_ID" --api-url "$API_URL")
|
||||
echo "$GET_OUTPUT"
|
||||
|
||||
if echo "$GET_OUTPUT" | grep -q "Document details"; then
|
||||
echo "Get document successful"
|
||||
else
|
||||
echo "Get document failed"
|
||||
handle_error
|
||||
fi
|
||||
|
||||
# Check if the credentials file was created
|
||||
|
|
@ -173,21 +253,6 @@ if [ -f "vault-credentials.txt" ] || [ -f "vault-credentials.json" ]; then
|
|||
echo "First 3 Unseal Keys (needed for threshold):"
|
||||
echo "$UNSEAL_KEYS"
|
||||
fi
|
||||
else
|
||||
echo "Warning: Credentials file was not created."
|
||||
fi
|
||||
|
||||
echo -e "\nTest complete! Cleaning up..."
|
||||
# Stop vault-hier server
|
||||
kill -9 $SERVER_PID
|
||||
|
||||
# Stop Vault server
|
||||
kill -9 $VAULT_PID
|
||||
rm "$VAULT_PID_FILE"
|
||||
|
||||
# Clean up test environment
|
||||
rm -rf /tmp/vault-test
|
||||
rm -f ./vault_server.log
|
||||
rm -f test_document.txt
|
||||
|
||||
echo "All cleaned up. Test successful!"
|
||||
echo -e "\nTest complete! All tests passed."
|
||||
|
|
|
|||
Loading…
Reference in a new issue