string_test.go

package sbm

import (
	"testing"

	"github.com/vault-thirteen/auxie/tester"
)

func Test_trimHeader(t *testing.T) {
	var tst = tester.New(t)
	var err error
	var rawHeader []byte
	var result []byte
	var resultExpected []byte

	// Test #1. Normal Test.
	rawHeader = append([]byte("ABC"), '\r', '\n')
	resultExpected = []byte("ABC")
	result, err = trimHeader(rawHeader)
	tst.MustBeNoError(err)
	tst.MustBeEqual(result, resultExpected)

	// Test #2. The last Symbol is not LF.
	rawHeader = append([]byte("ABC"), '\r', '\r')
	resultExpected = []byte(nil)
	result, err = trimHeader(rawHeader)
	tst.MustBeAnError(err)
	tst.MustBeEqual(err.Error(), ErrHeaderEnding)
	tst.MustBeEqual(result, resultExpected)

	// Test #3. The pre-last Symbol is not CR.
	rawHeader = append([]byte("ABC"), '\n', '\n')
	resultExpected = []byte(nil)
	result, err = trimHeader(rawHeader)
	tst.MustBeAnError(err)
	tst.MustBeEqual(err.Error(), ErrHeaderEnding)
	tst.MustBeEqual(result, resultExpected)

	// Test #4. The Header is too short.
	rawHeader = []byte{'\r'}
	resultExpected = []byte(nil)
	result, err = trimHeader(rawHeader)
	tst.MustBeAnError(err)
	tst.MustBeEqual(err.Error(), ErrHeaderSize)
	tst.MustBeEqual(result, resultExpected)

	// Test #5. Empty Header.
	rawHeader = []byte{'\r', '\n'}
	resultExpected = []byte{}
	result, err = trimHeader(rawHeader)
	tst.MustBeNoError(err)
	tst.MustBeEqual(result, resultExpected)
}