feat: complete registration invitation and auth redesign

2026-08-19 00:30:11 +08:00
parent e063e7879e
commit 219ecadf4c
47 changed files with 9014 additions and 224 deletions
@@ -0,0 +1,77 @@
+package main
+
+import (
+	"testing"
+	"time"
+
+	"github.com/tinode/chat/server/store"
+	"github.com/tinode/chat/server/store/types"
+)
+
+type passwordResetEventStore struct {
+	testHTTPInfrastructureStore
+	delivered []string
+}
+
+func (s *passwordResetEventStore) MarkEventDelivered(_ types.TenantID, eventID string, _ time.Time) error {
+	s.delivered = append(s.delivered, eventID)
+	return nil
+}
+
+func TestDispatchPasswordResetEvent(t *testing.T) {
+	oldInfrastructure := store.HTTPInfrastructure
+	oldHub, oldSessions := globals.hub, globals.sessionStore
+	defer func() {
+		store.HTTPInfrastructure = oldInfrastructure
+		globals.hub, globals.sessionStore = oldHub, oldSessions
+	}()
+
+	infrastructure := &passwordResetEventStore{}
+	infrastructure.events = []types.BusinessEvent{{
+		EventID: "password-reset-1", EventType: "security.password_reset",
+		TargetType: "user", Target: "42", Payload: []byte(`{"user_id":42}`),
+	}}
+	store.HTTPInfrastructure = infrastructure
+	globals.hub = &Hub{routeSrv: make(chan *ServerComMessage, 1)}
+	globals.sessionStore = nil
+
+	dispatchIMBusinessEvents(types.TenantID(1))
+
+	if len(infrastructure.delivered) != 1 || infrastructure.delivered[0] != "password-reset-1" {
+		t.Fatalf("password reset event was not marked delivered: %v", infrastructure.delivered)
+	}
+	if len(globals.hub.routeSrv) != 0 {
+		t.Fatal("password reset event was routed as a regular client event")
+	}
+}
+
+func TestDispatchRegistrationInviteUsageEvent(t *testing.T) {
+	oldInfrastructure, oldInvites := store.HTTPInfrastructure, store.RegistrationInvites
+	oldHub := globals.hub
+	defer func() {
+		store.HTTPInfrastructure, store.RegistrationInvites = oldInfrastructure, oldInvites
+		globals.hub = oldHub
+	}()
+
+	infrastructure := &passwordResetEventStore{}
+	infrastructure.events = []types.BusinessEvent{{
+		EventID: "invite-use-1", EventType: "registration.invite_used",
+		TargetType: "user", Target: "42", Payload: []byte(`{"invite_code_id":99}`),
+	}}
+	invites := &testRegistrationInviteStore{}
+	store.HTTPInfrastructure = infrastructure
+	store.RegistrationInvites = invites
+	globals.hub = &Hub{routeSrv: make(chan *ServerComMessage, 1)}
+
+	dispatchIMBusinessEvents(types.TenantID(1))
+
+	if invites.syncedCode != 99 {
+		t.Fatalf("invite usage projection code=%d, want 99", invites.syncedCode)
+	}
+	if len(infrastructure.delivered) != 1 || infrastructure.delivered[0] != "invite-use-1" {
+		t.Fatalf("invite usage event was not marked delivered: %v", infrastructure.delivered)
+	}
+	if len(globals.hub.routeSrv) != 0 {
+		t.Fatal("invite usage event was routed as a regular client event")
+	}
+}