diff options
Diffstat (limited to 'internal/processing/polls')
| -rw-r--r-- | internal/processing/polls/poll_test.go | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/internal/processing/polls/poll_test.go b/internal/processing/polls/poll_test.go index 421f05361..f7c0f9f02 100644 --- a/internal/processing/polls/poll_test.go +++ b/internal/processing/polls/poll_test.go @@ -68,7 +68,7 @@ func (suite *PollTestSuite) TearDownTest() { func (suite *PollTestSuite) TestPollGet() { // Create a new context for this test. - ctx, cncl := context.WithCancel(context.Background()) + ctx, cncl := context.WithCancel(suite.T().Context()) defer cncl() // Perform test for all requester + poll combos. @@ -111,7 +111,7 @@ func (suite *PollTestSuite) testPollGet(ctx context.Context, requester *gtsmodel func (suite *PollTestSuite) TestPollVote() { // Create a new context for this test. - ctx, cncl := context.WithCancel(context.Background()) + ctx, cncl := context.WithCancel(suite.T().Context()) defer cncl() // randomChoices generates random vote choices in poll. |