mirror of
https://github.com/rubenhensen/k8scd.git
synced 2026-09-19 02:52:56 +02:00
Add nix-infra-machine
This commit is contained in:
Ruben Hensen
@@ -0,0 +1,166 @@
|
||||
#!/usr/bin/env bash
|
||||
# OpenSearch standalone test for nix-infra-machine
|
||||
#
|
||||
# This test:
|
||||
# 1. Deploys OpenSearch as a native service on custom port 9201
|
||||
# 2. Verifies the service is running
|
||||
# 3. Tests basic OpenSearch operations (index/query)
|
||||
# 4. Cleans up on teardown
|
||||
|
||||
# Custom ports for testing
|
||||
OPENSEARCH_HTTP_PORT=9201
|
||||
OPENSEARCH_TRANSPORT_PORT=9301
|
||||
|
||||
# Handle teardown command
|
||||
if [ "$CMD" = "teardown" ]; then
|
||||
echo "Tearing down OpenSearch test..."
|
||||
|
||||
# Stop OpenSearch service
|
||||
$NIX_INFRA fleet cmd -d "$WORK_DIR" --target="$TARGET" \
|
||||
'systemctl stop opensearch 2>/dev/null || true'
|
||||
|
||||
# Clean up data directory
|
||||
$NIX_INFRA fleet cmd -d "$WORK_DIR" --target="$TARGET" \
|
||||
'rm -rf /var/lib/opensearch'
|
||||
|
||||
echo "OpenSearch teardown complete"
|
||||
return 0
|
||||
fi
|
||||
|
||||
# ============================================================================
|
||||
# Test Setup
|
||||
# ============================================================================
|
||||
|
||||
_start=$(date +%s)
|
||||
|
||||
echo ""
|
||||
echo "========================================"
|
||||
echo "OpenSearch Standalone Test (port $OPENSEARCH_HTTP_PORT)"
|
||||
echo "========================================"
|
||||
echo ""
|
||||
|
||||
# Deploy the opensearch configuration to test nodes
|
||||
echo "Step 1: Deploying OpenSearch configuration..."
|
||||
$NIX_INFRA fleet deploy-apps -d "$WORK_DIR" --batch --env="$ENV" \
|
||||
--test-dir="$WORK_DIR/$TEST_DIR" \
|
||||
--target="$TARGET"
|
||||
|
||||
# Apply the configuration
|
||||
echo "Step 2: Applying NixOS configuration..."
|
||||
$NIX_INFRA fleet cmd -d "$WORK_DIR" --target="$TARGET" "nixos-rebuild switch --fast"
|
||||
|
||||
_setup=$(date +%s)
|
||||
|
||||
# ============================================================================
|
||||
# Test Verification
|
||||
# ============================================================================
|
||||
|
||||
echo ""
|
||||
echo "Step 3: Verifying OpenSearch deployment..."
|
||||
echo ""
|
||||
|
||||
# Wait for OpenSearch service and API to be ready
|
||||
for node in $TARGET; do
|
||||
wait_for_service "$node" "opensearch" --timeout=60
|
||||
wait_for_port "$node" "$OPENSEARCH_HTTP_PORT" --timeout=30
|
||||
wait_for_elasticsearch "$node" "$OPENSEARCH_HTTP_PORT" --timeout=60 # Same API as Elasticsearch
|
||||
done
|
||||
|
||||
# Check if the systemd service is active
|
||||
echo ""
|
||||
echo "Checking systemd service status..."
|
||||
for node in $TARGET; do
|
||||
assert_service_active "$node" "opensearch" || show_service_logs "$node" "opensearch" 50
|
||||
done
|
||||
|
||||
# Check if OpenSearch process is running
|
||||
echo ""
|
||||
echo "Checking OpenSearch process..."
|
||||
for node in $TARGET; do
|
||||
assert_process_running "$node" "-f opensearch" "OpenSearch"
|
||||
done
|
||||
|
||||
# Check if OpenSearch HTTP port is listening
|
||||
echo ""
|
||||
echo "Checking OpenSearch HTTP port ($OPENSEARCH_HTTP_PORT)..."
|
||||
for node in $TARGET; do
|
||||
assert_port_listening "$node" "$OPENSEARCH_HTTP_PORT" "HTTP port $OPENSEARCH_HTTP_PORT"
|
||||
done
|
||||
|
||||
# ============================================================================
|
||||
# Functional Tests
|
||||
# ============================================================================
|
||||
|
||||
echo ""
|
||||
echo "Step 4: Running functional tests..."
|
||||
echo ""
|
||||
|
||||
# Test OpenSearch connection and basic operations
|
||||
for node in $TARGET; do
|
||||
echo "Testing OpenSearch operations on $node..."
|
||||
|
||||
# Test cluster health endpoint
|
||||
echo " Checking cluster health..."
|
||||
health_result=$(cmd_clean "$node" "curl -s http://127.0.0.1:$OPENSEARCH_HTTP_PORT/_cluster/health")
|
||||
if assert_contains "$health_result" "cluster_name" "Cluster health endpoint accessible"; then
|
||||
status=$(echo "$health_result" | jq -r '.status' 2>/dev/null || echo "unknown")
|
||||
print_info "Cluster status" "$status"
|
||||
fi
|
||||
|
||||
# Create a test index
|
||||
echo " Creating test index..."
|
||||
create_result=$(cmd_clean "$node" "curl -s -X PUT 'http://127.0.0.1:$OPENSEARCH_HTTP_PORT/test-index' -H 'Content-Type: application/json' -d '{\"settings\": {\"number_of_shards\": 1, \"number_of_replicas\": 0}}'")
|
||||
assert_contains_all "$create_result" "Index creation successful" "acknowledged" "true"
|
||||
|
||||
# Insert a test document
|
||||
echo " Inserting test document..."
|
||||
insert_result=$(cmd_clean "$node" "curl -s -X POST 'http://127.0.0.1:$OPENSEARCH_HTTP_PORT/test-index/_doc/1' -H 'Content-Type: application/json' -d '{\"name\": \"test\", \"value\": 42}'")
|
||||
if [[ "$insert_result" == *"created"* ]] || [[ "$insert_result" == *"_id"* ]]; then
|
||||
echo -e " ${GREEN}✓${NC} Document insert successful [pass]"
|
||||
else
|
||||
echo -e " ${RED}✗${NC} Document insert failed: $insert_result [fail]"
|
||||
fi
|
||||
|
||||
# Force refresh to make document searchable
|
||||
cmd "$node" "curl -s -X POST 'http://127.0.0.1:$OPENSEARCH_HTTP_PORT/test-index/_refresh'" > /dev/null 2>&1
|
||||
|
||||
# Query the test document
|
||||
echo " Querying test document..."
|
||||
query_result=$(cmd_clean "$node" "curl -s 'http://127.0.0.1:$OPENSEARCH_HTTP_PORT/test-index/_doc/1'")
|
||||
assert_contains_all "$query_result" "Document query successful" "found" "true"
|
||||
|
||||
# Test search functionality
|
||||
echo " Testing search..."
|
||||
search_result=$(cmd_clean "$node" "curl -s -X GET 'http://127.0.0.1:$OPENSEARCH_HTTP_PORT/test-index/_search' -H 'Content-Type: application/json' -d '{\"query\": {\"match\": {\"name\": \"test\"}}}'")
|
||||
assert_contains_all "$search_result" "Search operation successful" "hits" "value"
|
||||
|
||||
# List indices
|
||||
echo " Listing indices..."
|
||||
indices_result=$(cmd_clean "$node" "curl -s 'http://127.0.0.1:$OPENSEARCH_HTTP_PORT/_cat/indices?v'")
|
||||
assert_contains "$indices_result" "test-index" "Index listing successful"
|
||||
|
||||
# Clean up test index
|
||||
echo " Cleaning up test index..."
|
||||
cmd "$node" "curl -s -X DELETE 'http://127.0.0.1:$OPENSEARCH_HTTP_PORT/test-index'" > /dev/null 2>&1
|
||||
print_cleanup "Test index cleaned up"
|
||||
done
|
||||
|
||||
# ============================================================================
|
||||
# Test Summary
|
||||
# ============================================================================
|
||||
|
||||
_end=$(date +%s)
|
||||
|
||||
echo ""
|
||||
echo "========================================"
|
||||
echo "OpenSearch Test Summary"
|
||||
echo "========================================"
|
||||
|
||||
printf '+ setup %s\n' $(printTime $_start $_setup)
|
||||
printf '+ tests %s\n' $(printTime $_setup $_end)
|
||||
printf '= TOTAL %s\n' $(printTime $_start $_end)
|
||||
|
||||
echo ""
|
||||
echo "========================================"
|
||||
echo "OpenSearch Test Complete"
|
||||
echo "========================================"
|
||||
Reference in New Issue
Block a user