Add tests for obfuscated2 clientside

mtglib/internal/obfuscated2/client_handshake_test.go

@@ -0,0 +1,85 @@
+package obfuscated2_test
+
+import (
+	"bytes"
+	"testing"
+
+	"github.com/9seconds/mtg/v2/mtglib/internal/obfuscated2"
+	"github.com/9seconds/mtg/v2/testlib"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
+	"github.com/stretchr/testify/suite"
+)
+
+type ClientHandshakeTestSuite struct {
+	suite.Suite
+	SnapshotTestSuite
+}
+
+func (suite *ClientHandshakeTestSuite) SetupSuite() {
+	suite.NoError(suite.IngestSnapshots("client_snapshots", "snapshot-ok-"))
+}
+
+func (suite *ClientHandshakeTestSuite) TestCannotRead() {
+	buf := bytes.NewBuffer([]byte{1, 2, 3})
+	_, _, _, err := obfuscated2.ClientHandshake([]byte{1, 2, 3}, buf) // nolint: dogsled
+
+	suite.Error(err)
+}
+
+func (suite *ClientHandshakeTestSuite) TestOk() {
+	for nameV, snapshotV := range suite.snapshots {
+		snapshot := snapshotV
+
+		suite.T().Run(nameV, func(t *testing.T) {
+			buf := bytes.NewBuffer(snapshot.Frame.data)
+
+			dc, encryptor, decryptor, err := obfuscated2.ClientHandshake(
+				snapshot.Secret.data, buf)
+			assert.NoError(t, err)
+			assert.EqualValues(t, snapshot.DC, dc)
+
+			writeData := make([]byte, len(snapshot.Encrypted.Text.data))
+			readData := make([]byte, len(snapshot.Decrypted.Text.data))
+
+			connMock := &testlib.NetConnMock{}
+			connMock.On("Read", mock.Anything).
+				Once().
+				Return(len(snapshot.Decrypted.Text.data), nil).
+				Run(func(args mock.Arguments) {
+					arr := args.Get(0).([]byte)
+					copy(arr, snapshot.Decrypted.Cipher.data)
+				})
+			connMock.On("Write", mock.Anything).
+				Once().
+				Return(len(snapshot.Encrypted.Text.data), nil).
+				Run(func(args mock.Arguments) {
+					arr := args.Get(0).([]byte)
+					copy(writeData, arr)
+				})
+
+			conn := &obfuscated2.Conn{
+				Conn:      connMock,
+				Encryptor: encryptor,
+				Decryptor: decryptor,
+			}
+
+			n, err := conn.Read(readData)
+			assert.Equal(t, len(readData), n)
+			assert.NoError(t, err)
+			assert.Equal(t, snapshot.Decrypted.Text.data, readData)
+
+			n, err = conn.Write(snapshot.Encrypted.Text.data)
+			assert.Equal(t, len(writeData), n)
+			assert.NoError(t, err)
+			assert.Equal(t, snapshot.Encrypted.Cipher.data, writeData)
+
+			connMock.AssertExpectations(t)
+		})
+	}
+}
+
+func TestClientHandshake(t *testing.T) {
+	t.Parallel()
+	suite.Run(t, &ClientHandshakeTestSuite{})
+}