294 lines
8.1 KiB
Plaintext
294 lines
8.1 KiB
Plaintext
|
# Test Environment Management
|
||
|
|
# Nushell integration for containerized test environments
|
||
|
|
|
||
|
|
use lib_provisioning *
|
||
|
|
|
||
|
|
const DEFAULT_ORCHESTRATOR = "http://localhost:8080"
|
||
|
|
|
||
|
|
# Create test environment
|
||
|
|
export def "test env create" [
|
||
|
|
config: record # Test environment configuration
|
||
|
|
--infra (-i): string # Infrastructure context
|
||
|
|
--auto-start # Auto-start tests after creation
|
||
|
|
--auto-cleanup # Auto-cleanup after completion
|
||
|
|
--orchestrator: string = $DEFAULT_ORCHESTRATOR
|
||
|
|
]: nothing -> record {
|
||
|
|
let request = {
|
||
|
|
config: $config,
|
||
|
|
infra: $infra,
|
||
|
|
auto_start: $auto_start,
|
||
|
|
auto_cleanup: $auto_cleanup
|
||
|
|
}
|
||
|
|
|
||
|
|
let response = (http post $"($orchestrator)/test/environments/create"
|
||
|
|
--content-type "application/json"
|
||
|
|
($request | to json))
|
||
|
|
|
||
|
|
if $response.success {
|
||
|
|
_print $"✅ Test environment created: ($response.environment_id)"
|
||
|
|
$response
|
||
|
|
} else {
|
||
|
|
_print $"🛑 Failed to create environment: ($response.message)"
|
||
|
|
$response
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# Create single taskserv test environment
|
||
|
|
export def "test env single" [
|
||
|
|
taskserv: string # Taskserv name
|
||
|
|
--base-image: string = "ubuntu:22.04"
|
||
|
|
--cpu: int = 1000 # CPU millicores
|
||
|
|
--memory: int = 2048 # Memory MB
|
||
|
|
--infra (-i): string
|
||
|
|
--auto-start
|
||
|
|
--auto-cleanup
|
||
|
|
]: nothing -> record {
|
||
|
|
let config = {
|
||
|
|
type: "single_taskserv",
|
||
|
|
taskserv: $taskserv,
|
||
|
|
base_image: $base_image,
|
||
|
|
environment: {},
|
||
|
|
resources: {
|
||
|
|
cpu_millicores: $cpu,
|
||
|
|
memory_mb: $memory,
|
||
|
|
disk_gb: null
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
test env create $config --infra $infra --auto-start=$auto_start --auto-cleanup=$auto_cleanup
|
||
|
|
}
|
||
|
|
|
||
|
|
# Create server simulation test environment
|
||
|
|
export def "test env server" [
|
||
|
|
server_name: string # Server name
|
||
|
|
taskservs: list<string> # List of taskservs
|
||
|
|
--base-image: string = "ubuntu:22.04"
|
||
|
|
--cpu: int = 2000
|
||
|
|
--memory: int = 4096
|
||
|
|
--infra (-i): string
|
||
|
|
--auto-start
|
||
|
|
--auto-cleanup
|
||
|
|
]: nothing -> record {
|
||
|
|
let config = {
|
||
|
|
type: "server_simulation",
|
||
|
|
server_name: $server_name,
|
||
|
|
taskservs: $taskservs,
|
||
|
|
base_image: $base_image,
|
||
|
|
auto_resolve_dependencies: true,
|
||
|
|
environment: {},
|
||
|
|
resources: {
|
||
|
|
cpu_millicores: $cpu,
|
||
|
|
memory_mb: $memory,
|
||
|
|
disk_gb: null
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
test env create $config --infra $infra --auto-start=$auto_start --auto-cleanup=$auto_cleanup
|
||
|
|
}
|
||
|
|
|
||
|
|
# Create cluster topology test environment
|
||
|
|
export def "test env cluster" [
|
||
|
|
cluster_type: string # Cluster type (kubernetes, etcd)
|
||
|
|
topology: record # Topology configuration
|
||
|
|
--infra (-i): string
|
||
|
|
--auto-start
|
||
|
|
--auto-cleanup
|
||
|
|
]: nothing -> record {
|
||
|
|
let config = {
|
||
|
|
type: "cluster_topology",
|
||
|
|
...$topology
|
||
|
|
}
|
||
|
|
|
||
|
|
test env create $config --infra $infra --auto-start=$auto_start --auto-cleanup=$auto_cleanup
|
||
|
|
}
|
||
|
|
|
||
|
|
# List test environments
|
||
|
|
export def "test env list" [
|
||
|
|
--orchestrator: string = $DEFAULT_ORCHESTRATOR
|
||
|
|
]: nothing -> table {
|
||
|
|
let response = (http get $"($orchestrator)/test/environments")
|
||
|
|
|
||
|
|
if $response.success {
|
||
|
|
$response.data | select id env_type status created_at containers
|
||
|
|
} else {
|
||
|
|
[]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# Get test environment details
|
||
|
|
export def "test env get" [
|
||
|
|
env_id: string
|
||
|
|
--orchestrator: string = $DEFAULT_ORCHESTRATOR
|
||
|
|
]: nothing -> record {
|
||
|
|
let response = (http get $"($orchestrator)/test/environments/($env_id)")
|
||
|
|
|
||
|
|
if $response.success {
|
||
|
|
$response.data
|
||
|
|
} else {
|
||
|
|
error make {msg: $"Environment ($env_id) not found"}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# Run tests in environment
|
||
|
|
export def "test env run" [
|
||
|
|
env_id: string
|
||
|
|
--tests: list<string> = []
|
||
|
|
--timeout: int
|
||
|
|
--orchestrator: string = $DEFAULT_ORCHESTRATOR
|
||
|
|
]: nothing -> table {
|
||
|
|
let request = {
|
||
|
|
tests: $tests,
|
||
|
|
timeout_seconds: $timeout
|
||
|
|
}
|
||
|
|
|
||
|
|
let response = (http post $"($orchestrator)/test/environments/($env_id)/run"
|
||
|
|
--content-type "application/json"
|
||
|
|
($request | to json))
|
||
|
|
|
||
|
|
if $response.success {
|
||
|
|
_print $"\n(_ansi cyan_bold)Test Results(_ansi reset)"
|
||
|
|
|
||
|
|
let results = $response.data
|
||
|
|
let passed = ($results | where passed == true | length)
|
||
|
|
let total = ($results | length)
|
||
|
|
|
||
|
|
for result in $results {
|
||
|
|
let icon = if $result.passed {
|
||
|
|
$"(_ansi green)✓(_ansi reset)"
|
||
|
|
} else {
|
||
|
|
$"(_ansi red)✗(_ansi reset)"
|
||
|
|
}
|
||
|
|
_print $"($icon) ($result.name): ($result.duration_ms)ms"
|
||
|
|
if not $result.passed and ($result.error | is-not-empty) {
|
||
|
|
_print $" Error: ($result.error)"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
_print $"\n($passed)/($total) tests passed"
|
||
|
|
$results
|
||
|
|
} else {
|
||
|
|
error make {msg: "Failed to run tests"}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# Get environment logs
|
||
|
|
export def "test env logs" [
|
||
|
|
env_id: string
|
||
|
|
--orchestrator: string = $DEFAULT_ORCHESTRATOR
|
||
|
|
]: nothing -> list<string> {
|
||
|
|
let response = (http get $"($orchestrator)/test/environments/($env_id)/logs")
|
||
|
|
|
||
|
|
if $response.success {
|
||
|
|
$response.data
|
||
|
|
} else {
|
||
|
|
[]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# Cleanup test environment
|
||
|
|
export def "test env cleanup" [
|
||
|
|
env_id: string
|
||
|
|
--orchestrator: string = $DEFAULT_ORCHESTRATOR
|
||
|
|
]: nothing -> nothing {
|
||
|
|
let response = (http delete $"($orchestrator)/test/environments/($env_id)")
|
||
|
|
|
||
|
|
if $response.success {
|
||
|
|
_print $"✅ Environment ($env_id) cleaned up"
|
||
|
|
} else {
|
||
|
|
_print $"🛑 Failed to cleanup environment"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# Show environment status
|
||
|
|
export def "test env status" [
|
||
|
|
env_id: string
|
||
|
|
--orchestrator: string = $DEFAULT_ORCHESTRATOR
|
||
|
|
]: nothing -> nothing {
|
||
|
|
let env = (test env get $env_id --orchestrator $orchestrator)
|
||
|
|
|
||
|
|
_print $"\n(_ansi cyan_bold)Test Environment Status(_ansi reset)"
|
||
|
|
_print $"ID: ($env.id)"
|
||
|
|
_print $"Type: ($env.env_type)"
|
||
|
|
_print $"Status: ($env.status)"
|
||
|
|
_print $"Created: ($env.created_at)"
|
||
|
|
|
||
|
|
_print $"\n(_ansi cyan)Containers:(_ansi reset)"
|
||
|
|
for container in $env.containers {
|
||
|
|
_print $" • ($container.container_name) - ($container.status)"
|
||
|
|
if ($container.ip_address | is-not-empty) {
|
||
|
|
_print $" IP: ($container.ip_address)"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ($env.test_results | length) > 0 {
|
||
|
|
_print $"\n(_ansi cyan)Test Results:(_ansi reset)"
|
||
|
|
let passed = ($env.test_results | where passed == true | length)
|
||
|
|
let total = ($env.test_results | length)
|
||
|
|
_print $" ($passed)/($total) tests passed"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# Load topology template
|
||
|
|
export def "test topology load" [
|
||
|
|
template_name: string
|
||
|
|
]: nothing -> record {
|
||
|
|
let config_path = $"($env.PROVISIONING_PATH?)/config/test-topologies.toml"
|
||
|
|
|
||
|
|
if not ($config_path | path exists) {
|
||
|
|
error make {msg: $"Topology templates not found: ($config_path)"}
|
||
|
|
}
|
||
|
|
|
||
|
|
let topologies = (open $config_path)
|
||
|
|
|
||
|
|
if $template_name in ($topologies | columns) {
|
||
|
|
$topologies | get $template_name
|
||
|
|
} else {
|
||
|
|
error make {msg: $"Topology template ($template_name) not found"}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# List available topology templates
|
||
|
|
export def "test topology list" []: nothing -> table {
|
||
|
|
let config_path = $"($env.PROVISIONING_PATH?)/config/test-topologies.toml"
|
||
|
|
|
||
|
|
if not ($config_path | path exists) {
|
||
|
|
return []
|
||
|
|
}
|
||
|
|
|
||
|
|
let topologies = (open $config_path)
|
||
|
|
|
||
|
|
$topologies | transpose name config | select name
|
||
|
|
}
|
||
|
|
|
||
|
|
# Quick test - create, run, and cleanup
|
||
|
|
export def "test quick" [
|
||
|
|
taskserv: string
|
||
|
|
--infra (-i): string
|
||
|
|
]: nothing -> nothing {
|
||
|
|
_print $"🧪 Quick test for ($taskserv)"
|
||
|
|
|
||
|
|
let env_response = (test env single $taskserv --infra $infra --auto-start)
|
||
|
|
|
||
|
|
if not $env_response.success {
|
||
|
|
_print $"🛑 Failed to create environment"
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
let env_id = $env_response.environment_id
|
||
|
|
|
||
|
|
_print $"⏳ Waiting for environment to be ready..."
|
||
|
|
sleep 5sec
|
||
|
|
|
||
|
|
let status = (test env get $env_id)
|
||
|
|
|
||
|
|
if $status.status == "ready" or $status.status == "running" {
|
||
|
|
_print $"✅ Environment ready, running tests..."
|
||
|
|
test env run $env_id
|
||
|
|
}
|
||
|
|
|
||
|
|
_print $"🧹 Cleaning up..."
|
||
|
|
test env cleanup $env_id
|
||
|
|
|
||
|
|
_print $"✅ Quick test completed"
|
||
|
|
}
|