Fix test (one more time)

output_http.go

@@ -148,6 +148,7 @@ func NewHTTPOutput(address string, config *HTTPOutputConfig) io.Writer {
 func (o *HTTPOutput) workerMaster() {
 	for {
 		newWorkers := <-o.needWorker
+		atomic.AddInt64(&o.activeWorkers, int64(newWorkers))
 		for i := 0; i < newWorkers; i++ {
 			go o.startWorker()
 		}
@@ -171,7 +172,6 @@ func (o *HTTPOutput) sessionWorkerMaster() {
 
 			if !ok {
 				atomic.AddInt64(&o.activeWorkers, 1)
-
 				worker = newHTTPWorker(o, nil)
 				o.workerSessions[sessionID] = worker
 			}
@@ -201,8 +201,6 @@ func (o *HTTPOutput) startWorker() {
 		ResponseBufferSize: o.config.BufferSize,
 	})
 
-	atomic.AddInt64(&o.activeWorkers, 1)
-
 	for {
 		select {
 		case data := <-o.queue:

output_http_test.go

@@ -57,12 +57,11 @@ func TestHTTPOutput(t *testing.T) {
 		input.EmitGET()
 	}
 
-	wg.Wait()
-
-	if output.(*HTTPOutput).activeWorkers != 200 {
+	if output.(*HTTPOutput).activeWorkers < 200 {
 		t.Error("Should create workers for each request", output.(*HTTPOutput).activeWorkers)
 	}
 
+	wg.Wait()
 	close(quit)
 
 	Settings.modifierConfig = HTTPModifierConfig{}