diff options
Diffstat (limited to 'internal/processing/conversations/conversations_test.go')
| -rw-r--r-- | internal/processing/conversations/conversations_test.go | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/internal/processing/conversations/conversations_test.go b/internal/processing/conversations/conversations_test.go index 0f96ecbac..40145c2fb 100644 --- a/internal/processing/conversations/conversations_test.go +++ b/internal/processing/conversations/conversations_test.go @@ -75,7 +75,7 @@ type ConversationsTestSuite struct { } func (suite *ConversationsTestSuite) getClientMsg(timeout time.Duration) (*messages.FromClientAPI, bool) { - ctx := context.Background() + ctx := suite.T().Context() ctx, cncl := context.WithTimeout(ctx, timeout) defer cncl() return suite.state.Workers.Client.Queue.PopCtx(ctx) @@ -130,8 +130,8 @@ func (suite *ConversationsTestSuite) TearDownTest() { (*gtsmodel.ConversationToStatus)(nil), } for _, model := range conversationModels { - if err := suite.db.DropTable(context.Background(), model); err != nil { - log.Error(context.Background(), err) + if err := suite.db.DropTable(suite.T().Context(), model); err != nil { + log.Error(suite.T().Context(), err) } } |