@@ -67,7 +67,7 @@ func testUserQuietHoursScheduleStore() *atomic.Pointer[schedule.UserQuietHoursSc
6767
6868func TestAcquireJob_LongPoll (t * testing.T ) {
6969t .Parallel ()
70- //nolint:dogsled // ૮・ᴥ・ა
70+ //nolint:dogsled
7171srv ,_ ,_ ,_ := setup (t ,false ,& overrides {acquireJobLongPollDuration :time .Microsecond })
7272job ,err := srv .AcquireJob (context .Background (),nil )
7373require .NoError (t ,err )
@@ -76,7 +76,7 @@ func TestAcquireJob_LongPoll(t *testing.T) {
7676
7777func TestAcquireJobWithCancel_Cancel (t * testing.T ) {
7878t .Parallel ()
79- //nolint:dogsled // ૮ ˶′ﻌ ‵˶ ა
79+ //nolint:dogsled
8080srv ,_ ,_ ,_ := setup (t ,false ,nil )
8181ctx ,cancel := context .WithTimeout (context .Background (),testutil .WaitShort )
8282defer cancel ()
@@ -101,8 +101,8 @@ func TestAcquireJobWithCancel_Cancel(t *testing.T) {
101101func TestHeartbeat (t * testing.T ) {
102102t .Parallel ()
103103
104- ctx , cancel := context . WithCancel ( context . Background ())
105- t . Cleanup ( cancel )
104+ numBeats := 3
105+ ctx := testutil . Context ( t , testutil . WaitShort )
106106heartbeatChan := make (chan struct {})
107107heartbeatFn := func (hbCtx context.Context )error {
108108t .Logf ("heartbeat" )
@@ -114,28 +114,17 @@ func TestHeartbeat(t *testing.T) {
114114return nil
115115}
116116}
117- //nolint:dogsled // ｡:ﾟ૮ ˶ˆ ﻌ ˆ˶ ა ﾟ:｡
117+ //nolint:dogsled
118118_ ,_ ,_ ,_ = setup (t ,false ,& overrides {
119119ctx :ctx ,
120120heartbeatFn :heartbeatFn ,
121121heartbeatInterval :testutil .IntervalFast ,
122122})
123123
124- _ ,ok := <- heartbeatChan
125- require .True (t ,ok ,"first heartbeat not received" )
126- _ ,ok = <- heartbeatChan
127- require .True (t ,ok ,"second heartbeat not received" )
128- cancel ()
129- // Close the channel to ensure we don't receive any more heartbeats.
130- // The test will fail if we do.
131- defer func () {
132- if r := recover ();r != nil {
133- t .Fatalf ("heartbeat received after cancel: %v" ,r )
134- }
135- }()
136-
137- close (heartbeatChan )
138- <- time .After (testutil .IntervalMedium )
124+ for i := 0 ;i < numBeats ;i ++ {
125+ testutil .RequireRecvCtx (ctx ,t ,heartbeatChan )
126+ }
127+ // goleak.VerifyTestMain ensures that the heartbeat goroutine does not leak
139128}
140129
141130func TestAcquireJob (t * testing.T ) {