Test probe manager picks up newly added tracepoints

1 files changed, 32 insertions, 0 deletions

diff --git a/internal/probemanager/manager_test.go b/internal/probemanager/manager_test.go
index 44dc259..f610ea3 100644
--- a/internal/probemanager/manager_test.go
+++ b/internal/probemanager/manager_test.go
@@ -151,3 +151,35 @@ func TestManagerAttachAllReturnsProgramError(t *testing.T) {
 		t.Fatalf("expected state to capture attach error, got %+v", states)
 	}
 }
+
+func TestManagerAttachAllPicksUpNewTracepointsOnLaterCall(t *testing.T) {
+	attacher := &fakeAttacher{
+		programs: map[string]*fakeProgram{
+			"handle_sys_enter_read":  {},
+			"handle_sys_exit_read":   {},
+			"handle_sys_enter_write": {},
+			"handle_sys_exit_write":  {},
+		},
+		errs: map[string]error{},
+	}
+	mgr := NewManager(attacher)
+
+	if err := mgr.AttachAll(nil, []string{"sys_enter_read", "sys_exit_read"}); err != nil {
+		t.Fatalf("AttachAll(read) returned error: %v", err)
+	}
+	states := mgr.States()
+	if len(states) != 1 || states[0].Syscall != "read" {
+		t.Fatalf("expected only read after first call, got %+v", states)
+	}
+
+	if err := mgr.AttachAll(nil, []string{"sys_enter_read", "sys_exit_read", "sys_enter_write", "sys_exit_write"}); err != nil {
+		t.Fatalf("AttachAll(read+write) returned error: %v", err)
+	}
+	states = mgr.States()
+	if len(states) != 2 {
+		t.Fatalf("expected new syscall to appear after second call, got %+v", states)
+	}
+	if states[0].Syscall != "read" || states[1].Syscall != "write" {
+		t.Fatalf("unexpected syscall ordering/content: %+v", states)
+	}
+}