From 857b007c0e05f50d1667dab69294c09ea4f5e84f Mon Sep 17 00:00:00 2001 From: taylanisikdemir Date: Thu, 21 Mar 2024 14:32:25 -0700 Subject: [PATCH] Address data race in task processor test (#5807) --- service/history/task/processor_test.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/service/history/task/processor_test.go b/service/history/task/processor_test.go index 73220ed7c86..682a0119f26 100644 --- a/service/history/task/processor_test.go +++ b/service/history/task/processor_test.go @@ -104,9 +104,11 @@ func (s *queueTaskProcessorSuite) TestGetOrCreateShardTaskScheduler_ProcessorNot func (s *queueTaskProcessorSuite) TestGetOrCreateShardTaskScheduler_ShardProcessorAlreadyExists() { mockScheduler := task.NewMockScheduler(s.controller) + mockScheduler.EXPECT().Stop().Times(1) s.processor.shardSchedulers[s.mockShard] = mockScheduler s.processor.Start() + defer s.processor.Stop() scheduler, err := s.processor.getOrCreateShardTaskScheduler(s.mockShard) s.NoError(err) s.Equal(mockScheduler, scheduler) @@ -116,6 +118,7 @@ func (s *queueTaskProcessorSuite) TestGetOrCreateShardTaskScheduler_ShardProcess s.Empty(s.processor.shardSchedulers) s.processor.Start() + defer s.processor.Stop() scheduler, err := s.processor.getOrCreateShardTaskScheduler(s.mockShard) s.NoError(err)