Function TestHostNetworkPodForTrace

shell/trace_test.go:63–141

(t *testing.T)

}

func TestHostNetworkPodForTrace(t *testing.T) {
	config := TraceConfig{
		RetinaShellImage: "mcr.microsoft.com/containernetworking/retina-shell:v1.0.0",
	}

	pod := hostNetworkPodForTrace(config, "kube-system", "node-001")

	// Test basic pod properties
	t.Run("namespace", func(t *testing.T) {
		if pod.Namespace != "kube-system" {
			t.Errorf("Expected namespace kube-system, got %s", pod.Namespace)
		}
	})

	t.Run("node selector", func(t *testing.T) {
		if pod.Spec.NodeName != "node-001" {
			t.Errorf("Expected nodeName node-001, got %s", pod.Spec.NodeName)
		}
	})

	t.Run("host network enabled", func(t *testing.T) {
		if !pod.Spec.HostNetwork {
			t.Error("Expected HostNetwork to be true")
		}
	})

	t.Run("host PID enabled", func(t *testing.T) {
		if !pod.Spec.HostPID {
			t.Error("Expected HostPID to be true for bpftrace")
		}
	})

	t.Run("tolerates all taints", func(t *testing.T) {
		found := false
		for _, tol := range pod.Spec.Tolerations {
			if tol.Operator == v1.TolerationOpExists {
				found = true
				break
			}
		}
		if !found {
			t.Error("Expected toleration with Operator=Exists to run on any node")
		}
	})

	t.Run("restart policy never", func(t *testing.T) {
		if pod.Spec.RestartPolicy != v1.RestartPolicyNever {
			t.Errorf("Expected RestartPolicy Never, got %s", pod.Spec.RestartPolicy)
		}
	})

	t.Run("image set correctly", func(t *testing.T) {
		if len(pod.Spec.Containers) != 1 {
			t.Fatalf("Expected 1 container, got %d", len(pod.Spec.Containers))
		}
		if pod.Spec.Containers[0].Image != config.RetinaShellImage {
			t.Errorf("Expected image %s, got %s", config.RetinaShellImage, pod.Spec.Containers[0].Image)
		}