1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
package proto_test
import (
"bytes"
"testing"
"github.com/mesos/mesos-go/api/v1/lib"
"github.com/mesos/mesos-go/api/v1/lib/encoding"
. "github.com/mesos/mesos-go/api/v1/lib/encoding/proto"
)
type FakeMessage string
func (f *FakeMessage) Marshal() ([]byte, error) { return ([]byte)(*f), nil }
func (f *FakeMessage) MarshalJSON() ([]byte, error) { return nil, nil }
func TestEncoder(t *testing.T) {
// write a proto message, validate that we're actually marshaling proto
buf := bytes.Buffer{}
enc := NewEncoder(encoding.SinkWriter(&buf))
err := enc.Encode(&mesos.FrameworkID{Value: "hello"})
if err != nil {
t.Fatal(err)
}
data := string(buf.Bytes())
if data != "\n\x05hello" {
// \n == 00001 010 == {field-id} {tag-type}
// \x05 == length of the string
t.Fatalf("expected `hello` instead of %q", data)
}
// write a non-proto message, verify panic
caughtPanic := false
func() {
defer func() {
if x := recover(); x != nil {
caughtPanic = true
}
}()
m := FakeMessage("hello")
enc.Encode(&m)
t.Fatal("expected panic, but Encode completed normally")
}()
if !caughtPanic {
t.Fatal("Encode failed to complete normally, but we didn't see a panic? should never happen")
}
}
|
