99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
|
package playback
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
var errNoMatches = errors.New("no matches found")
|
||
|
|
|
||
|
|
func withoutValue(data map[string]string, field string) map[string]string {
|
||
|
|
result := make(map[string]string)
|
||
|
|
for k, v := range data {
|
||
|
|
result[k] = v
|
||
|
|
}
|
||
|
|
result[field] = ""
|
||
|
|
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParseAuthHeader(t *testing.T) {
|
||
|
|
defaultHeader := "AWS4-HMAC-SHA256 Credential=oid0cid/20210809/us-east-1/s3/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date, Signature=2811ccb9e242f41426738fb1f"
|
||
|
|
|
||
|
|
defaultAuthInfo := map[string]string{
|
||
|
|
"access_key_id": "oid0cid",
|
||
|
|
"service": "s3",
|
||
|
|
"region": "us-east-1",
|
||
|
|
"v4_signature": "2811ccb9e242f41426738fb1f",
|
||
|
|
"signed_header_fields": "host;x-amz-content-sha256;x-amz-date",
|
||
|
|
"date": "20210809",
|
||
|
|
}
|
||
|
|
for _, tc := range []struct {
|
||
|
|
title string
|
||
|
|
header string
|
||
|
|
err error
|
||
|
|
expected map[string]string
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
title: "correct full header",
|
||
|
|
header: defaultHeader,
|
||
|
|
err: nil,
|
||
|
|
expected: defaultAuthInfo,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
title: "correct with empty region",
|
||
|
|
header: strings.Replace(defaultHeader, "/us-east-1/", "//", 1),
|
||
|
|
err: nil,
|
||
|
|
expected: withoutValue(defaultAuthInfo, "region"),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
title: "empty access key",
|
||
|
|
header: strings.Replace(defaultHeader, "oid0cid", "", 1),
|
||
|
|
err: errNoMatches,
|
||
|
|
expected: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
title: "empty service",
|
||
|
|
header: strings.Replace(defaultHeader, "/s3/", "//", 1),
|
||
|
|
err: errNoMatches,
|
||
|
|
expected: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
title: "empty date",
|
||
|
|
header: strings.Replace(defaultHeader, "/20210809/", "//", 1),
|
||
|
|
err: errNoMatches,
|
||
|
|
expected: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
title: "empty v4_signature",
|
||
|
|
header: strings.Replace(defaultHeader, "Signature=2811ccb9e242f41426738fb1f",
|
||
|
|
"Signature=", 1),
|
||
|
|
err: errNoMatches,
|
||
|
|
expected: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
title: "empty signed_fields",
|
||
|
|
header: strings.Replace(defaultHeader, "SignedHeaders=host;x-amz-content-sha256;x-amz-date",
|
||
|
|
"SignedHeaders=", 1),
|
||
|
|
err: errNoMatches,
|
||
|
|
expected: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
title: "empty signed_fields",
|
||
|
|
header: strings.Replace(defaultHeader, "SignedHeaders=host;x-amz-content-sha256;x-amz-date",
|
||
|
|
"SignedHeaders=", 1),
|
||
|
|
err: errNoMatches,
|
||
|
|
expected: nil,
|
||
|
|
},
|
||
|
|
} {
|
||
|
|
t.Run(tc.title, func(t *testing.T) {
|
||
|
|
authInfo, err := parseAuthHeader(tc.header)
|
||
|
|
require.Equal(t, err, tc.err, tc.header)
|
||
|
|
require.Equal(t, tc.expected, authInfo, tc.header)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|