summaryrefslogtreecommitdiff
path: root/integrationtests/polling_test.go
blob: d6b520ca8bfda63ea77b99b9ab94bb16179e5513 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package integrationtests

import (
	"strings"
	"testing"
)

const (
	pollingParquetDuration    = 10
	pollingWorkloadStartupEnv = "IOR_WORKLOAD_STARTUP_DELAY_MS=1000"
)

var pollingTraceArgs = []string{"-trace-syscalls", "epoll_ctl,epoll_wait,epoll_pwait,epoll_pwait2,poll,ppoll,select,pselect6"}

func TestPollingEpollTracepoints(t *testing.T) {
	h := newTestHarness(t)
	h.WorkloadEnv = []string{pollingWorkloadStartupEnv}
	result, pid, err := h.RunWithIorArgs("polling-epoll", defaultDuration, pollingTraceArgs)
	if err != nil {
		t.Fatalf("run scenario polling-epoll: %v", err)
	}

	AssertNoUnexpectedPID(t, result, pid)
	AssertNoUnexpectedComm(t, result, "ioworkload")
	AssertEventsPresent(t, result, []ExpectedEvent{
		{Tracepoint: "enter_epoll_ctl", Comm: "ioworkload", MinCount: 1},
		{Tracepoint: "enter_epoll_wait", Comm: "ioworkload", MinCount: 1},
		{Tracepoint: "enter_epoll_pwait", Comm: "ioworkload", MinCount: 1},
		{Tracepoint: "enter_poll", Comm: "ioworkload", MinCount: 1},
		{Tracepoint: "enter_ppoll", Comm: "ioworkload", MinCount: 1},
		{Tracepoint: "enter_select", Comm: "ioworkload", MinCount: 1},
		{Tracepoint: "enter_pselect6", Comm: "ioworkload", MinCount: 1},
	})

	if !hasTracepoint(result, "enter_epoll_pwait2") {
		t.Log("enter_epoll_pwait2 not observed; treating as unsupported-kernel path")
	}
}

func TestPollingEpollReadyCountInParquet(t *testing.T) {
	h := newTestHarness(t)
	h.WorkloadEnv = []string{pollingWorkloadStartupEnv}
	path, pid, err := h.RunParquetWithIorArgs("polling-epoll", pollingParquetDuration, pollingTraceArgs)
	if err != nil {
		t.Fatalf("run polling-epoll parquet scenario: %v", err)
	}

	rows := filterRecordsByPID(readParquetRecords(t, path), uint32(pid))
	if len(rows) == 0 {
		t.Fatalf("expected parquet rows for workload PID %d", pid)
	}

	wantReadyCount := map[string]bool{
		"epoll_wait":  false,
		"epoll_pwait": false,
		"poll":        false,
		"ppoll":       false,
		"select":      false,
		"pselect6":    false,
	}
	var sawPwait2 bool
	var sawPwait2ReadyCount bool
	for _, row := range rows {
		switch row.Syscall {
		case "epoll_wait", "epoll_pwait", "poll", "ppoll", "select", "pselect6":
			if row.Ret > 0 {
				wantReadyCount[row.Syscall] = true
			}
			if row.Bytes != 0 {
				t.Fatalf("%s bytes = %d, want 0 for ready-count events", row.Syscall, row.Bytes)
			}
		case "epoll_pwait2":
			sawPwait2 = true
			if row.Ret > 0 {
				sawPwait2ReadyCount = true
			}
			if row.Bytes != 0 {
				t.Fatalf("%s bytes = %d, want 0 for ready-count events", row.Syscall, row.Bytes)
			}
		}
	}

	for syscall, ok := range wantReadyCount {
		if !ok {
			t.Fatalf("expected %s row with positive ready-count ret in parquet output", syscall)
		}
	}
	if sawPwait2 && !sawPwait2ReadyCount {
		t.Fatalf("expected epoll_pwait2 row with positive ready-count ret when epoll_pwait2 is present")
	}
	if !sawPwait2 {
		t.Log("epoll_pwait2 parquet rows not observed; treating as unsupported-kernel path")
	}
}

func hasTracepoint(result TestResult, tracepoint string) bool {
	for _, rec := range result.Records {
		if strings.Contains(rec.TraceID.String(), tracepoint) {
			return true
		}
	}
	return false
}