task 07: add KindMem and separate address-space byte accounting

1 files changed, 46 insertions, 0 deletions

diff --git a/internal/eventloop_memory_test.go b/internal/eventloop_memory_test.go
new file mode 100644
index 0000000..4ac9580
--- /dev/null
+++ b/internal/eventloop_memory_test.go
@@ -0,0 +1,46 @@
+package internal
+
+import (
+	"testing"
+
+	"ior/internal/event"
+	"ior/internal/globalfilter"
+	"ior/internal/types"
+)
+
+func TestHandleMemExitAppliesPairFilter(t *testing.T) {
+	el := mustNewEventLoop(t, eventLoopConfig{
+		filter: globalfilter.Filter{
+			Syscall: &globalfilter.StringFilter{Pattern: "openat"},
+		},
+	})
+
+	enter := &types.MemEvent{
+		EventType: types.ENTER_MEM_EVENT,
+		TraceId:   types.SYS_ENTER_MUNMAP,
+		Time:      100,
+		Pid:       91,
+		Tid:       92,
+		Length:    4096,
+	}
+	exit := &types.RetEvent{
+		EventType: types.EXIT_RET_EVENT,
+		TraceId:   types.SYS_EXIT_MUNMAP,
+		Time:      200,
+		Pid:       91,
+		Tid:       92,
+		Ret:       0,
+	}
+	ep := &event.Pair{EnterEv: enter, ExitEv: exit}
+
+	if ok := el.handleMemExit(ep, enter); ok {
+		t.Fatal("handleMemExit should reject pair due to filter mismatch")
+	}
+}
+
+func TestInitRawHandlersRegistersMemoryEvents(t *testing.T) {
+	el := mustNewEventLoop(t, eventLoopConfig{})
+	if _, ok := el.rawHandlers[types.ENTER_MEM_EVENT]; !ok {
+		t.Fatal("ENTER_MEM_EVENT handler is not registered")
+	}
+}