fix: scope notifications by tenant

This commit is contained in:
2026-01-09 15:49:49 +08:00
parent c446dec5c6
commit 4d90f547e6
9 changed files with 62 additions and 44 deletions

View File

@@ -44,16 +44,17 @@ func (s *NotificationTestSuite) Test_CRUD() {
ctx := s.T().Context()
database.Truncate(ctx, s.DB, models.TableNameNotification)
tenantID := int64(1)
uID := int64(100)
ctx = context.WithValue(ctx, consts.CtxKeyUser, uID)
Convey("should send notification", func() {
err := Notification.Send(ctx, uID, "system", "Welcome", "Hello World")
err := Notification.Send(ctx, tenantID, uID, "system", "Welcome", "Hello World")
So(err, ShouldBeNil)
var list *requests.Pager
for i := 0; i < 5; i++ {
list, err = Notification.List(ctx, uID, 1, "")
list, err = Notification.List(ctx, tenantID, uID, 1, "")
So(err, ShouldBeNil)
if list.Total > 0 {
break
@@ -69,7 +70,7 @@ func (s *NotificationTestSuite) Test_CRUD() {
// Mark Read
// Need ID
n, _ := models.NotificationQuery.WithContext(ctx).Where(models.NotificationQuery.UserID.Eq(uID)).First()
err = Notification.MarkRead(ctx, uID, n.ID)
err = Notification.MarkRead(ctx, tenantID, uID, n.ID)
So(err, ShouldBeNil)
nReload, _ := models.NotificationQuery.WithContext(ctx).Where(models.NotificationQuery.ID.Eq(n.ID)).First()