diff options
| -rw-r--r-- | securecookie.go | 16 | ||||
| -rw-r--r-- | securecookie_test.go | 13 |
2 files changed, 25 insertions, 4 deletions
diff --git a/securecookie.go b/securecookie.go index 892b412..2f34bfe 100644 --- a/securecookie.go +++ b/securecookie.go @@ -357,12 +357,17 @@ func CodecsFromPairs(keyPairs ...[]byte) []Codec { return codecs } +var errNoCodecs = errors.New("securecookie: no codecs provided") + // EncodeMulti encodes a cookie value using a group of codecs. // // The codecs are tried in order. Multiple codecs are accepted to allow // key rotation. -func EncodeMulti(name string, value interface{}, - codecs ...Codec) (string, error) { +func EncodeMulti(name string, value interface{}, codecs ...Codec) (string, error) { + if len(codecs) == 0 { + return "", errNoCodecs + } + var errors MultiError for _, codec := range codecs { if encoded, err := codec.Encode(name, value); err == nil { @@ -378,8 +383,11 @@ func EncodeMulti(name string, value interface{}, // // The codecs are tried in order. Multiple codecs are accepted to allow // key rotation. -func DecodeMulti(name string, value string, dst interface{}, - codecs ...Codec) error { +func DecodeMulti(name string, value string, dst interface{}, codecs ...Codec) error { + if len(codecs) == 0 { + return errNoCodecs + } + var errors MultiError for _, codec := range codecs { if err := codec.Decode(name, value, dst); err == nil { diff --git a/securecookie_test.go b/securecookie_test.go index 099c290..fe0cdb1 100644 --- a/securecookie_test.go +++ b/securecookie_test.go @@ -144,6 +144,19 @@ func TestMultiError(t *testing.T) { } } +func TestMultiNoCodecs(t *testing.T) { + _, err := EncodeMulti("foo", "bar") + if err != errNoCodecs { + t.Errorf("EncodeMulti: bad value for error, got: %v", err) + } + + var dst []byte + err = DecodeMulti("foo", "bar", &dst) + if err != errNoCodecs { + t.Errorf("DecodeMulti: bad value for error, got: %v", err) + } +} + // ---------------------------------------------------------------------------- type FooBar struct { |