144 lines
4.7 KiB
Go
144 lines
4.7 KiB
Go
// Copyright 2024 The Forgejo Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package forgejo
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNotifications(t *testing.T) {
|
|
log.Println("== TestNotifications ==")
|
|
|
|
// init user2
|
|
c := newTestClient()
|
|
|
|
user1, _, err := c.GetMyUserInfo()
|
|
assert.NoError(t, err)
|
|
user2 := createTestUser(t, "notify2", c)
|
|
|
|
// create 2 repos
|
|
repoA, err := createTestRepo(t, "TestNotifications_A", c)
|
|
assert.NoError(t, err)
|
|
|
|
c.sudo = user2.UserName
|
|
repoB, err := createTestRepo(t, "TestNotifications_B", c)
|
|
assert.NoError(t, err)
|
|
_, err = c.WatchRepo(user1.UserName, repoA.Name)
|
|
c.sudo = ""
|
|
assert.NoError(t, err)
|
|
|
|
c.sudo = user2.UserName
|
|
notifications, _, err := c.ReadNotifications(MarkNotificationOptions{})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, notifications, 0)
|
|
count, _, err := c.CheckNotifications()
|
|
assert.EqualValues(t, 0, count)
|
|
assert.NoError(t, err)
|
|
c.sudo = ""
|
|
_, _, err = c.CreateIssue(repoA.Owner.UserName, repoA.Name, CreateIssueOption{Title: "A Issue", Closed: false})
|
|
assert.NoError(t, err)
|
|
issue, _, err := c.CreateIssue(repoB.Owner.UserName, repoB.Name, CreateIssueOption{Title: "B Issue", Closed: false})
|
|
assert.NoError(t, err)
|
|
time.Sleep(time.Second * 5)
|
|
|
|
// CheckNotifications of user2
|
|
c.sudo = user2.UserName
|
|
count, _, err = c.CheckNotifications()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 2, count)
|
|
|
|
// ListNotifications
|
|
nList, _, err := c.ListNotifications(ListNotificationOptions{})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, nList, 2)
|
|
for _, n := range nList {
|
|
assert.EqualValues(t, true, n.Unread)
|
|
assert.EqualValues(t, "Issue", n.Subject.Type)
|
|
assert.EqualValues(t, NotifySubjectOpen, nList[0].Subject.State)
|
|
assert.EqualValues(t, NotifySubjectOpen, nList[1].Subject.State)
|
|
if n.Subject.Title == "A Issue" {
|
|
assert.EqualValues(t, repoA.Name, n.Repository.Name)
|
|
} else if n.Subject.Title == "B Issue" {
|
|
assert.EqualValues(t, repoB.Name, n.Repository.Name)
|
|
} else {
|
|
assert.Error(t, fmt.Errorf("ListNotifications returned a Issue witch should not"))
|
|
}
|
|
}
|
|
|
|
// ListRepoNotifications
|
|
nList, _, err = c.ListRepoNotifications(repoA.Owner.UserName, repoA.Name, ListNotificationOptions{})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, nList, 1)
|
|
assert.EqualValues(t, "A Issue", nList[0].Subject.Title)
|
|
// ReadRepoNotifications
|
|
notifications, _, err = c.ReadRepoNotifications(repoA.Owner.UserName, repoA.Name, MarkNotificationOptions{})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, notifications, 1)
|
|
|
|
// GetThread
|
|
n, _, err := c.GetNotification(nList[0].ID)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, false, n.Unread)
|
|
assert.EqualValues(t, "A Issue", n.Subject.Title)
|
|
|
|
// ReadNotifications
|
|
notifications, _, err = c.ReadNotifications(MarkNotificationOptions{})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, notifications, 1)
|
|
nList, _, err = c.ListNotifications(ListNotificationOptions{})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, nList, 0)
|
|
|
|
// ReadThread
|
|
iState := StateClosed
|
|
c.sudo = ""
|
|
_, _, err = c.EditIssue(repoB.Owner.UserName, repoB.Name, issue.Index, EditIssueOption{State: &iState})
|
|
assert.NoError(t, err)
|
|
time.Sleep(time.Second * 5)
|
|
|
|
c.sudo = user2.UserName
|
|
nList, _, err = c.ListNotifications(ListNotificationOptions{})
|
|
assert.NoError(t, err)
|
|
count, _, err = c.CheckNotifications()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, count)
|
|
if assert.Len(t, nList, 1) {
|
|
assert.EqualValues(t, NotifySubjectClosed, nList[0].Subject.State)
|
|
notification, _, err := c.ReadNotification(nList[0].ID)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, notification.ID, nList[0].ID)
|
|
}
|
|
|
|
c.sudo = ""
|
|
notifications, _, err = c.ReadNotifications(MarkNotificationOptions{})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, notifications, 2)
|
|
nList, _, err = c.ListNotifications(ListNotificationOptions{Status: []NotifyStatus{NotifyStatusRead}})
|
|
assert.NoError(t, err)
|
|
if assert.Len(t, nList, 2) {
|
|
notification, _, err := c.ReadNotification(nList[0].ID, NotifyStatusPinned)
|
|
assert.EqualValues(t, notification.ID, nList[0].ID)
|
|
assert.NoError(t, err)
|
|
|
|
notification, _, err = c.ReadNotification(nList[1].ID, NotifyStatusUnread)
|
|
assert.EqualValues(t, notification.ID, nList[1].ID)
|
|
assert.NoError(t, err)
|
|
}
|
|
nList, _, err = c.ListNotifications(ListNotificationOptions{Status: []NotifyStatus{NotifyStatusPinned, NotifyStatusUnread}})
|
|
assert.NoError(t, err)
|
|
if assert.Len(t, nList, 2) {
|
|
assert.EqualValues(t, NotifySubjectOpen, nList[0].Subject.State)
|
|
assert.EqualValues(t, NotifySubjectOpen, nList[1].Subject.State)
|
|
}
|
|
}
|