Add tests

run-tests.sh

@@ -0,0 +1,126 @@
+#!/bin/bash
+
+# Test runner script for LatticeLM Gateway
+# Usage: ./run-tests.sh [option]
+#
+# Options:
+#   all       - Run all tests (default)
+#   coverage  - Run tests with coverage report
+#   verbose   - Run tests with verbose output
+#   config    - Run config tests only
+#   providers - Run provider tests only
+#   conv      - Run conversation tests only
+#   watch     - Watch mode (requires entr)
+
+set -e
+
+COLOR_GREEN='\033[0;32m'
+COLOR_BLUE='\033[0;34m'
+COLOR_YELLOW='\033[1;33m'
+COLOR_RED='\033[0;31m'
+COLOR_RESET='\033[0m'
+
+print_header() {
+    echo -e "${COLOR_BLUE}========================================${COLOR_RESET}"
+    echo -e "${COLOR_BLUE}$1${COLOR_RESET}"
+    echo -e "${COLOR_BLUE}========================================${COLOR_RESET}"
+}
+
+print_success() {
+    echo -e "${COLOR_GREEN}✓ $1${COLOR_RESET}"
+}
+
+print_error() {
+    echo -e "${COLOR_RED}✗ $1${COLOR_RESET}"
+}
+
+print_info() {
+    echo -e "${COLOR_YELLOW}ℹ $1${COLOR_RESET}"
+}
+
+run_all_tests() {
+    print_header "Running All Tests"
+    go test ./internal/... || exit 1
+    print_success "All tests passed!"
+}
+
+run_verbose_tests() {
+    print_header "Running Tests (Verbose)"
+    go test ./internal/... -v || exit 1
+    print_success "All tests passed!"
+}
+
+run_coverage_tests() {
+    print_header "Running Tests with Coverage"
+    go test ./internal/... -cover -coverprofile=coverage.out || exit 1
+    print_success "Tests passed! Generating HTML report..."
+    go tool cover -html=coverage.out -o coverage.html
+    print_success "Coverage report generated: coverage.html"
+    print_info "Open coverage.html in your browser to view detailed coverage"
+}
+
+run_config_tests() {
+    print_header "Running Config Tests"
+    go test ./internal/config -v -cover || exit 1
+    print_success "Config tests passed!"
+}
+
+run_provider_tests() {
+    print_header "Running Provider Tests"
+    go test ./internal/providers/... -v -cover || exit 1
+    print_success "Provider tests passed!"
+}
+
+run_conversation_tests() {
+    print_header "Running Conversation Tests"
+    go test ./internal/conversation -v -cover || exit 1
+    print_success "Conversation tests passed!"
+}
+
+run_watch_mode() {
+    if ! command -v entr &> /dev/null; then
+        print_error "entr is not installed. Install it with: apt-get install entr"
+        exit 1
+    fi
+    print_header "Running Tests in Watch Mode"
+    print_info "Watching for file changes... (Press Ctrl+C to stop)"
+    find ./internal -name '*.go' | entr -c sh -c 'go test ./internal/... || true'
+}
+
+# Main script
+case "${1:-all}" in
+    all)
+        run_all_tests
+        ;;
+    coverage)
+        run_coverage_tests
+        ;;
+    verbose)
+        run_verbose_tests
+        ;;
+    config)
+        run_config_tests
+        ;;
+    providers)
+        run_provider_tests
+        ;;
+    conv)
+        run_conversation_tests
+        ;;
+    watch)
+        run_watch_mode
+        ;;
+    *)
+        echo "Usage: $0 {all|coverage|verbose|config|providers|conv|watch}"
+        echo ""
+        echo "Options:"
+        echo "  all       - Run all tests (default)"
+        echo "  coverage  - Run tests with coverage report"
+        echo "  verbose   - Run tests with verbose output"
+        echo "  config    - Run config tests only"
+        echo "  providers - Run provider tests only"
+        echo "  conv      - Run conversation tests only"
+        echo "  watch     - Watch mode (requires entr)"
+        exit 1
+        ;;
+esac