LCBro

#!/bin/bash # Test script for remote CDP server functionality # Tests connection to remote CDP servers and validates API responses set -e # Configuration DEFAULT_TIMEOUT=10 DEFAULT_SSL_MODE="auto" VERBOSE=false # Colors for output RED='\033[0;31m' GREEN='\033[0;32m' YELLOW='\033[1;33m' BLUE='\033[0;34m' NC='\033[0m' # No Color # Logging functions log_info() { echo -e "${BLUE}[INFO]${NC} $1" } log_success() { echo -e "${GREEN}[SUCCESS]${NC} $1" } log_warning() { echo -e "${YELLOW}[WARNING]${NC} $1" } log_error() { echo -e "${RED}[ERROR]${NC} $1" } # Help function show_help() { cat << EOF Remote CDP Server Test Script Usage: $0 [OPTIONS] <URL> Options: -t, --timeout SECONDS Request timeout in seconds [default: $DEFAULT_TIMEOUT] -s, --ssl-mode MODE SSL mode: auto, enabled, disabled, insecure [default: $DEFAULT_SSL_MODE] -k, --api-key KEY API key for authentication -h, --header HEADER Additional HTTP header (format: "Name: Value") -v, --verbose Verbose output --help Show this help Examples: $0 https://cdp.example.com:9222 $0 -k "secret-key" https://cdp.example.com:9222 $0 -s insecure https://localhost:9222 $0 -h "X-Client-ID: test" https://cdp.example.com:9222 EOF } # Test remote CDP server test_remote_cdp() { local url="$1" local timeout="$2" local ssl_mode="$3" local api_key="$4" local headers=("${@:5}") log_info "Testing remote CDP server: $url" log_info "Timeout: ${timeout}s, SSL Mode: $ssl_mode" # Prepare headers array local header_args=() for header in "${headers[@]}"; do header_args+=("-H" "$header") done # Add API key if provided if [ -n "$api_key" ]; then header_args+=("-H" "Authorization: Bearer $api_key") header_args+=("-H" "X-API-Key: $api_key") log_info "Using API key: ${api_key:0:8}..." fi # Test server health test_health "$url" "$timeout" "${header_args[@]}" # Test server info test_server_info "$url" "$timeout" "${header_args[@]}" # Test browsers list test_browsers_list "$url" "$timeout" "${header_args[@]}" } # Test server health endpoint test_health() { local url="$1" local timeout="$2" shift 2 local header_args=("$@") log_info "Testing health endpoint..." local health_url="${url%/}/api/health" if [ "$VERBOSE" = true ]; then log_info "Health URL: $health_url" log_info "Headers: ${header_args[*]}" fi local response local status_code if response=$(curl -s -w "\n%{http_code}" --max-time "$timeout" "${header_args[@]}" "$health_url" 2>/dev/null); then status_code=$(echo "$response" | tail -n1) response_body=$(echo "$response" | head -n -1) if [ "$status_code" -eq 200 ]; then log_success "Health check passed (HTTP $status_code)" if [ "$VERBOSE" = true ]; then echo "$response_body" | jq . 2>/dev/null || echo "$response_body" fi else log_warning "Health check failed (HTTP $status_code)" if [ "$VERBOSE" = true ]; then echo "$response_body" fi fi else log_error "Health check failed: connection error" return 1 fi } # Test server info endpoint test_server_info() { local url="$1" local timeout="$2" shift 2 local header_args=("$@") log_info "Testing server info endpoint..." local info_url="${url%/}/api/info" if [ "$VERBOSE" = true ]; then log_info "Info URL: $info_url" fi local response local status_code if response=$(curl -s -w "\n%{http_code}" --max-time "$timeout" "${header_args[@]}" "$info_url" 2>/dev/null); then status_code=$(echo "$response" | tail -n1) response_body=$(echo "$response" | head -n -1) if [ "$status_code" -eq 200 ]; then log_success "Server info retrieved (HTTP $status_code)" if command -v jq >/dev/null 2>&1; then echo "$response_body" | jq . else echo "$response_body" fi else log_warning "Server info failed (HTTP $status_code)" if [ "$VERBOSE" = true ]; then echo "$response_body" fi fi else log_error "Server info failed: connection error" return 1 fi } # Test browsers list endpoint test_browsers_list() { local url="$1" local timeout="$2" shift 2 local header_args=("$@") log_info "Testing browsers list endpoint..." local browsers_url="${url%/}/api/browsers" if [ "$VERBOSE" = true ]; then log_info "Browsers URL: $browsers_url" fi local response local status_code if response=$(curl -s -w "\n%{http_code}" --max-time "$timeout" "${header_args[@]}" "$browsers_url" 2>/dev/null); then status_code=$(echo "$response" | tail -n1) response_body=$(echo "$response" | head -n -1) if [ "$status_code" -eq 200 ]; then log_success "Browsers list retrieved (HTTP $status_code)" if command -v jq >/dev/null 2>&1; then local browser_count browser_count=$(echo "$response_body" | jq '.browsers | length' 2>/dev/null || echo "0") log_info "Found $browser_count browsers" echo "$response_body" | jq '.browsers[] | {id, title, type, url}' 2>/dev/null || echo "$response_body" else echo "$response_body" fi else log_warning "Browsers list failed (HTTP $status_code)" if [ "$VERBOSE" = true ]; then echo "$response_body" fi fi else log_error "Browsers list failed: connection error" return 1 fi } # Validate URL validate_url() { local url="$1" if [[ ! "$url" =~ ^https?:// ]]; then log_error "URL must start with http:// or https://" return 1 fi # Basic URL validation if ! curl -s --head --max-time 5 "$url" >/dev/null 2>&1; then log_warning "URL may not be reachable: $url" fi return 0 } # Main function main() { local url="" local timeout="$DEFAULT_TIMEOUT" local ssl_mode="$DEFAULT_SSL_MODE" local api_key="" local headers=() # Parse command line arguments while [[ $# -gt 0 ]]; do case $1 in -t|--timeout) timeout="$2" shift 2 ;; -s|--ssl-mode) ssl_mode="$2" shift 2 ;; -k|--api-key) api_key="$2" shift 2 ;; -h|--header) headers+=("$2") shift 2 ;; -v|--verbose) VERBOSE=true shift ;; --help) show_help exit 0 ;; -*) log_error "Unknown option: $1" show_help exit 1 ;; *) if [ -z "$url" ]; then url="$1" else log_error "Multiple URLs provided" exit 1 fi shift ;; esac done # Check if URL is provided if [ -z "$url" ]; then log_error "URL is required" show_help exit 1 fi # Validate URL if ! validate_url "$url"; then exit 1 fi # Check dependencies if ! command -v curl >/dev/null 2>&1; then log_error "curl is required but not installed" exit 1 fi # Display configuration echo log_info "Remote CDP Server Test" log_info "======================" log_info "URL: $url" log_info "Timeout: ${timeout}s" log_info "SSL Mode: $ssl_mode" if [ -n "$api_key" ]; then log_info "API Key: ${api_key:0:8}..." fi if [ ${#headers[@]} -gt 0 ]; then log_info "Headers: ${#headers[@]} custom headers" fi echo # Run tests if test_remote_cdp "$url" "$timeout" "$ssl_mode" "$api_key" "${headers[@]}"; then echo log_success "All tests completed successfully" exit 0 else echo log_error "Some tests failed" exit 1 fi } # Run main function main "$@"