message_encode_test.go

package irc

import (
	"io/ioutil"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestEncodeIgnoreRaw(t *testing.T) {
	raw := sadCrab
	m, err := ParseMessage(raw)
	if assert.NoError(t, err) {
		m.Raw = "THIS SHOULD NOT BE USED"

		ms := m.String()
		assert.Equal(t, raw, ms, "encoded message should be equal")
		assert.Equal(t, len(ms), m.Len(), "encoded message length should be equal")
	}
}

func TestStringBytesEqual(t *testing.T) {
	raw := sadCrab
	m, err := ParseMessage(raw)
	if assert.NoError(t, err) {
		ms := m.String()
		mb := m.Bytes()

		assert.Equal(t, ms, string(mb), "encoded message should be equal")
	}
}

func TestMarshalText(t *testing.T) {
	m, err := ParseMessage(sadCrab)
	assert.NoError(t, err)

	expected := m.Bytes()
	got, err := m.MarshalText()
	assert.NoError(t, err)

	assert.Equal(t, expected, got)
}

func TestUnmarshalText(t *testing.T) {
	expected, err := ParseMessage(sadCrab)
	assert.NoError(t, err)

	var m Message
	err = m.UnmarshalText([]byte(sadCrab))
	assert.NoError(t, err)

	assert.Equal(t, expected, &m)
}

func BenchmarkLen(b *testing.B) {
	m, err := ParseMessage(rawTwitch)
	if err != nil {
		b.Fatal(err)
	}

	b.SetBytes(int64(len(rawTwitch)))
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		m.Len()
	}
}

func BenchmarkEncodeBuffer(b *testing.B) {
	m, err := ParseMessage(rawTwitch)
	if err != nil {
		b.Fatal(err)
	}

	b.SetBytes(int64(len(rawTwitch)))
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		buf := m.buffer()
		bufferPool.Put(buf)
	}
}

func BenchmarkEncodeWriteTo(b *testing.B) {
	m, err := ParseMessage(rawTwitch)
	if err != nil {
		b.Fatal(err)
	}

	b.SetBytes(int64(len(rawTwitch)))
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		m.WriteToWithNewline(ioutil.Discard) //nolint:errcheck
	}
}

func BenchmarkEncodeBytes(b *testing.B) {
	m, err := ParseMessage(rawTwitch)
	if err != nil {
		b.Fatal(err)
	}

	b.SetBytes(int64(len(rawTwitch)))
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		m.Bytes()
	}
}

func BenchmarkEncodeString(b *testing.B) {
	m, err := ParseMessage(rawTwitch)
	if err != nil {
		b.Fatal(err)
	}

	b.SetBytes(int64(len(rawTwitch)))
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		m.String()
	}
}