69 lines
1.3 KiB
Go
69 lines
1.3 KiB
Go
|
package api
|
||
|
|
||
|
import (
|
||
|
"net/http"
|
||
|
"net/url"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/stretchr/testify/require"
|
||
|
)
|
||
|
|
||
|
func TestFilter(t *testing.T) {
|
||
|
key1, val1 := "key1", "val1"
|
||
|
key2, val2 := "key2", "val2"
|
||
|
key3, val3 := "key3", "val3"
|
||
|
|
||
|
anyVal := ""
|
||
|
|
||
|
notNilHandler := http.HandlerFunc(func(http.ResponseWriter, *http.Request) {})
|
||
|
|
||
|
t.Run("queries", func(t *testing.T) {
|
||
|
f := NewHandlerFilter().
|
||
|
Add(NewFilter().
|
||
|
QueriesMatch(key1, val1, key2, anyVal).
|
||
|
Queries(key3).
|
||
|
Handler(notNilHandler))
|
||
|
|
||
|
r, err := http.NewRequest(http.MethodGet, "https://localhost:8084", nil)
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
query := make(url.Values)
|
||
|
query.Set(key1, val1)
|
||
|
query.Set(key2, val2)
|
||
|
query.Set(key3, val3)
|
||
|
r.URL.RawQuery = query.Encode()
|
||
|
|
||
|
h := f.match(r)
|
||
|
require.NotNil(t, h)
|
||
|
|
||
|
query.Set(key1, val2)
|
||
|
r.URL.RawQuery = query.Encode()
|
||
|
|
||
|
h = f.match(r)
|
||
|
require.Nil(t, h)
|
||
|
})
|
||
|
|
||
|
t.Run("headers", func(t *testing.T) {
|
||
|
f := NewHandlerFilter().
|
||
|
Add(NewFilter().
|
||
|
HeadersMatch(key1, val1, key2, anyVal).
|
||
|
Headers(key3).
|
||
|
Handler(notNilHandler))
|
||
|
|
||
|
r, err := http.NewRequest(http.MethodGet, "https://localhost:8084", nil)
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
r.Header.Set(key1, val1)
|
||
|
r.Header.Set(key2, val2)
|
||
|
r.Header.Set(key3, val3)
|
||
|
|
||
|
h := f.match(r)
|
||
|
require.NotNil(t, h)
|
||
|
|
||
|
r.Header.Set(key1, val2)
|
||
|
|
||
|
h = f.match(r)
|
||
|
require.Nil(t, h)
|
||
|
})
|
||
|
}
|