Fix LiveGenerator test

pkg/scan/request_test.go

@@ -223,27 +223,22 @@ func TestScanRequests(t *testing.T) {
 
 func TestLiveRequestGeneratorContextExit(t *testing.T) {
 	t.Parallel()
+
 	rg := NewLiveRequestGenerator(5 * time.Second)
-	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Microsecond)
-	defer cancel()
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
 	requests, err := rg.GenerateRequests(ctx, newScanRange())
 	require.NoError(t, err)
-	capacity := cap(requests)
-	cnt := 0
-	time.Sleep(1 * time.Microsecond)
+	// consume all requests
 loop:
 	for {
 		select {
 		case _, ok := <-requests:
 			if !ok {
 				break loop
 			}
-			if cnt > capacity {
-				require.FailNow(t, "number of elements more than channel capacity")
-			}
-			cnt++
 		case <-time.After(waitTimeout):
-			require.FailNow(t, "wait timeout")
+			require.Fail(t, "test timeout")
 		}
 	}
 }