mtg/mtglib/internal/obfuscated2/handshake_frame_internal_test.go

package obfuscated2

import (
	"encoding/base64"
	"testing"

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

type HandshakeFrameTestSuite struct {
	suite.Suite
}

func (suite *HandshakeFrameTestSuite) Decode(value string) []byte {
	v, err := base64.RawStdEncoding.DecodeString(value)
	suite.NoError(err)

	return v
}

func (suite *HandshakeFrameTestSuite) Encode(value []byte) string {
	return base64.RawStdEncoding.EncodeToString(value)
}

func (suite *HandshakeFrameTestSuite) TestOk() {
	hf := handshakeFrame{}
	testFrame := suite.Decode(
		"L9TmCzzxl9bPKODBpZeVM/qqNUxQ/axxBup1S2ymbIfUd6f7YSyzzM9EmTFv2/XzGqJGEHuj2zofmUGBLghu5g")
	copy(hf.data[:], testFrame)

	suite.Equal("zyjgwaWXlTP6qjVMUP2scQbqdUtspmyH1Hen+2Ess8w", suite.Encode(hf.key()))
	suite.Equal("z0SZMW/b9fMaokYQe6PbOg", suite.Encode(hf.iv()))
	suite.Equal("H5lBgQ", suite.Encode(hf.connectionType()))
	suite.EqualValues(2094, hf.dc())

	inverted := hf.invert()
	suite.Equal("OtujexBGohrz9dtvMZlEz8yzLGH7p3fUh2ymbEt16gY", suite.Encode(inverted.key()))
	suite.Equal("caz9UEw1qvozlZelweAozw", suite.Encode(inverted.iv()))
	suite.Equal("H5lBgQ", suite.Encode(inverted.connectionType()))
	suite.EqualValues(2094, inverted.dc())
}

func TestHandshakeFrame(t *testing.T) {
	t.Parallel()
	suite.Run(t, &HandshakeFrameTestSuite{})
}