handshake_protocol_internal_test.go

package avroipc

import (
	"github.com/stretchr/testify/require"
	"testing"
)

func Test_getMD5(t *testing.T) {
	t.Run("empty value", func(t *testing.T) {
		actual := getMD5("")
		expected := []byte{0xd4, 0x1d, 0x8c, 0xd9, 0x8f, 0x0, 0xb2, 0x4, 0xe9, 0x80, 0x9, 0x98, 0xec, 0xf8, 0x42, 0x7e}

		require.Equal(t, expected, actual)
	})

	t.Run("normal value", func(t *testing.T) {
		actual := getMD5("test string")
		expected := []byte{0x6f, 0x8d, 0xb5, 0x99, 0xde, 0x98, 0x6f, 0xab, 0x7a, 0x21, 0x62, 0x5b, 0x79, 0x16, 0x58, 0x9c}

		require.Equal(t, expected, actual)
	})
}

func TestHandshakeProtocol_PrepareRequest(t *testing.T) {
	t.Run("without client protocol", func(t *testing.T) {
		expected := []byte{
			// Client hash.
			0x49, 0x87, 0x43, 0x7b, 0xf5, 0x9, 0xdf, 0xde, 0x62, 0x36, 0x72, 0x99, 0xef, 0x40, 0xc8, 0x2f,
			// Client protocol.
			0x0,
			// Server hash.
			0x49, 0x87, 0x43, 0x7b, 0xf5, 0x9, 0xdf, 0xde, 0x62, 0x36, 0x72, 0x99, 0xef, 0x40, 0xc8, 0x2f,
			// Metadata
			0x0,
			// Empty message.
			0x0, 0x0,
		}

		p, err := NewHandshakeProtocol()
		require.NoError(t, err)

		actual, err := p.PrepareRequest()
		require.NoError(t, err)
		require.Equal(t, expected, actual)
	})

	t.Run("with client protocol", func(t *testing.T) {
		expected := []byte{
			// Client hash.
			0x49, 0x87, 0x43, 0x7b, 0xf5, 0x9, 0xdf, 0xde, 0x62, 0x36, 0x72, 0x99, 0xef, 0x40, 0xc8, 0x2f,
			// Client protocol.
			0x2, 0x8, 0x74, 0x65, 0x73, 0x74,
			// Server hash.
			0x49, 0x87, 0x43, 0x7b, 0xf5, 0x9, 0xdf, 0xde, 0x62, 0x36, 0x72, 0x99, 0xef, 0x40, 0xc8, 0x2f,
			// Metadata
			0x0,
			// Empty message.
			0x0, 0x0,
		}

		p, err := NewHandshakeProtocol()
		require.NoError(t, err)

		h := p.(*handshakeProtocol)
		h.clientProtocol = "test"
		h.needClientProtocol = true

		actual, err := p.PrepareRequest()
		require.NoError(t, err)
		require.Equal(t, expected, actual)
	})
}

func TestHandshakeProtocol_ProcessResponse(t *testing.T) {
	t.Run("bad match", func(t *testing.T) {
		response := []byte{
			// Match.
			0x7,
		}

		p, err := NewHandshakeProtocol()
		require.NoError(t, err)

		needResend, err := p.ProcessResponse(response)
		require.Error(t, err)
		require.Contains(t, err.Error(), "cannot decode binary enum")
		require.False(t, needResend)
	})

	t.Run("short buffer", func(t *testing.T) {
		response := []byte{
			// Match.
			0x0,
		}

		p, err := NewHandshakeProtocol()
		require.NoError(t, err)

		needResend, err := p.ProcessResponse(response)
		require.Error(t, err)
		require.Contains(t, err.Error(), "short buffer")
		require.False(t, needResend)
	})

	t.Run("both match", func(t *testing.T) {
		response := []byte{
			// Match.
			0x0,
			// Server protocol.
			0x0,
			// Server hash.
			0x0,
			// Metadata
			0x0,
		}

		p, err := NewHandshakeProtocol()
		require.NoError(t, err)

		needResend, err := p.ProcessResponse(response)
		require.NoError(t, err)
		require.False(t, needResend)
	})

	t.Run("none match", func(t *testing.T) {
		response := []byte{
			// Match.
			0x4,
			// Server protocol.
			0x0,
			// Server hash.
			0x2, 0x49, 0x87, 0x43, 0x7b, 0xf5, 0x9, 0xdf, 0xde, 0x62, 0x36, 0x72, 0x99, 0xef, 0x40, 0xc8, 0x2f,
			// Metadata
			0x0,
		}

		p, err := NewHandshakeProtocol()
		require.NoError(t, err)

		needResend, err := p.ProcessResponse(response)
		require.NoError(t, err)
		require.True(t, needResend)
	})

	t.Run("repeated none match", func(t *testing.T) {
		response := []byte{
			// Match.
			0x4,
			// Server protocol.
			0x0,
			// Server hash.
			0x2, 0x49, 0x87, 0x43, 0x7b, 0xf5, 0x9, 0xdf, 0xde, 0x62, 0x36, 0x72, 0x99, 0xef, 0x40, 0xc8, 0x2f,
			// Metadata
			0x0,
		}

		p, err := NewHandshakeProtocol()
		require.NoError(t, err)

		h := p.(*handshakeProtocol)
		h.needClientProtocol = true

		needResend, err := p.ProcessResponse(response)
		require.Error(t, err)
		require.Contains(t, err.Error(), "unknown client's protocol")
		require.False(t, needResend)
	})

	t.Run("client match", func(t *testing.T) {
		response := []byte{
			// Match.
			0x2,
			// Server protocol.
			0x2, 0x8, 0x74, 0x65, 0x73, 0x74,
			// Server hash.
			0x2, 0x49, 0x87, 0x43, 0x7b, 0xf5, 0x9, 0xdf, 0xde, 0x62, 0x36, 0x72, 0x99, 0xef, 0x40, 0xc8, 0x2f,
			// Metadata
			0x0,
		}

		p, err := NewHandshakeProtocol()
		require.NoError(t, err)

		needResend, err := p.ProcessResponse(response)
		require.NoError(t, err)
		require.False(t, needResend)
	})

	t.Run("bad client match order", func(t *testing.T) {
		response := []byte{
			// Match.
			0x2,
			// Server protocol.
			0x2, 0x8, 0x74, 0x65, 0x73, 0x74,
			// Server hash.
			0x2, 0x49, 0x87, 0x43, 0x7b, 0xf5, 0x9, 0xdf, 0xde, 0x62, 0x36, 0x72, 0x99, 0xef, 0x40, 0xc8, 0x2f,
			// Metadata
			0x0,
		}

		p, err := NewHandshakeProtocol()
		require.NoError(t, err)

		h := p.(*handshakeProtocol)
		h.needClientProtocol = true

		needResend, err := p.ProcessResponse(response)
		require.Error(t, err)
		require.Contains(t, err.Error(), "unknown client's protocol")
		require.False(t, needResend)
	})
}