Function TestGenerateNfqueueDropProbe

shell/tracescript_test.go:509–554

(t *testing.T)

}

func TestGenerateNfqueueDropProbe(t *testing.T) {
	config := TraceConfig{
		EnableNfqueueDrops: true,
	}

	gen := NewScriptGenerator(config)
	script := gen.Generate()

	// Verify fexit probe is present
	if !strings.Contains(script, "fexit:vmlinux:__nf_queue") {
		t.Error("script missing fexit:vmlinux:__nf_queue probe")
	}

	// Verify it checks return value
	if !strings.Contains(script, "retval") {
		t.Error("script missing retval check")
	}

	// Verify it accesses args->skb
	if !strings.Contains(script, "args->skb") {
		t.Error("script missing args->skb access")
	}

	// Verify it reads queue number
	if !strings.Contains(script, "args->queuenum") {
		t.Error("script missing args->queuenum access")
	}

	// Verify NFQ_DROP event type in table output
	if !strings.Contains(script, "NFQ_DROP") {
		t.Error("script missing NFQ_DROP event type")
	}

	// Verify __nf_queue probe name in output
	if !strings.Contains(script, "__nf_queue") {
		t.Error("script missing __nf_queue probe name in output")
	}

	// Verify errno decoding
	if !strings.Contains(script, "ESRCH") {
		t.Error("script missing ESRCH errno name")
	}
	if !strings.Contains(script, "ENOMEM") {
		t.Error("script missing ENOMEM errno name")
	}
}

func TestGenerateNfqueueDropProbeJSON(t *testing.T) {
	config := TraceConfig{