115 lines
2.1 KiB
Go
115 lines
2.1 KiB
Go
package origin
|
||
|
||
import (
|
||
"net/http/httptest"
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/require"
|
||
)
|
||
|
||
func TestPatternCheck(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
testCases := []struct {
|
||
name string
|
||
origin string
|
||
originPatterns []string
|
||
success bool
|
||
}{
|
||
{
|
||
name: "empty_origin",
|
||
origin: "",
|
||
originPatterns: []string{
|
||
"*.example.com",
|
||
},
|
||
success: true,
|
||
},
|
||
{
|
||
name: "origin_patterns_match",
|
||
origin: "https://two.Example.com",
|
||
originPatterns: []string{
|
||
"*.example.com",
|
||
"foo.com",
|
||
},
|
||
success: true,
|
||
},
|
||
{
|
||
name: "origin_patterns_no_match",
|
||
origin: "https://two.Example.com",
|
||
originPatterns: []string{
|
||
"foo.com",
|
||
"bar.com",
|
||
},
|
||
success: false,
|
||
},
|
||
{
|
||
name: "origin_patterns_uppercase_in_pattern",
|
||
origin: "https://two.example.com",
|
||
originPatterns: []string{
|
||
"https://*.Example.com",
|
||
},
|
||
success: true,
|
||
},
|
||
{
|
||
name: "origin_patterns_uppercase_in_pattern_and_origin",
|
||
origin: "HTTPS://App.Example.com",
|
||
originPatterns: []string{
|
||
"https://APP.example.COM",
|
||
},
|
||
success: true,
|
||
},
|
||
{
|
||
name: "origin_patterns_uppercase_in_pattern_no_match",
|
||
origin: "https://two.example.org",
|
||
originPatterns: []string{
|
||
"https://*.Example.com",
|
||
},
|
||
success: false,
|
||
},
|
||
{
|
||
name: "origin_patterns_cyrillic_e_in_origin",
|
||
origin: "https://two.еxample.com",
|
||
originPatterns: []string{
|
||
"*.example.com",
|
||
"foo.com",
|
||
},
|
||
success: false,
|
||
},
|
||
{
|
||
name: "file_origin",
|
||
origin: "file://",
|
||
originPatterns: []string{
|
||
"file://*",
|
||
},
|
||
success: true,
|
||
},
|
||
{
|
||
name: "null_origin",
|
||
origin: "null",
|
||
originPatterns: []string{
|
||
"null",
|
||
},
|
||
success: true,
|
||
},
|
||
}
|
||
|
||
for _, tc := range testCases {
|
||
tc := tc
|
||
t.Run(tc.name, func(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
r := httptest.NewRequest("GET", "https://example.com/websocket/connection", nil)
|
||
r.Header.Set("Origin", tc.origin)
|
||
|
||
a, err := NewPatternChecker(tc.originPatterns)
|
||
require.NoError(t, err)
|
||
ok := a.Check(r)
|
||
if tc.success {
|
||
require.True(t, ok)
|
||
} else {
|
||
require.False(t, ok)
|
||
}
|
||
})
|
||
}
|
||
}
|