fix: scope notifications by tenant
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user