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",
	}, 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" || 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())
	}
}