#!/bin/bash

# Tester for deadlock_demo exercise
# Tests deadlock demonstration and solution

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m'

EXERCISE="deadlock_demo"
PASSED=0
FAILED=0

echo "========================================"
echo "Testing: $EXERCISE"
echo "========================================"

if [ ! -f "./$EXERCISE" ]; then
    echo -e "${RED}✗ Executable ./$EXERCISE not found${NC}"
    echo "Please compile: gcc -Wall -Wextra -Werror -pthread deadlock_demo.c -o deadlock_demo"
    exit 1
fi

# Test 1: Deadlock mode (should hang or timeout)
echo -n "Test 1: Deadlock demonstration (mode 0)... "
timeout 3 ./$EXERCISE 0 > /tmp/deadlock_output.txt 2>&1
EXIT_CODE=$?

if [ $EXIT_CODE -eq 124 ]; then
    # Timeout occurred - this is expected for deadlock
    echo -e "${GREEN}✓ PASSED${NC} (program deadlocked as expected)"
    ((PASSED++))
else
    OUTPUT=$(cat /tmp/deadlock_output.txt 2>/dev/null)
    if echo "$OUTPUT" | grep -qi "deadlock"; then
        echo -e "${GREEN}✓ PASSED${NC} (deadlock detected/explained)"
        ((PASSED++))
    else
        echo -e "${RED}✗ FAILED${NC}"
        echo "Program should demonstrate deadlock (hang) or explain it"
        echo "Exit code: $EXIT_CODE"
        ((FAILED++))
    fi
fi

# Test 2: Solution mode (should complete)
echo -n "Test 2: Deadlock solution (mode 1)... "
timeout 5 ./$EXERCISE 1 > /tmp/deadlock_solution.txt 2>&1
EXIT_CODE=$?

if [ $EXIT_CODE -eq 0 ]; then
    echo -e "${GREEN}✓ PASSED${NC} (completed successfully)"
    ((PASSED++))
else
    echo -e "${RED}✗ FAILED${NC}"
    echo "Solution mode should complete without hanging"
    echo "Exit code: $EXIT_CODE"
    ((FAILED++))
fi

# Test 3: Check solution output
echo -n "Test 3: Solution produces output... "
OUTPUT=$(cat /tmp/deadlock_solution.txt 2>/dev/null)
if [ -n "$OUTPUT" ] && [ $(echo "$OUTPUT" | wc -l) -gt 2 ]; then
    echo -e "${GREEN}✓ PASSED${NC}"
    ((PASSED++))
else
    echo -e "${RED}✗ FAILED${NC}"
    echo "Expected meaningful output from solution mode"
    ((FAILED++))
fi

# Test 4: Verify both threads complete in solution mode
echo -n "Test 4: Both threads complete (solution)... "
if echo "$OUTPUT" | grep -qiE "(thread 1|t1)" && \
   echo "$OUTPUT" | grep -qiE "(thread 2|t2)"; then
    echo -e "${GREEN}✓ PASSED${NC}"
    ((PASSED++))
else
    echo -e "${YELLOW}⚠ WARNING${NC}"
    echo "Could not clearly identify both threads in output"
    echo "Output: $OUTPUT"
    ((PASSED++))
fi

# Test 5: Multiple runs of solution (should always work)
echo -n "Test 5: Solution consistency (5 runs)... "
ALL_SUCCESS=true
for i in {1..5}; do
    timeout 5 ./$EXERCISE 1 > /dev/null 2>&1
    if [ $? -ne 0 ]; then
        ALL_SUCCESS=false
        break
    fi
done

if [ "$ALL_SUCCESS" = true ]; then
    echo -e "${GREEN}✓ PASSED${NC}"
    ((PASSED++))
else
    echo -e "${RED}✗ FAILED${NC}"
    echo "Solution mode failed on iteration $i"
    ((FAILED++))
fi

# Test 6: Check for mutex usage
echo -n "Test 6: Mutex implementation check... "
if grep -q "pthread_mutex" deadlock_demo.c 2>/dev/null; then
    MUTEX_COUNT=$(grep -c "pthread_mutex" deadlock_demo.c 2>/dev/null)
    if [ "$MUTEX_COUNT" -ge 4 ]; then
        echo -e "${GREEN}✓ PASSED${NC} (found mutex operations)"
        ((PASSED++))
    else
        echo -e "${YELLOW}⚠ WARNING${NC} (few mutex operations found)"
        ((PASSED++))
    fi
else
    echo -e "${RED}✗ FAILED${NC}"
    echo "Source file should use pthread_mutex functions"
    ((FAILED++))
fi

# Test 7: Invalid argument handling
echo -n "Test 7: Invalid argument handling... "
timeout 2 ./$EXERCISE 2 > /dev/null 2>&1
EXIT_CODE1=$?
timeout 2 ./$EXERCISE abc > /dev/null 2>&1
EXIT_CODE2=$?
timeout 2 ./$EXERCISE > /dev/null 2>&1
EXIT_CODE3=$?

if [ $EXIT_CODE3 -ne 0 ]; then
    echo -e "${GREEN}✓ PASSED${NC} (handles invalid/missing arguments)"
    ((PASSED++))
else
    echo -e "${YELLOW}⚠ WARNING${NC}"
    echo "Program should handle invalid arguments gracefully"
    ((PASSED++))
fi

# Test 8: Memory leak check (solution mode)
if command -v valgrind &> /dev/null; then
    echo -n "Test 8: Memory leak check... "
    VALGRIND_OUTPUT=$(timeout 10 valgrind --leak-check=full --error-exitcode=42 ./$EXERCISE 1 2>&1)
    if [ $? -ne 42 ]; then
        echo -e "${GREEN}✓ PASSED${NC}"
        ((PASSED++))
    else
        echo -e "${RED}✗ FAILED${NC}"
        echo "Memory leaks detected!"
        ((FAILED++))
    fi
else
    echo -e "${YELLOW}⊘ Test 8: Valgrind not installed, skipping${NC}"
fi

# Cleanup
rm -f /tmp/deadlock_output.txt /tmp/deadlock_solution.txt

# Summary
echo "========================================"
echo -e "Results: ${GREEN}$PASSED passed${NC}, ${RED}$FAILED failed${NC}"
echo "========================================"
echo ""
echo "Note: Understanding deadlock is crucial for Philosophers!"
echo "Always lock resources in a consistent order to avoid it."

[ $FAILED -eq 0 ] && exit 0 || exit 1