package proxyd import ( "testing" "github.com/stretchr/testify/assert" ) func TestStripXFF(t *testing.T) { tests := []struct { in, out string }{ {"1.2.3, 4.5.6, 7.8.9", "1.2.3"}, {"1.2.3,4.5.6", "1.2.3"}, {" 1.2.3 , 4.5.6 ", "1.2.3"}, } for _, test := range tests { actual := stripXFF(test.in) assert.Equal(t, test.out, actual) } }