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)
	}
}