41 lines
1.7 KiB
Go
41 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
"time"
|
|
|
|
"proxy-pool/internal/loadgen"
|
|
)
|
|
|
|
func TestExecutePassesBoundedWorkloadAndWritesJSON(t *testing.T) {
|
|
var received loadgen.Options
|
|
var stdout, stderr bytes.Buffer
|
|
code := execute(context.Background(), []string{
|
|
"-target", "https://target.example/path", "-proxy", "http://gateway.example:8080", "-method", "post",
|
|
"-requests", "3", "-concurrency", "2", "-timeout", "2s", "-header", "X-Run: fixed", "-body", `{"count":1}`,
|
|
}, func(_ context.Context, options loadgen.Options) (loadgen.Report, error) {
|
|
received = options
|
|
return loadgen.Report{Requests: 3, Completed: 3, Succeeded: 3, Duration: time.Second}, nil
|
|
}, &stdout, &stderr)
|
|
if code != 0 || received.TargetURL != "https://target.example/path" || received.ProxyURL != "http://gateway.example:8080" ||
|
|
received.Method != "post" || received.Requests != 3 || received.Duration != 0 || received.Concurrency != 2 ||
|
|
received.Headers.Get("X-Run") != "fixed" || string(received.RequestBody) != `{"count":1}` || stderr.Len() != 0 {
|
|
t.Fatalf("execute() = %d; options=%+v stderr=%q", code, received, stderr.String())
|
|
}
|
|
var report loadgen.Report
|
|
if err := json.Unmarshal(stdout.Bytes(), &report); err != nil || report.Succeeded != 3 {
|
|
t.Fatalf("JSON report = (%+v, %v)", report, err)
|
|
}
|
|
}
|
|
|
|
func TestExecuteRejectsInvalidWorkload(t *testing.T) {
|
|
var stdout, stderr bytes.Buffer
|
|
code := execute(context.Background(), []string{"-target", "http://target.example"}, loadgen.Run, &stdout, &stderr)
|
|
if code != 2 || stdout.Len() != 0 || stderr.Len() == 0 {
|
|
t.Fatalf("execute() = %d stdout=%q stderr=%q", code, stdout.String(), stderr.String())
|
|
}
|
|
}
|