proxy_test.go

package proxykit_test

import (
	"strings"
	"testing"

	"github.com/colduction/proxykit-go"
)

var _ proxykit.ProxyValidator = (*proxykit.Proxy)(nil)

func TestIsValidHostnamePort(t *testing.T) {
	tests := []struct {
		address string
		valid   bool
	}{
		{address: "proxy.example.com:8080", valid: true},
		{address: "127.0.0.1:1", valid: true},
		{address: "192.0.2.10:8080", valid: true},
		{address: "[::1]:80", valid: true},
		{address: "[2001:db8::1]:443", valid: true},
		{address: "proxy-name:65535", valid: true},
		{address: ":80", valid: true},
		{address: "proxy.example.com:0", valid: false},
		{address: "proxy.example.com:65536", valid: false},
		{address: "proxy..example.com:80", valid: false},
		{address: "proxy_example.com:80", valid: false},
		{address: strings.Repeat("a", 64) + ":80", valid: false},
		{address: "::1:80", valid: false},
	}
	for _, test := range tests {
		if got := proxykit.IsValidHostnamePort(test.address); got != test.valid {
			t.Errorf("IsValidHostnamePort(%q) = %v, want %v", test.address, got, test.valid)
		}
	}
}

func TestProxyValidatorMethods(t *testing.T) {
	proxy := &proxykit.Proxy{
		Scheme:   proxykit.HTTP,
		Host:     "proxy.example.com:8080",
		Username: "user",
		Password: "pass",
	}
	if !proxy.IsValidScheme() {
		t.Error("IsValidScheme() = false, want true")
	}
	if !proxy.IsValidHostnamePort() {
		t.Error("IsValidHostnamePort() = false, want true")
	}
	if !proxy.IsValidCredentials() {
		t.Error("IsValidCredentials() = false, want true")
	}

	var nilProxy *proxykit.Proxy
	if nilProxy.IsValidScheme() ||
		nilProxy.IsValidHostnamePort() ||
		nilProxy.IsValidCredentials() {
		t.Error("nil proxy validation methods must report false")
	}
}