test(02-01): add failing tests for protocol types, registry, and label store

- TestRequestMarshal, TestResponseWithSessions
- TestRegistryUpdate, TestRegistryWaitingSince, TestRegistryRemoveStale
- TestLabelStoreSetGet, TestLabelStorePersistence, TestLabelStoreLoadMissing

daemon_test.go

@@ -0,0 +1,179 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+)
+
+func TestRegistryUpdate(t *testing.T) {
+	reg := NewRegistry()
+
+	info := SessionInfo{
+		PID:       1234,
+		SessionID: "sess-1",
+		Cwd:       "/home/user/project",
+		State:     "Working",
+	}
+
+	reg.Update(info)
+
+	list := reg.List()
+	if len(list) != 1 {
+		t.Fatalf("list len = %d, want 1", len(list))
+	}
+	if list[0].SessionID != "sess-1" {
+		t.Errorf("session_id = %q, want %q", list[0].SessionID, "sess-1")
+	}
+}
+
+func TestRegistryWaitingSince(t *testing.T) {
+	reg := NewRegistry()
+
+	// Session starts Working
+	reg.Update(SessionInfo{
+		PID:       1234,
+		SessionID: "sess-1",
+		State:     "Working",
+	})
+
+	list := reg.List()
+	if list[0].WaitingSince != nil {
+		t.Error("WaitingSince should be nil when Working")
+	}
+
+	// Session transitions to NeedsInput
+	reg.Update(SessionInfo{
+		PID:       1234,
+		SessionID: "sess-1",
+		State:     "Needs Input",
+	})
+
+	list = reg.List()
+	if list[0].WaitingSince == nil {
+		t.Fatal("WaitingSince should be set when NeedsInput")
+	}
+
+	waitStart := *list[0].WaitingSince
+
+	// Session goes back to Working -> WaitingSince reset
+	reg.Update(SessionInfo{
+		PID:       1234,
+		SessionID: "sess-1",
+		State:     "Working",
+	})
+
+	list = reg.List()
+	if list[0].WaitingSince != nil {
+		t.Errorf("WaitingSince should be nil after returning to Working, got %v", list[0].WaitingSince)
+	}
+
+	_ = waitStart
+}
+
+func TestRegistryRemoveStale(t *testing.T) {
+	reg := NewRegistry()
+
+	reg.Update(SessionInfo{SessionID: "sess-1", State: "Working"})
+	reg.Update(SessionInfo{SessionID: "sess-2", State: "Working"})
+
+	// Only sess-1 is still active
+	active := map[string]bool{"sess-1": true}
+	reg.RemoveStale(active)
+
+	list := reg.List()
+	if len(list) != 1 {
+		t.Fatalf("list len = %d, want 1", len(list))
+	}
+	if list[0].SessionID != "sess-1" {
+		t.Errorf("session_id = %q, want %q", list[0].SessionID, "sess-1")
+	}
+}
+
+func TestLabelStoreSetGet(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "labels.json")
+
+	ls, err := NewLabelStore(path)
+	if err != nil {
+		t.Fatalf("new: %v", err)
+	}
+
+	if err := ls.Set("sess-1", "review MR"); err != nil {
+		t.Fatalf("set: %v", err)
+	}
+
+	got := ls.Get("sess-1")
+	if got != "review MR" {
+		t.Errorf("get = %q, want %q", got, "review MR")
+	}
+
+	// Non-existent key returns empty
+	if got := ls.Get("unknown"); got != "" {
+		t.Errorf("get unknown = %q, want empty", got)
+	}
+}
+
+func TestLabelStorePersistence(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "labels.json")
+
+	ls, err := NewLabelStore(path)
+	if err != nil {
+		t.Fatalf("new: %v", err)
+	}
+	if err := ls.Set("sess-1", "review MR"); err != nil {
+		t.Fatalf("set: %v", err)
+	}
+
+	// Create a new LabelStore from the same file
+	ls2, err := NewLabelStore(path)
+	if err != nil {
+		t.Fatalf("new2: %v", err)
+	}
+
+	got := ls2.Get("sess-1")
+	if got != "review MR" {
+		t.Errorf("persisted get = %q, want %q", got, "review MR")
+	}
+}
+
+func TestLabelStoreLoadMissing(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "nonexistent", "labels.json")
+
+	ls, err := NewLabelStore(path)
+	if err != nil {
+		t.Fatalf("should not error on missing file: %v", err)
+	}
+
+	if got := ls.Get("anything"); got != "" {
+		t.Errorf("get = %q, want empty", got)
+	}
+}
+
+func TestRegistryUpdateTimestamp(t *testing.T) {
+	reg := NewRegistry()
+
+	before := time.Now()
+	reg.Update(SessionInfo{
+		SessionID: "sess-1",
+		State:     "Needs Input",
+	})
+	after := time.Now()
+
+	list := reg.List()
+	if list[0].WaitingSince == nil {
+		t.Fatal("WaitingSince should be set")
+	}
+	ws := *list[0].WaitingSince
+	if ws.Before(before) || ws.After(after) {
+		t.Errorf("WaitingSince %v not between %v and %v", ws, before, after)
+	}
+}
+
+// Placeholder to verify file exists
+func init() {
+	_ = os.TempDir()
+}