@@ -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