forked from TrueCloudLab/frostfs-s3-gw
68 lines
1.3 KiB
Go
68 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)
|
|
})
|
|
}
|