#!/usr/bin/env bash

set -euo pipefail

# Comprehensive test runner for Bifrost PR validation

# This script runs all test suites to validate changes

echo "🧪 Starting Bifrost Test Suite..."

echo "=================================="

# Color codes for output

RED='\033[0;31m'

GREEN='\033[0;32m'

YELLOW='\033[1;33m'

NC='\033[0m' # No Color

# Track test results

TESTS_PASSED=0

TESTS_FAILED=0

# Function to report test result

report_result() {

local test_name=$1

local result=$2

if [ "$result" -eq 0 ]; then

echo -e "${GREEN}✅ $test_name passed${NC}"

((TESTS_PASSED++))

else

echo -e "${RED}❌ $test_name failed${NC}"

((TESTS_FAILED++))

fi

}

# 1. Core Build Validation

echo ""

echo "📦 1/5 - Validating Core Build..."

echo "-----------------------------------"

cd core

if go mod download && go build ./...; then

report_result "Core Build" 0

else

report_result "Core Build" 1

fi

cd ..

# 2. Build MCP Test Servers

echo ""

echo "🔌 2/5 - Building MCP Test Servers..."

echo "-----------------------------------"

MCP_BUILD_FAILED=0

for mcp_dir in examples/mcps/*/; do

if [ -d "$mcp_dir" ]; then

mcp_name=$(basename "$mcp_dir")

if [ -f "$mcp_dir/go.mod" ]; then

echo " Building $mcp_name (Go)..."

mkdir -p "$mcp_dir/bin"

if cd "$mcp_dir" && GOWORK=off go build -o "bin/$mcp_name" . && cd - > /dev/null; then

echo -e " ${GREEN}✓ $mcp_name${NC}"

else

echo -e " ${RED}✗ $mcp_name${NC}"

MCP_BUILD_FAILED=1

cd - > /dev/null 2>&1 || true

fi

elif [ -f "$mcp_dir/package.json" ]; then

echo " Building $mcp_name (TypeScript)..."

if cd "$mcp_dir" && npm install --silent && npm run build && cd - > /dev/null; then

echo -e " ${GREEN}✓ $mcp_name${NC}"

else

echo -e " ${RED}✗ $mcp_name${NC}"

MCP_BUILD_FAILED=1

cd - > /dev/null 2>&1 || true

fi

fi

fi

done

report_result "MCP Test Servers Build" $MCP_BUILD_FAILED

# 3. Core Provider Tests

echo ""

echo "🔧 3/5 - Running Core Provider Tests..."

echo "-----------------------------------"

cd core

if go test -v -run . ./...; then

report_result "Core Provider Tests" 0

else

report_result "Core Provider Tests" 1

fi

cd ..

# 4. Governance Tests

echo ""

echo "🛡️ 4/5 - Running Governance Tests..."

echo "-----------------------------------"

if [ -d "tests/governance" ]; then

cd tests/governance

# Check if virtual environment exists, create if not

if [ ! -d "venv" ]; then

echo "Creating Python virtual environment..."

python3 -m venv venv

fi

# Activate virtual environment

source venv/bin/activate

# Install dependencies

echo "Installing Python dependencies..."

pip install -q -r requirements.txt

# Run tests

if pytest -v; then

report_result "Governance Tests" 0

else

report_result "Governance Tests" 1

fi

deactivate

cd ../..

else

echo -e "${YELLOW}⚠️ Governance tests directory not found, skipping...${NC}"

fi

# 5. Integration Tests

echo ""

echo "🔗 5/5 - Running Integration Tests..."

echo "-----------------------------------"

if [ -d "tests/integrations" ]; then

cd tests/integrations

# Check if virtual environment exists, create if not

if [ ! -d "venv" ]; then

echo "Creating Python virtual environment..."

python3 -m venv venv

fi

# Activate virtual environment

source venv/bin/activate

# Install dependencies

echo "Installing Python dependencies..."

pip install -q -r requirements.txt

# Run tests

if python run_all_tests.py; then

report_result "Integration Tests" 0

else

report_result "Integration Tests" 1

fi

deactivate

cd ../..

else

echo -e "${YELLOW}⚠️ Integration tests directory not found, skipping...${NC}"

fi

# Final Summary

echo ""

echo "=================================="

echo "🏁 Test Suite Complete!"

echo "=================================="

echo -e "${GREEN}Passed: $TESTS_PASSED${NC}"

echo -e "${RED}Failed: $TESTS_FAILED${NC}"

echo ""

if [ "$TESTS_FAILED" -gt 0 ]; then

echo -e "${RED}❌ Some tests failed. Please review the output above.${NC}"

exit 1

else

echo -e "${GREEN}✅ All tests passed successfully!${NC}"

exit 0

fi