go-i2p/lib/i2np/i2np_test.go

package i2np

import (
	"github.com/hkparker/go-i2p/lib/common"
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestReadI2NPTypeWithNoData(t *testing.T) {
	assert := assert.New(t)

	mtype, err := ReadI2NPType([]byte{})
	assert.Equal(0, mtype)
	assert.Equal(ERR_I2NP_NOT_ENOUGH_DATA, err)
}

func TestReadI2NPTypeWithValidData(t *testing.T) {
	assert := assert.New(t)

	mtype, err := ReadI2NPType([]byte{0x01})
	assert.Equal(1, mtype)
	assert.Nil(err)
}

func TestReadI2NPNTCPMessageIDWithMissingData(t *testing.T) {
	assert := assert.New(t)

	mid, err := ReadI2NPNTCPMessageID([]byte{0x00, 0x00, 0x00, 0x00})
	assert.Equal(0, mid)
	assert.Equal(ERR_I2NP_NOT_ENOUGH_DATA, err)
}

func TestReadI2NPNTCPMessageIDWithValidData(t *testing.T) {
	assert := assert.New(t)

	mid, err := ReadI2NPNTCPMessageID([]byte{0x01, 0x00, 0x00, 0x00, 0x01})
	assert.Equal(1, mid)
	assert.Nil(err)
}

func TestReadI2NPNTCPMessageDateWithMissingData(t *testing.T) {
	assert := assert.New(t)

	date, err := ReadI2NPNTCPMessageDate([]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00})
	assert.Equal(common.Date{}, date)
	assert.Equal(ERR_I2NP_NOT_ENOUGH_DATA, err)
}

func TestReadI2NPNTCPMessageDateWithValidData(t *testing.T) {
	assert := assert.New(t)

	date, err := ReadI2NPNTCPMessageDate([]byte{0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x05, 0x26, 0x5c, 0x00})
	assert.Equal(int64(86400), date.Time().Unix())
	assert.Nil(err)
}

func TestReadI2NPNTCPMessageSizeWithMissingData(t *testing.T) {
	assert := assert.New(t)

	size, err := ReadI2NPNTCPMessageSize([]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00})
	assert.Equal(0, size)
	assert.Equal(ERR_I2NP_NOT_ENOUGH_DATA, err)
}

func TestReadI2NPNTCPMessageSizeWithValidData(t *testing.T) {
	assert := assert.New(t)

	size, err := ReadI2NPNTCPMessageSize([]byte{0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x05, 0x26, 0x5c, 0x00, 0x00, 0x01})
	assert.Equal(1, size)
	assert.Nil(err)
}

func TestReadI2NPNTCPMessageChecksumWithMissingData(t *testing.T) {
	assert := assert.New(t)

	checksum, err := ReadI2NPNTCPMessageChecksum([]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00})
	assert.Equal(0, checksum)
	assert.Equal(ERR_I2NP_NOT_ENOUGH_DATA, err)
}

func TestReadI2NPNTCPMessageChecksumWithValidData(t *testing.T) {
	assert := assert.New(t)

	checksum, err := ReadI2NPNTCPMessageChecksum([]byte{0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x05, 0x26, 0x5c, 0x00, 0x00, 0x01, 0x01})
	assert.Equal(1, checksum)
	assert.Nil(err)
}