Function TestDownloadServiceMethods

cli/cmd/capture/download_test.go:547–621

(t *testing.T)

}

func TestDownloadServiceMethods(t *testing.T) {
	ctx := context.Background()

	// Setup test objects
	objects := []runtime.Object{
		NewLinuxNode("test-node"),
		NewDownloadNamespace("test-namespace"),
	}

	kubeClient := newDownloadKubeClient(objects)
	config := &rest.Config{}
	service := NewDownloadService(kubeClient, config, "test-namespace")

	t.Run("createDownloadPod creates pod correctly", func(t *testing.T) {
		downloadCmd := &DownloadCmd{
			ContainerImage:   "mcr.microsoft.com/azurelinux/busybox:1.36",
			MountPath:        "/host/tmp/captures",
			KeepAliveCommand: []string{"sh", "-c", "echo 'Download pod ready'; sleep 3600"},
		}

		pod, err := service.createDownloadPod(ctx, "test-node", "/tmp/captures", testCapture, downloadCmd)
		if err != nil {
			t.Fatalf("Expected no error, got: %v", err)
		}

		if pod == nil {
			t.Fatal("Expected pod to be created, got nil")
		}

		if pod.Spec.NodeName != "test-node" {
			t.Errorf("Expected NodeName to be 'test-node', got: %s", pod.Spec.NodeName)
		}

		if len(pod.Spec.Containers) != 1 {
			t.Errorf("Expected 1 container, got: %d", len(pod.Spec.Containers))
		}

		container := pod.Spec.Containers[0]
		if container.Image != downloadCmd.ContainerImage {
			t.Errorf("Expected container image %s, got: %s", downloadCmd.ContainerImage, container.Image)
		}
	})

	t.Run("waitForPodReady handles pod states correctly", func(t *testing.T) {
		// This test is limited due to the fake client behavior
		// In a real scenario, we would test timeout and different pod phases
		podName := "test-pod-ready"

		// Create a pod that should be running
		pod := &corev1.Pod{
			ObjectMeta: metav1.ObjectMeta{
				Name:      podName,
				Namespace: service.namespace,
			},
			Status: corev1.PodStatus{
				Phase: corev1.PodRunning,
			},
		}