提交 97a36884 作者: Matt Bell

commands; Fixed tests for Reader output

上级 0419ce1d
package commands package commands
import ( import (
"bytes"
"fmt" "fmt"
"strings" "strings"
"testing" "testing"
...@@ -27,21 +28,25 @@ func TestMarshalling(t *testing.T) { ...@@ -27,21 +28,25 @@ func TestMarshalling(t *testing.T) {
req.SetOption(EncShort, JSON) req.SetOption(EncShort, JSON)
bytes, err := res.Marshal() reader, err := res.Marshal()
if err != nil { if err != nil {
t.Error(err, "Should have passed") t.Error(err, "Should have passed")
} }
output := string(bytes) var buf bytes.Buffer
buf.ReadFrom(reader)
output := buf.String()
if removeWhitespace(output) != "{\"Foo\":\"beep\",\"Bar\":\"boop\",\"Baz\":1337}" { if removeWhitespace(output) != "{\"Foo\":\"beep\",\"Bar\":\"boop\",\"Baz\":1337}" {
t.Error("Incorrect JSON output") t.Error("Incorrect JSON output")
} }
res.SetError(fmt.Errorf("Oops!"), ErrClient) res.SetError(fmt.Errorf("Oops!"), ErrClient)
bytes, err = res.Marshal() reader, err = res.Marshal()
if err != nil { if err != nil {
t.Error("Should have passed") t.Error("Should have passed")
} }
output = string(bytes) buf.Reset()
buf.ReadFrom(reader)
output = buf.String()
fmt.Println(removeWhitespace(output)) fmt.Println(removeWhitespace(output))
if removeWhitespace(output) != "{\"Message\":\"Oops!\",\"Code\":1}" { if removeWhitespace(output) != "{\"Message\":\"Oops!\",\"Code\":1}" {
t.Error("Incorrect JSON output") t.Error("Incorrect JSON output")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论