2017-01-30 17:21:49 +03:00
// Copyright 2017 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 models
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestCreateOrUpdateIssueNotifications ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
issue := AssertExistsAndLoadBean ( t , & Issue { ID : 1 } ) . ( * Issue )
2020-04-06 19:33:34 +03:00
assert . NoError ( t , CreateOrUpdateIssueNotifications ( issue . ID , 0 , 2 , 0 ) )
2017-01-30 17:21:49 +03:00
2017-10-24 20:36:19 +03:00
// User 9 is inactive, thus notifications for user 1 and 4 are created
notf := AssertExistsAndLoadBean ( t , & Notification { UserID : 1 , IssueID : issue . ID } ) . ( * Notification )
2017-01-30 17:21:49 +03:00
assert . Equal ( t , NotificationStatusUnread , notf . Status )
2017-02-07 14:47:55 +03:00
CheckConsistencyFor ( t , & Issue { ID : issue . ID } )
2017-10-24 20:36:19 +03:00
notf = AssertExistsAndLoadBean ( t , & Notification { UserID : 4 , IssueID : issue . ID } ) . ( * Notification )
assert . Equal ( t , NotificationStatusUnread , notf . Status )
2017-01-30 17:21:49 +03:00
}
func TestNotificationsForUser ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
user := AssertExistsAndLoadBean ( t , & User { ID : 2 } ) . ( * User )
statuses := [ ] NotificationStatus { NotificationStatusRead , NotificationStatusUnread }
notfs , err := NotificationsForUser ( user , statuses , 1 , 10 )
assert . NoError ( t , err )
2020-01-09 14:56:32 +03:00
if assert . Len ( t , notfs , 3 ) {
assert . EqualValues ( t , 5 , notfs [ 0 ] . ID )
2017-08-28 12:17:45 +03:00
assert . EqualValues ( t , user . ID , notfs [ 0 ] . UserID )
2017-12-07 08:52:57 +03:00
assert . EqualValues ( t , 4 , notfs [ 1 ] . ID )
assert . EqualValues ( t , user . ID , notfs [ 1 ] . UserID )
2020-01-09 14:56:32 +03:00
assert . EqualValues ( t , 2 , notfs [ 2 ] . ID )
assert . EqualValues ( t , user . ID , notfs [ 2 ] . UserID )
2017-08-28 12:17:45 +03:00
}
2017-01-30 17:21:49 +03:00
}
func TestNotification_GetRepo ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
notf := AssertExistsAndLoadBean ( t , & Notification { RepoID : 1 } ) . ( * Notification )
repo , err := notf . GetRepo ( )
assert . NoError ( t , err )
assert . Equal ( t , repo , notf . Repository )
assert . EqualValues ( t , notf . RepoID , repo . ID )
}
func TestNotification_GetIssue ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
notf := AssertExistsAndLoadBean ( t , & Notification { RepoID : 1 } ) . ( * Notification )
issue , err := notf . GetIssue ( )
assert . NoError ( t , err )
assert . Equal ( t , issue , notf . Issue )
assert . EqualValues ( t , notf . IssueID , issue . ID )
}
func TestGetNotificationCount ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
2017-12-07 08:52:57 +03:00
user := AssertExistsAndLoadBean ( t , & User { ID : 1 } ) . ( * User )
cnt , err := GetNotificationCount ( user , NotificationStatusRead )
2017-01-30 17:21:49 +03:00
assert . NoError ( t , err )
assert . EqualValues ( t , 0 , cnt )
2017-12-07 08:52:57 +03:00
cnt , err = GetNotificationCount ( user , NotificationStatusUnread )
2017-01-30 17:21:49 +03:00
assert . NoError ( t , err )
assert . EqualValues ( t , 1 , cnt )
}
func TestSetNotificationStatus ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
user := AssertExistsAndLoadBean ( t , & User { ID : 2 } ) . ( * User )
notf := AssertExistsAndLoadBean ( t ,
& Notification { UserID : user . ID , Status : NotificationStatusRead } ) . ( * Notification )
2021-09-18 02:40:50 +03:00
_ , err := SetNotificationStatus ( notf . ID , user , NotificationStatusPinned )
assert . NoError ( t , err )
2017-01-30 17:21:49 +03:00
AssertExistsAndLoadBean ( t ,
& Notification { ID : notf . ID , Status : NotificationStatusPinned } )
2021-09-18 02:40:50 +03:00
_ , err = SetNotificationStatus ( 1 , user , NotificationStatusRead )
assert . Error ( t , err )
_ , err = SetNotificationStatus ( NonexistentID , user , NotificationStatusRead )
assert . Error ( t , err )
2017-01-30 17:21:49 +03:00
}
2017-12-07 08:52:57 +03:00
func TestUpdateNotificationStatuses ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
user := AssertExistsAndLoadBean ( t , & User { ID : 2 } ) . ( * User )
notfUnread := AssertExistsAndLoadBean ( t ,
& Notification { UserID : user . ID , Status : NotificationStatusUnread } ) . ( * Notification )
notfRead := AssertExistsAndLoadBean ( t ,
& Notification { UserID : user . ID , Status : NotificationStatusRead } ) . ( * Notification )
notfPinned := AssertExistsAndLoadBean ( t ,
& Notification { UserID : user . ID , Status : NotificationStatusPinned } ) . ( * Notification )
assert . NoError ( t , UpdateNotificationStatuses ( user , NotificationStatusUnread , NotificationStatusRead ) )
AssertExistsAndLoadBean ( t ,
& Notification { ID : notfUnread . ID , Status : NotificationStatusRead } )
AssertExistsAndLoadBean ( t ,
& Notification { ID : notfRead . ID , Status : NotificationStatusRead } )
AssertExistsAndLoadBean ( t ,
& Notification { ID : notfPinned . ID , Status : NotificationStatusPinned } )
}