61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
package bencode
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
)
|
|
|
|
type testStruct struct {
|
|
Field1 string `bencode:"my field1"`
|
|
Field2 int64 `bencode:"my field2"`
|
|
Field3 int64 `bencode:"my field3,omitempty"`
|
|
}
|
|
|
|
type testOldTag struct {
|
|
Field1 string "my field1"
|
|
Field2 int64 "my field2"
|
|
}
|
|
|
|
func TestMarshalling(t *testing.T) {
|
|
ts1 := testStruct{Field1: "foo", Field2: 123456}
|
|
buf, err := Marshal(ts1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if bytes.Contains(buf, []byte("omitempty")) || bytes.Contains(buf, []byte("field3")) {
|
|
t.Errorf("should not have the string 'omitempty' or 'field3' in %q", buf)
|
|
}
|
|
|
|
ts2 := testStruct{}
|
|
err = Unmarshal(buf, &ts2)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if ts1.Field1 != ts2.Field1 {
|
|
t.Errorf("expected %q, got %q", ts1.Field1, ts2.Field1)
|
|
}
|
|
if ts1.Field2 != ts2.Field2 {
|
|
t.Errorf("expected %q, got %q", ts1.Field2, ts2.Field2)
|
|
}
|
|
}
|
|
|
|
func TestOldMarshalling(t *testing.T) {
|
|
ts1 := testOldTag{"foo", 123456}
|
|
buf, err := Marshal(ts1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
ts2 := testStruct{}
|
|
err = Unmarshal(buf, &ts2)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if ts1.Field1 != ts2.Field1 {
|
|
t.Errorf("expected %q, got %q", ts1.Field1, ts2.Field1)
|
|
}
|
|
if ts1.Field2 != ts2.Field2 {
|
|
t.Errorf("expected %q, got %q", ts1.Field2, ts2.Field2)
|
|
}
|
|
}
|