kubernetes-mcp-server

package kubevirt import ( "context" "strings" "testing" "k8s.io/apimachinery/pkg/apis/meta/v1/unstructured" "k8s.io/apimachinery/pkg/runtime" "k8s.io/client-go/dynamic/fake" ) // createTestVM creates a test VirtualMachine with the given name, namespace, and runStrategy func createTestVM(name, namespace string, runStrategy RunStrategy) *unstructured.Unstructured { vm := &unstructured.Unstructured{} vm.SetUnstructuredContent(map[string]interface{}{ "apiVersion": "kubevirt.io/v1", "kind": "VirtualMachine", "metadata": map[string]interface{}{ "name": name, "namespace": namespace, }, "spec": map[string]interface{}{ "runStrategy": string(runStrategy), }, }) return vm } func TestStartVM(t *testing.T) { tests := []struct { name string initialVM *unstructured.Unstructured wantStarted bool wantError bool errorContains string }{ { name: "Start VM that is Halted", initialVM: createTestVM("test-vm", "default", RunStrategyHalted), wantStarted: true, wantError: false, }, { name: "Start VM that is already running (Always)", initialVM: createTestVM("test-vm", "default", RunStrategyAlways), wantStarted: false, wantError: false, }, { name: "Start VM without runStrategy", initialVM: &unstructured.Unstructured{ Object: map[string]interface{}{ "apiVersion": "kubevirt.io/v1", "kind": "VirtualMachine", "metadata": map[string]interface{}{ "name": "test-vm", "namespace": "default", }, "spec": map[string]interface{}{}, }, }, wantStarted: true, wantError: false, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { scheme := runtime.NewScheme() client := fake.NewSimpleDynamicClient(scheme, tt.initialVM) ctx := context.Background() vm, wasStarted, err := StartVM(ctx, client, tt.initialVM.GetNamespace(), tt.initialVM.GetName()) if tt.wantError { if err == nil { t.Errorf("Expected error, got nil") return } if tt.errorContains != "" && !strings.Contains(err.Error(), tt.errorContains) { t.Errorf("Error = %v, want to contain %q", err, tt.errorContains) } return } if err != nil { t.Errorf("Unexpected error: %v", err) return } if vm == nil { t.Errorf("Expected non-nil VM, got nil") return } if wasStarted != tt.wantStarted { t.Errorf("wasStarted = %v, want %v", wasStarted, tt.wantStarted) } // Verify the VM's runStrategy is Always strategy, found, err := GetVMRunStrategy(vm) if err != nil { t.Errorf("Failed to get runStrategy: %v", err) return } if !found { t.Errorf("runStrategy not found") return } if strategy != RunStrategyAlways { t.Errorf("Strategy = %q, want %q", strategy, RunStrategyAlways) } }) } } func TestStartVMNotFound(t *testing.T) { scheme := runtime.NewScheme() client := fake.NewSimpleDynamicClient(scheme) ctx := context.Background() _, _, err := StartVM(ctx, client, "default", "non-existent-vm") if err == nil { t.Errorf("Expected error for non-existent VM, got nil") return } if !strings.Contains(err.Error(), "failed to get VirtualMachine") { t.Errorf("Error = %v, want to contain 'failed to get VirtualMachine'", err) } } func TestStopVM(t *testing.T) { tests := []struct { name string initialVM *unstructured.Unstructured wantStopped bool wantError bool errorContains string }{ { name: "Stop VM that is running (Always)", initialVM: createTestVM("test-vm", "default", RunStrategyAlways), wantStopped: true, wantError: false, }, { name: "Stop VM that is already stopped (Halted)", initialVM: createTestVM("test-vm", "default", RunStrategyHalted), wantStopped: false, wantError: false, }, { name: "Stop VM without runStrategy", initialVM: &unstructured.Unstructured{ Object: map[string]interface{}{ "apiVersion": "kubevirt.io/v1", "kind": "VirtualMachine", "metadata": map[string]interface{}{ "name": "test-vm", "namespace": "default", }, "spec": map[string]interface{}{}, }, }, wantStopped: true, wantError: false, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { scheme := runtime.NewScheme() client := fake.NewSimpleDynamicClient(scheme, tt.initialVM) ctx := context.Background() vm, wasStopped, err := StopVM(ctx, client, tt.initialVM.GetNamespace(), tt.initialVM.GetName()) if tt.wantError { if err == nil { t.Errorf("Expected error, got nil") return } if tt.errorContains != "" && !strings.Contains(err.Error(), tt.errorContains) { t.Errorf("Error = %v, want to contain %q", err, tt.errorContains) } return } if err != nil { t.Errorf("Unexpected error: %v", err) return } if vm == nil { t.Errorf("Expected non-nil VM, got nil") return } if wasStopped != tt.wantStopped { t.Errorf("wasStopped = %v, want %v", wasStopped, tt.wantStopped) } // Verify the VM's runStrategy is Halted strategy, found, err := GetVMRunStrategy(vm) if err != nil { t.Errorf("Failed to get runStrategy: %v", err) return } if !found { t.Errorf("runStrategy not found") return } if strategy != RunStrategyHalted { t.Errorf("Strategy = %q, want %q", strategy, RunStrategyHalted) } }) } } func TestStopVMNotFound(t *testing.T) { scheme := runtime.NewScheme() client := fake.NewSimpleDynamicClient(scheme) ctx := context.Background() _, _, err := StopVM(ctx, client, "default", "non-existent-vm") if err == nil { t.Errorf("Expected error for non-existent VM, got nil") return } if !strings.Contains(err.Error(), "failed to get VirtualMachine") { t.Errorf("Error = %v, want to contain 'failed to get VirtualMachine'", err) } } func TestRestartVM(t *testing.T) { tests := []struct { name string initialVM *unstructured.Unstructured wantError bool errorContains string }{ { name: "Restart VM that is running (Always)", initialVM: createTestVM("test-vm", "default", RunStrategyAlways), wantError: false, }, { name: "Restart VM that is stopped (Halted)", initialVM: createTestVM("test-vm", "default", RunStrategyHalted), wantError: false, }, { name: "Restart VM without runStrategy", initialVM: &unstructured.Unstructured{ Object: map[string]interface{}{ "apiVersion": "kubevirt.io/v1", "kind": "VirtualMachine", "metadata": map[string]interface{}{ "name": "test-vm", "namespace": "default", }, "spec": map[string]interface{}{}, }, }, wantError: false, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { scheme := runtime.NewScheme() client := fake.NewSimpleDynamicClient(scheme, tt.initialVM) ctx := context.Background() vm, err := RestartVM(ctx, client, tt.initialVM.GetNamespace(), tt.initialVM.GetName()) if tt.wantError { if err == nil { t.Errorf("Expected error, got nil") return } if tt.errorContains != "" && !strings.Contains(err.Error(), tt.errorContains) { t.Errorf("Error = %v, want to contain %q", err, tt.errorContains) } return } if err != nil { t.Errorf("Unexpected error: %v", err) return } if vm == nil { t.Errorf("Expected non-nil VM, got nil") return } // Verify the VM's runStrategy is Always (after restart) strategy, found, err := GetVMRunStrategy(vm) if err != nil { t.Errorf("Failed to get runStrategy: %v", err) return } if !found { t.Errorf("runStrategy not found") return } if strategy != RunStrategyAlways { t.Errorf("Strategy = %q, want %q after restart", strategy, RunStrategyAlways) } }) } } func TestRestartVMNotFound(t *testing.T) { scheme := runtime.NewScheme() client := fake.NewSimpleDynamicClient(scheme) ctx := context.Background() _, err := RestartVM(ctx, client, "default", "non-existent-vm") if err == nil { t.Errorf("Expected error for non-existent VM, got nil") return } if !strings.Contains(err.Error(), "failed to get VirtualMachine") { t.Errorf("Error = %v, want to contain 'failed to get VirtualMachine'", err) } }