diff options
Diffstat (limited to 'internal/webpush/realsender_test.go')
| -rw-r--r-- | internal/webpush/realsender_test.go | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/internal/webpush/realsender_test.go b/internal/webpush/realsender_test.go index cf9cf4cac..bddcbbc28 100644 --- a/internal/webpush/realsender_test.go +++ b/internal/webpush/realsender_test.go @@ -167,7 +167,7 @@ func (suite *RealSenderStandardTestSuite) simulatePushNotification( expectDeletedSubscription bool, ) error { // Don't let the test run forever if the push notification was not sent for some reason. - ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) + ctx, cancel := context.WithTimeout(suite.T().Context(), 3*time.Second) defer cancel() notification, err := suite.state.DB.GetNotificationByID(ctx, notificationID) @@ -263,7 +263,7 @@ func (suite *RealSenderStandardTestSuite) TestSendPolicyMismatch() { StatusOrEditID: "01F8MHAMCHF6Y650WCRSCP4WMY", Read: util.Ptr(false), } - if err := suite.db.PutNotification(context.Background(), notification); !suite.NoError(err) { + if err := suite.db.PutNotification(suite.T().Context(), notification); !suite.NoError(err) { suite.FailNow(err.Error()) return } |