encode.time_test.go

package msgpack

import (
	"bytes"
	"fmt"
	"testing"
	"time"

	"github.com/blugnu/test"
)

func TestTimeEncoders(t *testing.T) {
	// ARRANGE
	ept := time.Date(1970, 1, 1, 0, 0, 0, 0, time.UTC)
	buf := &bytes.Buffer{}
	enc := Encoder{out: buf}

	// ARRANGE
	testcases := []struct {
		name string
		t    time.Time
		ts   string
		sut  func(time.Time) error
		test func(error)
	}{
		{name: "EncodeRFC3339", t: ept, sut: func(tt time.Time) error { return enc.EncodeRFC3339(tt) }, test: func(error) {
			test.BytesEqual(t, append([]byte{maskFixString | 20}, []byte("1970-01-01T00:00:00Z")...), buf.Bytes())
		}},
		{name: "EncodeRFC3339Micro", t: ept, sut: func(tt time.Time) error { return enc.EncodeRFC3339Micro(tt) }, test: func(error) {
			test.BytesEqual(t, append([]byte{maskFixString | 27}, []byte("1970-01-01T00:00:00.000000Z")...), buf.Bytes(), test.BytesString)
		}},
		{name: "EncodeRFC3339",
			t:   time.Date(2106, 02, 07, 06, 28, 16, 0, time.UTC),
			sut: func(tt time.Time) error { return enc.EncodeRFC3339(tt) }, test: func(error) {
				test.BytesEqual(t, append([]byte{maskFixString | 20}, []byte("2106-02-07T06:28:16Z")...), buf.Bytes())
			}},
		{name: "encodeTimestamp (max ts32)",
			t: time.Unix(2<<31-1, 0).UTC(),
			sut: func(tt time.Time) error {
				test.Equal(t, "2106-02-07T06:28:15Z", tt.Format(time.RFC3339))
				return encodeTimestamp(&enc, tt)
			},
			test: func(error) {
				test.BytesEqual(t, []byte{typeFixExt4, 0xff, 0xff, 0xff, 0xff, 0xff}, buf.Bytes())
			}},
		{name: "encodeTimestamp (min ts64)",
			t: time.Unix(2<<31, 0).UTC(),
			sut: func(tt time.Time) error {
				test.Equal(t, "2106-02-07T06:28:16Z", tt.Format(time.RFC3339))
				return encodeTimestamp(&enc, tt)
			},
			test: func(error) {
				test.BytesEqual(t, []byte{typeFixExt8, 0xff, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00}, buf.Bytes())
			}},
		{name: "encodeTimestamp (max ts64)",
			t: time.Unix(2<<33-1, 0).UTC(),
			sut: func(tt time.Time) error {
				test.Equal(t, "2514-05-30T01:53:03Z", tt.Format(time.RFC3339))
				return encodeTimestamp(&enc, tt)
			},
			test: func(error) {
				test.BytesEqual(t, []byte{typeFixExt8, 0xff, 0x00, 0x00, 0x00, 0x03, 0xff, 0xff, 0xff, 0xff}, buf.Bytes())
			}},
		{name: "encodeTimestamp (min pos ts96)",
			t: time.Unix(2<<33, 0).UTC(),
			sut: func(tt time.Time) error {
				test.Equal(t, "2514-05-30T01:53:04Z", tt.Format(time.RFC3339))
				return encodeTimestamp(&enc, tt)
			},
			test: func(error) {
				test.BytesEqual(t, []byte{typeExt8, 12, 0xff, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x00}, buf.Bytes())
			}},
		{name: "encodeTimestamp (max ts96)",
			t: time.Unix(2<<62-1, 999999999).UTC(),
			sut: func(tt time.Time) error {
				test.Equal(t, "292277026596-12-04T15:30:07.999999999Z", tt.Format(time.RFC3339Nano))
				return encodeTimestamp(&enc, tt)
			},
			test: func(error) {
				test.BytesEqual(t, []byte{typeExt8, 12, 0xff, 0x2a, 0x4a, 0xe5, 0xff, 0x7f, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff}, buf.Bytes())
			}},
	}
	for _, tc := range testcases {
		t.Run(fmt.Sprintf("%s(%s)", tc.name, tc.t), func(t *testing.T) {
			// ARRANGE
			buf.Reset()

			// ACT
			err := tc.sut(tc.t)

			// ASSERT
			tc.test(err)
		})
	}
}