package prox5

import "testing"

func Test_filter(t *testing.T) {
	type args struct {
		in string
	}
	type test struct {
		name         string
		args         args
		wantFiltered string
		wantOk       bool
	}
	var tests = []test{
		{
			name: "simple",
			args: args{
				in: "127.0.0.1:1080",
			},
			wantFiltered: "127.0.0.1:1080",
			wantOk:       true,
		},
		{
			name: "withAuth",
			args: args{
				in: "127.0.0.1:1080:user:pass",
			},
			wantFiltered: "user:pass@127.0.0.1:1080",
			wantOk:       true,
		},
		{
			name: "withAuthAlt",
			args: args{
				in: "user:pass@127.0.0.1:1080",
			},
			wantFiltered: "user:pass@127.0.0.1:1080",
			wantOk:       true,
		},
		{
			name: "simpleDomain",
			args: args{
				in: "yeet.com:1080",
			},
			wantFiltered: "yeet.com:1080",
			wantOk:       true,
		},
		{
			name: "domainWithAuth",
			args: args{
				in: "yeet.com:1080:user:pass",
			},
			wantFiltered: "user:pass@yeet.com:1080",
			wantOk:       true,
		},
		{
			name: "ipv6",
			args: args{
				in: "[fe80::2ef0:5dff:fe7f:c299]:1080",
			},
			wantFiltered: "[fe80::2ef0:5dff:fe7f:c299]:1080",
			wantOk:       true,
		},
		{
			name: "ipv6WithAuth",
			args: args{
				in: "[fe80::2ef0:5dff:fe7f:c299]:1080:user:pass",
			},
			wantFiltered: "user:pass@[fe80::2ef0:5dff:fe7f:c299]:1080",
			wantOk:       true,
		},
		{
			name: "invalid",
			args: args{
				in: "yeet",
			},
			wantFiltered: "",
			wantOk:       false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			gotFiltered, gotOk := filter(tt.args.in)
			if gotFiltered != tt.wantFiltered {
				t.Errorf("filter() gotFiltered = %v, want %v", gotFiltered, tt.wantFiltered)
			}
			if gotOk != tt.wantOk {
				t.Errorf("filter() gotOk = %v, want %v", gotOk, tt.wantOk)
			}
		})
	}
}