@@ -20,7 +20,7 @@ func TestCreateOrUpdateIssueNotifications(t *testing.T) {
2020 assert .NoError (t , unittest .PrepareTestDatabase ())
2121 issue := unittest .AssertExistsAndLoadBean (t , & issues_model.Issue {ID : 1 })
2222
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 ))
2424
2525 // User 9 is inactive, thus notifications for user 1 and 4 are created
2626 notf := unittest .AssertExistsAndLoadBean (t , & activities_model.Notification {UserID : 1 , IssueID : issue .ID })
@@ -34,7 +34,7 @@ func TestCreateOrUpdateIssueNotifications(t *testing.T) {
3434func TestNotificationsForUser (t * testing.T ) {
3535 assert .NoError (t , unittest .PrepareTestDatabase ())
3636 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 {
3838 UserID : user .ID ,
3939 Status : []activities_model.NotificationStatus {
4040 activities_model .NotificationStatusRead ,
@@ -55,7 +55,7 @@ func TestNotificationsForUser(t *testing.T) {
5555func TestNotification_GetRepo (t * testing.T ) {
5656 assert .NoError (t , unittest .PrepareTestDatabase ())
5757 notf := unittest .AssertExistsAndLoadBean (t , & activities_model.Notification {RepoID : 1 })
58- repo , err := notf .GetRepo (db . DefaultContext )
58+ repo , err := notf .GetRepo (t . Context () )
5959 assert .NoError (t , err )
6060 assert .Equal (t , repo , notf .Repository )
6161 assert .Equal (t , notf .RepoID , repo .ID )
@@ -64,7 +64,7 @@ func TestNotification_GetRepo(t *testing.T) {
6464func TestNotification_GetIssue (t * testing.T ) {
6565 assert .NoError (t , unittest .PrepareTestDatabase ())
6666 notf := unittest .AssertExistsAndLoadBean (t , & activities_model.Notification {RepoID : 1 })
67- issue , err := notf .GetIssue (db . DefaultContext )
67+ issue , err := notf .GetIssue (t . Context () )
6868 assert .NoError (t , err )
6969 assert .Equal (t , issue , notf .Issue )
7070 assert .Equal (t , notf .IssueID , issue .ID )
@@ -73,7 +73,7 @@ func TestNotification_GetIssue(t *testing.T) {
7373func TestGetNotificationCount (t * testing.T ) {
7474 assert .NoError (t , unittest .PrepareTestDatabase ())
7575 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 {
7777 UserID : user .ID ,
7878 Status : []activities_model.NotificationStatus {
7979 activities_model .NotificationStatusRead ,
@@ -82,7 +82,7 @@ func TestGetNotificationCount(t *testing.T) {
8282 assert .NoError (t , err )
8383 assert .EqualValues (t , 0 , cnt )
8484
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 {
8686 UserID : user .ID ,
8787 Status : []activities_model.NotificationStatus {
8888 activities_model .NotificationStatusUnread ,
@@ -97,14 +97,14 @@ func TestSetNotificationStatus(t *testing.T) {
9797 user := unittest .AssertExistsAndLoadBean (t , & user_model.User {ID : 2 })
9898 notf := unittest .AssertExistsAndLoadBean (t ,
9999 & 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 )
101101 assert .NoError (t , err )
102102 unittest .AssertExistsAndLoadBean (t ,
103103 & activities_model.Notification {ID : notf .ID , Status : activities_model .NotificationStatusPinned })
104104
105- _ , err = activities_model .SetNotificationStatus (db . DefaultContext , 1 , user , activities_model .NotificationStatusRead )
105+ _ , err = activities_model .SetNotificationStatus (t . Context () , 1 , user , activities_model .NotificationStatusRead )
106106 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 )
108108 assert .Error (t , err )
109109}
110110
@@ -117,7 +117,7 @@ func TestUpdateNotificationStatuses(t *testing.T) {
117117 & activities_model.Notification {UserID : user .ID , Status : activities_model .NotificationStatusRead })
118118 notfPinned := unittest .AssertExistsAndLoadBean (t ,
119119 & 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 ))
121121 unittest .AssertExistsAndLoadBean (t ,
122122 & activities_model.Notification {ID : notfUnread .ID , Status : activities_model .NotificationStatusRead })
123123 unittest .AssertExistsAndLoadBean (t ,
@@ -130,11 +130,11 @@ func TestSetIssueReadBy(t *testing.T) {
130130 assert .NoError (t , unittest .PrepareTestDatabase ())
131131 user := unittest .AssertExistsAndLoadBean (t , & user_model.User {ID : 1 })
132132 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 {
134134 return activities_model .SetIssueReadBy (ctx , issue .ID , user .ID )
135135 }))
136136
137- nt , err := activities_model .GetIssueNotification (db . DefaultContext , user .ID , issue .ID )
137+ nt , err := activities_model .GetIssueNotification (t . Context () , user .ID , issue .ID )
138138 assert .NoError (t , err )
139139 assert .Equal (t , activities_model .NotificationStatusRead , nt .Status )
140140}
0 commit comments