feat: add inbox

internal/hook/hook_test.go

@@ -3,10 +3,12 @@ package hook
 import (
 	"bytes"
 	"encoding/json"
+	"errors"
 	"strings"
 	"testing"
 
 	"github.com/longbin/agent-notify/internal/config"
+	"github.com/longbin/agent-notify/internal/inbox"
 )
 
 func TestCursorStopHookDisabled(t *testing.T) {
@@ -46,3 +48,105 @@ func TestClaudeStopHookActiveSkips(t *testing.T) {
 		t.Fatalf("expected {}, got %q", out.String())
 	}
 }
+
+func TestCursorHookUploadsInboxRecord(t *testing.T) {
+	stubCursorSend(t)
+	var uploaded inbox.Record
+	stubInbox(t,
+		func(rec inbox.Record, cfg config.Config) error {
+			uploaded = rec
+			return nil
+		},
+		func(rec inbox.Record) error {
+			t.Fatalf("unexpected fallback append: %+v", rec)
+			return nil
+		},
+	)
+
+	cfg := config.Default()
+	err := RunCursor(bytes.NewReader([]byte(`{"workspace_roots":["/tmp/proj"]}`)), cfg, "stop", &bytes.Buffer{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if uploaded.Agent != "Cursor" || uploaded.Event != "stop" || uploaded.CWD != "/tmp/proj" {
+		t.Fatalf("unexpected upload record: %+v", uploaded)
+	}
+}
+
+func TestCursorHookFallbacksWhenInboxUploadFails(t *testing.T) {
+	stubCursorSend(t)
+	var fallback inbox.Record
+	stubInbox(t,
+		func(rec inbox.Record, cfg config.Config) error {
+			return errors.New("offline")
+		},
+		func(rec inbox.Record) error {
+			fallback = rec
+			return nil
+		},
+	)
+
+	cfg := config.Default()
+	err := RunCursor(bytes.NewReader([]byte(`{"workspace_roots":["/tmp/proj"]}`)), cfg, "stop", &bytes.Buffer{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if fallback.Source != inbox.SourceFallback || fallback.Title == "" {
+		t.Fatalf("unexpected fallback record: %+v", fallback)
+	}
+}
+
+func TestDisabledHookDoesNotRecordInbox(t *testing.T) {
+	stubCursorSend(t)
+	stubInbox(t,
+		func(rec inbox.Record, cfg config.Config) error {
+			t.Fatalf("unexpected upload: %+v", rec)
+			return nil
+		},
+		func(rec inbox.Record) error {
+			t.Fatalf("unexpected fallback: %+v", rec)
+			return nil
+		},
+	)
+
+	cfg := config.Default()
+	cfg.Events.Stop = false
+	if err := RunCursor(bytes.NewReader([]byte(`{}`)), cfg, "stop", &bytes.Buffer{}); err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestClaudeHookUploadsInboxRecord(t *testing.T) {
+	var uploaded inbox.Record
+	stubInbox(t,
+		func(rec inbox.Record, cfg config.Config) error {
+			uploaded = rec
+			return nil
+		},
+		func(rec inbox.Record) error {
+			t.Fatalf("unexpected fallback append: %+v", rec)
+			return nil
+		},
+	)
+
+	cfg := config.Default()
+	var out bytes.Buffer
+	if err := RunClaude(strings.NewReader(`{"stop_hook_active":false}`), cfg, "stop", &out); err != nil {
+		t.Fatal(err)
+	}
+	if uploaded.Agent != "Claude" || uploaded.Event != "stop" || uploaded.Title == "" {
+		t.Fatalf("unexpected upload record: %+v", uploaded)
+	}
+}
+
+func stubInbox(t *testing.T, upload func(inbox.Record, config.Config) error, appendLocal func(inbox.Record) error) {
+	t.Helper()
+	prevUpload := uploadInboxRecord
+	prevAppend := appendInboxRecord
+	uploadInboxRecord = upload
+	appendInboxRecord = appendLocal
+	t.Cleanup(func() {
+		uploadInboxRecord = prevUpload
+		appendInboxRecord = prevAppend
+	})
+}