94 lines
1.8 KiB
Go
94 lines
1.8 KiB
Go
|
package protocol
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"reflect"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/aws/aws-sdk-go/aws"
|
||
|
|
)
|
||
|
|
|
||
|
|
var testJSONValueCases = []struct {
|
||
|
|
Value aws.JSONValue
|
||
|
|
Mode EscapeMode
|
||
|
|
String string
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
Value: aws.JSONValue{
|
||
|
|
"abc": 123.,
|
||
|
|
},
|
||
|
|
Mode: NoEscape,
|
||
|
|
String: `{"abc":123}`,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Value: aws.JSONValue{
|
||
|
|
"abc": 123.,
|
||
|
|
},
|
||
|
|
Mode: Base64Escape,
|
||
|
|
String: `eyJhYmMiOjEyM30=`,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Value: aws.JSONValue{
|
||
|
|
"abc": 123.,
|
||
|
|
},
|
||
|
|
Mode: QuotedEscape,
|
||
|
|
String: `"{\"abc\":123}"`,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestEncodeJSONValue(t *testing.T) {
|
||
|
|
for i, c := range testJSONValueCases {
|
||
|
|
str, err := EncodeJSONValue(c.Value, c.Mode)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("%d, expect no error, got %v", i, err)
|
||
|
|
}
|
||
|
|
if e, a := c.String, str; e != a {
|
||
|
|
t.Errorf("%d, expect %v encoded value, got %v", i, e, a)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDecodeJSONValue(t *testing.T) {
|
||
|
|
for i, c := range testJSONValueCases {
|
||
|
|
val, err := DecodeJSONValue(c.String, c.Mode)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("%d, expect no error, got %v", i, err)
|
||
|
|
}
|
||
|
|
if e, a := c.Value, val; !reflect.DeepEqual(e, a) {
|
||
|
|
t.Errorf("%d, expect %v encoded value, got %v", i, e, a)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestEncodeJSONValue_PanicUnkownMode(t *testing.T) {
|
||
|
|
defer func() {
|
||
|
|
if r := recover(); r == nil {
|
||
|
|
t.Errorf("expect panic, got none")
|
||
|
|
} else {
|
||
|
|
reason := fmt.Sprintf("%v", r)
|
||
|
|
if e, a := "unknown EscapeMode", reason; !strings.Contains(a, e) {
|
||
|
|
t.Errorf("expect %q to be in %v", e, a)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
val := aws.JSONValue{}
|
||
|
|
|
||
|
|
EncodeJSONValue(val, 123456)
|
||
|
|
}
|
||
|
|
func TestDecodeJSONValue_PanicUnkownMode(t *testing.T) {
|
||
|
|
defer func() {
|
||
|
|
if r := recover(); r == nil {
|
||
|
|
t.Errorf("expect panic, got none")
|
||
|
|
} else {
|
||
|
|
reason := fmt.Sprintf("%v", r)
|
||
|
|
if e, a := "unknown EscapeMode", reason; !strings.Contains(a, e) {
|
||
|
|
t.Errorf("expect %q to be in %v", e, a)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
DecodeJSONValue(`{"abc":123}`, 123456)
|
||
|
|
}
|