Add e2e tests

e2e/proxy_test.go

@@ -0,0 +1,103 @@
+package e2e
+
+import (
+	"testing"
+)
+
+// Note: We can't easily test actual proxy connections in E2E tests without
+// setting up real proxy servers. These tests verify the validation and
+// error handling around proxy configuration.
+
+func TestProxyValidSchemes(t *testing.T) {
+	t.Parallel()
+
+	// Valid proxy scheme should not cause a validation error
+	// (will fail at connection time since no proxy is running, but should pass validation)
+	for _, scheme := range []string{"http", "https", "socks5", "socks5h"} {
+		t.Run(scheme, func(t *testing.T) {
+			t.Parallel()
+
+			res := run("-U", "http://example.com", "-r", "1", "-z", "-q", "-o", "json",
+				"-X", scheme+"://127.0.0.1:9999")
+			assertExitCode(t, res, 0)
+
+			out := res.jsonOutput(t)
+			assertHasResponseKey(t, out, "dry-run")
+		})
+	}
+}
+
+func TestProxyInvalidScheme(t *testing.T) {
+	t.Parallel()
+
+	res := run("-U", "http://example.com", "-r", "1", "-q", "-o", "json",
+		"-X", "ftp://proxy.example.com:8080")
+	assertExitCode(t, res, 1)
+	assertContains(t, res.Stderr, "VALIDATION")
+}
+
+func TestMultipleProxiesDryRun(t *testing.T) {
+	t.Parallel()
+
+	// Multiple proxies with dry-run to verify they're accepted
+	res := run("-U", "http://example.com", "-r", "3", "-z", "-q", "-o", "json",
+		"-X", "http://127.0.0.1:8080",
+		"-X", "http://127.0.0.1:8081")
+	assertExitCode(t, res, 0)
+
+	out := res.jsonOutput(t)
+	assertResponseCount(t, out, 3)
+}
+
+func TestProxyConnectionFailure(t *testing.T) {
+	t.Parallel()
+
+	// Use a proxy that doesn't exist — should get a connection error
+	res := run("-U", "http://example.com", "-r", "1", "-q", "-o", "json",
+		"-X", "http://127.0.0.1:1")
+	// The process should still exit (may exit 0 with error in output or exit 1)
+	if res.ExitCode == 0 {
+		out := res.jsonOutput(t)
+		// Should NOT get a 200 — should have a proxy error
+		if _, ok := out.Responses["200"]; ok {
+			t.Error("expected proxy connection error, but got 200")
+		}
+	}
+}
+
+func TestProxyFromConfigFile(t *testing.T) {
+	t.Parallel()
+
+	config := `
+url: "http://example.com"
+requests: 1
+quiet: true
+output: json
+dryRun: true
+proxy:
+  - http://127.0.0.1:8080
+`
+	configPath := writeTemp(t, "proxy_config.yaml", config)
+
+	res := run("-f", configPath)
+	assertExitCode(t, res, 0)
+
+	out := res.jsonOutput(t)
+	assertHasResponseKey(t, out, "dry-run")
+}
+
+func TestProxyFromEnv(t *testing.T) {
+	t.Parallel()
+
+	res := runWithEnv(map[string]string{
+		"SARIN_URL":      "http://example.com",
+		"SARIN_REQUESTS": "1",
+		"SARIN_DRY_RUN":  "true",
+		"SARIN_OUTPUT":   "json",
+		"SARIN_PROXY":    "http://127.0.0.1:8080",
+	}, "-q")
+	assertExitCode(t, res, 0)
+
+	out := res.jsonOutput(t)
+	assertHasResponseKey(t, out, "dry-run")
+}