Fix mocks for poll handler generic tests

internal/server/singleprocess/poll_test.go

@@ -246,6 +246,11 @@ func TestPollQueuer_queue(t *testing.T) {
 			Run(func(args mock.Arguments) {
 				atomic.AddUint32(&peekCounter, 1)
 			})
+		mockH.On("GeneratePollJobs", mock.Anything, 42).
+			Return(nil, errors.New("oh no")).
+			Run(func(args mock.Arguments) {
+				atomic.AddUint32(&pollCounter, 1)
+			})
 		mockH.On("PollJob", mock.Anything, 42).
 			Return(nil, errors.New("oh no")).
 			Run(func(args mock.Arguments) {
@@ -290,10 +295,18 @@ func TestPollQueuer_queue(t *testing.T) {
 			Run(func(args mock.Arguments) {
 				atomic.AddUint32(&peekCounter, 1)
 			})
+		mockH.On("GeneratePollJobs", mock.Anything, 42).
+			Return(nil, errors.New("oh no")).
+			Run(func(args mock.Arguments) {
+				atomic.AddUint32(&pollCounter, 1)
+			})
 		mockH.On("PollJob", mock.Anything, 42).
 			Return(&pb.QueueJobRequest{}, nil).
 			Run(func(args mock.Arguments) {
-				atomic.AddUint32(&pollCounter, 1)
+			})
+		mockH.On("Complete", mock.Anything, 42).
+			Return(nil, nil).
+			Run(func(args mock.Arguments) {
 			})
 
 		// Start