Jake Hillion
ff4ce07b05
All checks were successful
continuous-integration/drone/push Build is passing
94 lines
2.3 KiB
Go
94 lines
2.3 KiB
Go
package proxy
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"mpbl3p/mocks"
|
|
"mpbl3p/shared"
|
|
"testing"
|
|
)
|
|
|
|
func TestPacket_Marshal(t *testing.T) {
|
|
testContent := []byte("A test string is the content of this packet.")
|
|
testPacket := NewSimplePacket(testContent)
|
|
|
|
t.Run("Length", func(t *testing.T) {
|
|
marshalled := testPacket.Marshal()
|
|
|
|
assert.Len(t, marshalled, len(testContent)+8)
|
|
})
|
|
}
|
|
|
|
func TestUnmarshalPacket(t *testing.T) {
|
|
testContent := []byte("A test string is the content of this packet.")
|
|
testPacket := NewSimplePacket(testContent)
|
|
testMarshalled := testPacket.Marshal()
|
|
|
|
t.Run("Length", func(t *testing.T) {
|
|
p, err := UnmarshalSimplePacket(testMarshalled)
|
|
|
|
require.Nil(t, err)
|
|
assert.Len(t, p.Contents(), len(testContent))
|
|
})
|
|
|
|
t.Run("Contents", func(t *testing.T) {
|
|
p, err := UnmarshalSimplePacket(testMarshalled)
|
|
|
|
require.Nil(t, err)
|
|
assert.Equal(t, p.Contents(), testContent)
|
|
})
|
|
}
|
|
|
|
func TestAppendMac(t *testing.T) {
|
|
testContent := []byte("A test string is the content of this packet.")
|
|
testMac := mocks.AlmostUselessMac{}
|
|
testPacket := NewSimplePacket(testContent)
|
|
testMarshalled := testPacket.Marshal()
|
|
|
|
appended := AppendMac(testMarshalled, testMac)
|
|
|
|
t.Run("Length", func(t *testing.T) {
|
|
assert.Len(t, appended, len(testMarshalled)+4)
|
|
})
|
|
|
|
t.Run("Mac", func(t *testing.T) {
|
|
assert.Equal(t, []byte{'a', 'b', 'c', 'd'}, appended[len(testMarshalled):])
|
|
})
|
|
|
|
t.Run("Original", func(t *testing.T) {
|
|
assert.Equal(t, testMarshalled, appended[:len(testMarshalled)])
|
|
})
|
|
}
|
|
|
|
func TestStripMac(t *testing.T) {
|
|
testContent := []byte("A test string is the content of this packet.")
|
|
testMac := mocks.AlmostUselessMac{}
|
|
testPacket := NewSimplePacket(testContent)
|
|
testMarshalled := testPacket.Marshal()
|
|
|
|
appended := AppendMac(testMarshalled, testMac)
|
|
|
|
t.Run("Length", func(t *testing.T) {
|
|
cut, err := StripMac(appended, testMac)
|
|
|
|
require.Nil(t, err)
|
|
assert.Len(t, cut, len(testMarshalled))
|
|
})
|
|
|
|
t.Run("IncorrectMac", func(t *testing.T) {
|
|
badMac := make([]byte, len(testMarshalled)+4)
|
|
copy(badMac, testMarshalled)
|
|
copy(badMac[:len(testMarshalled)], "dcba")
|
|
_, err := StripMac(badMac, testMac)
|
|
|
|
assert.Error(t, err, shared.ErrBadChecksum)
|
|
})
|
|
|
|
t.Run("Original", func(t *testing.T) {
|
|
cut, err := StripMac(appended, testMac)
|
|
|
|
require.Nil(t, err)
|
|
assert.Equal(t, testMarshalled, cut)
|
|
})
|
|
}
|