@@ -20,7 +20,7 @@ func TestCreateOrUpdateIssueNotifications(t *testing.T) {
20
20
assert .NoError (t , unittest .PrepareTestDatabase ())
21
21
issue := unittest .AssertExistsAndLoadBean (t , & issues_model.Issue {ID : 1 })
22
22
23
- assert .NoError (t , activities_model .CreateOrUpdateIssueNotifications (db . DefaultContext , issue .ID , 0 , 2 , 0 ))
23
+ assert .NoError (t , activities_model .CreateOrUpdateIssueNotifications (t . Context () , issue .ID , 0 , 2 , 0 ))
24
24
25
25
// User 9 is inactive, thus notifications for user 1 and 4 are created
26
26
notf := unittest .AssertExistsAndLoadBean (t , & activities_model.Notification {UserID : 1 , IssueID : issue .ID })
@@ -34,7 +34,7 @@ func TestCreateOrUpdateIssueNotifications(t *testing.T) {
34
34
func TestNotificationsForUser (t * testing.T ) {
35
35
assert .NoError (t , unittest .PrepareTestDatabase ())
36
36
user := unittest .AssertExistsAndLoadBean (t , & user_model.User {ID : 2 })
37
- notfs , err := db .Find [activities_model.Notification ](db . DefaultContext , activities_model.FindNotificationOptions {
37
+ notfs , err := db .Find [activities_model.Notification ](t . Context () , activities_model.FindNotificationOptions {
38
38
UserID : user .ID ,
39
39
Status : []activities_model.NotificationStatus {
40
40
activities_model .NotificationStatusRead ,
@@ -55,7 +55,7 @@ func TestNotificationsForUser(t *testing.T) {
55
55
func TestNotification_GetRepo (t * testing.T ) {
56
56
assert .NoError (t , unittest .PrepareTestDatabase ())
57
57
notf := unittest .AssertExistsAndLoadBean (t , & activities_model.Notification {RepoID : 1 })
58
- repo , err := notf .GetRepo (db . DefaultContext )
58
+ repo , err := notf .GetRepo (t . Context () )
59
59
assert .NoError (t , err )
60
60
assert .Equal (t , repo , notf .Repository )
61
61
assert .Equal (t , notf .RepoID , repo .ID )
@@ -64,7 +64,7 @@ func TestNotification_GetRepo(t *testing.T) {
64
64
func TestNotification_GetIssue (t * testing.T ) {
65
65
assert .NoError (t , unittest .PrepareTestDatabase ())
66
66
notf := unittest .AssertExistsAndLoadBean (t , & activities_model.Notification {RepoID : 1 })
67
- issue , err := notf .GetIssue (db . DefaultContext )
67
+ issue , err := notf .GetIssue (t . Context () )
68
68
assert .NoError (t , err )
69
69
assert .Equal (t , issue , notf .Issue )
70
70
assert .Equal (t , notf .IssueID , issue .ID )
@@ -73,7 +73,7 @@ func TestNotification_GetIssue(t *testing.T) {
73
73
func TestGetNotificationCount (t * testing.T ) {
74
74
assert .NoError (t , unittest .PrepareTestDatabase ())
75
75
user := unittest .AssertExistsAndLoadBean (t , & user_model.User {ID : 1 })
76
- cnt , err := db .Count [activities_model.Notification ](db . DefaultContext , activities_model.FindNotificationOptions {
76
+ cnt , err := db .Count [activities_model.Notification ](t . Context () , activities_model.FindNotificationOptions {
77
77
UserID : user .ID ,
78
78
Status : []activities_model.NotificationStatus {
79
79
activities_model .NotificationStatusRead ,
@@ -82,7 +82,7 @@ func TestGetNotificationCount(t *testing.T) {
82
82
assert .NoError (t , err )
83
83
assert .EqualValues (t , 0 , cnt )
84
84
85
- cnt , err = db .Count [activities_model.Notification ](db . DefaultContext , activities_model.FindNotificationOptions {
85
+ cnt , err = db .Count [activities_model.Notification ](t . Context () , activities_model.FindNotificationOptions {
86
86
UserID : user .ID ,
87
87
Status : []activities_model.NotificationStatus {
88
88
activities_model .NotificationStatusUnread ,
@@ -97,14 +97,14 @@ func TestSetNotificationStatus(t *testing.T) {
97
97
user := unittest .AssertExistsAndLoadBean (t , & user_model.User {ID : 2 })
98
98
notf := unittest .AssertExistsAndLoadBean (t ,
99
99
& activities_model.Notification {UserID : user .ID , Status : activities_model .NotificationStatusRead })
100
- _ , err := activities_model .SetNotificationStatus (db . DefaultContext , notf .ID , user , activities_model .NotificationStatusPinned )
100
+ _ , err := activities_model .SetNotificationStatus (t . Context () , notf .ID , user , activities_model .NotificationStatusPinned )
101
101
assert .NoError (t , err )
102
102
unittest .AssertExistsAndLoadBean (t ,
103
103
& activities_model.Notification {ID : notf .ID , Status : activities_model .NotificationStatusPinned })
104
104
105
- _ , err = activities_model .SetNotificationStatus (db . DefaultContext , 1 , user , activities_model .NotificationStatusRead )
105
+ _ , err = activities_model .SetNotificationStatus (t . Context () , 1 , user , activities_model .NotificationStatusRead )
106
106
assert .Error (t , err )
107
- _ , err = activities_model .SetNotificationStatus (db . DefaultContext , unittest .NonexistentID , user , activities_model .NotificationStatusRead )
107
+ _ , err = activities_model .SetNotificationStatus (t . Context () , unittest .NonexistentID , user , activities_model .NotificationStatusRead )
108
108
assert .Error (t , err )
109
109
}
110
110
@@ -117,7 +117,7 @@ func TestUpdateNotificationStatuses(t *testing.T) {
117
117
& activities_model.Notification {UserID : user .ID , Status : activities_model .NotificationStatusRead })
118
118
notfPinned := unittest .AssertExistsAndLoadBean (t ,
119
119
& activities_model.Notification {UserID : user .ID , Status : activities_model .NotificationStatusPinned })
120
- assert .NoError (t , activities_model .UpdateNotificationStatuses (db . DefaultContext , user , activities_model .NotificationStatusUnread , activities_model .NotificationStatusRead ))
120
+ assert .NoError (t , activities_model .UpdateNotificationStatuses (t . Context () , user , activities_model .NotificationStatusUnread , activities_model .NotificationStatusRead ))
121
121
unittest .AssertExistsAndLoadBean (t ,
122
122
& activities_model.Notification {ID : notfUnread .ID , Status : activities_model .NotificationStatusRead })
123
123
unittest .AssertExistsAndLoadBean (t ,
@@ -130,11 +130,11 @@ func TestSetIssueReadBy(t *testing.T) {
130
130
assert .NoError (t , unittest .PrepareTestDatabase ())
131
131
user := unittest .AssertExistsAndLoadBean (t , & user_model.User {ID : 1 })
132
132
issue := unittest .AssertExistsAndLoadBean (t , & issues_model.Issue {ID : 1 })
133
- assert .NoError (t , db .WithTx (db . DefaultContext , func (ctx context.Context ) error {
133
+ assert .NoError (t , db .WithTx (t . Context () , func (ctx context.Context ) error {
134
134
return activities_model .SetIssueReadBy (ctx , issue .ID , user .ID )
135
135
}))
136
136
137
- nt , err := activities_model .GetIssueNotification (db . DefaultContext , user .ID , issue .ID )
137
+ nt , err := activities_model .GetIssueNotification (t . Context () , user .ID , issue .ID )
138
138
assert .NoError (t , err )
139
139
assert .Equal (t , activities_model .NotificationStatusRead , nt .Status )
140
140
}
0 commit comments