From 38ff62fddcab66017c7e1cde08383e2da5a32de5 Mon Sep 17 00:00:00 2001
From: 9seconds <nineseconds@yandex.ru>
Date: Thu, 1 Apr 2021 15:41:51 +0300
Subject: [PATCH] Add tests for faketls conn

---
 mtglib/internal/faketls/conn_test.go | 153 +++++++++++++++++++++++++++
 1 file changed, 153 insertions(+)
 create mode 100644 mtglib/internal/faketls/conn_test.go

diff --git a/mtglib/internal/faketls/conn_test.go b/mtglib/internal/faketls/conn_test.go
new file mode 100644
index 0000000..41bff3d
--- /dev/null
+++ b/mtglib/internal/faketls/conn_test.go
@@ -0,0 +1,153 @@
+package faketls_test
+
+import (
+	"bytes"
+	"errors"
+	"io"
+	"math/rand"
+	"testing"
+
+	"github.com/9seconds/mtg/v2/mtglib/internal/faketls"
+	"github.com/9seconds/mtg/v2/mtglib/internal/faketls/record"
+	"github.com/9seconds/mtg/v2/testlib"
+	"github.com/stretchr/testify/mock"
+	"github.com/stretchr/testify/suite"
+)
+
+type ConnMock struct {
+	testlib.NetConnMock
+
+	readBuffer  bytes.Buffer
+	writeBuffer bytes.Buffer
+}
+
+func (m *ConnMock) Read(p []byte) (int, error) {
+	m.Called(p)
+
+	return m.readBuffer.Read(p)
+}
+
+func (m *ConnMock) Write(p []byte) (int, error) {
+	m.Called(p)
+
+	return m.writeBuffer.Write(p)
+}
+
+type ConnTestSuite struct {
+	suite.Suite
+
+	connMock *ConnMock
+	c        *faketls.Conn
+}
+
+func (suite *ConnTestSuite) SetupTest() {
+	suite.connMock = &ConnMock{}
+	suite.c = &faketls.Conn{
+		Conn: suite.connMock,
+	}
+}
+
+func (suite *ConnTestSuite) TearDownTest() {
+	suite.connMock.AssertExpectations(suite.T())
+}
+
+func (suite *ConnTestSuite) TestRead() {
+	suite.connMock.On("Read", mock.Anything).Return(0, nil)
+
+	rec := record.AcquireRecord()
+	defer record.ReleaseRecord(rec)
+
+	rec.Type = record.TypeChangeCipherSpec
+	rec.Version = record.Version12
+
+	rec.Payload.WriteByte(0x01)
+	rec.Dump(&suite.connMock.readBuffer) // nolint: errcheck
+	rec.Reset()
+
+	rec.Type = record.TypeApplicationData
+	rec.Version = record.Version12
+
+	rec.Payload.Write([]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10})
+	rec.Dump(&suite.connMock.readBuffer) // nolint: errcheck
+
+	resultBuffer := &bytes.Buffer{}
+	buf := make([]byte, 2)
+
+	for {
+		n, err := suite.c.Read(buf)
+		if errors.Is(err, io.EOF) {
+			break
+		}
+
+		resultBuffer.Write(buf[:n])
+	}
+
+	suite.Equal([]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}, resultBuffer.Bytes())
+}
+
+func (suite *ConnTestSuite) TestReadUnexpected() {
+	suite.connMock.On("Read", mock.Anything).Return(0, nil)
+
+	rec := record.AcquireRecord()
+	defer record.ReleaseRecord(rec)
+
+	rec.Type = record.TypeChangeCipherSpec
+	rec.Version = record.Version12
+
+	rec.Payload.WriteByte(0x01)
+	rec.Dump(&suite.connMock.readBuffer) // nolint: errcheck
+	rec.Reset()
+
+	rec.Type = record.TypeHandshake
+	rec.Version = record.Version12
+
+	rec.Payload.Write([]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10})
+	rec.Dump(&suite.connMock.readBuffer) // nolint: errcheck
+
+	buf := make([]byte, 2)
+
+	for {
+		_, err := suite.c.Read(buf)
+
+		switch {
+		case err == nil:
+		case errors.Is(err, io.EOF):
+			suite.FailNow("unexpected to finish")
+		default:
+			return
+		}
+	}
+}
+
+func (suite *ConnTestSuite) TestWrite() {
+	suite.connMock.On("Write", mock.Anything).Return(0, nil)
+
+	dataToRec := make([]byte, record.TLSMaxRecordSize*2)
+	rand.Read(dataToRec)
+
+	n, err := suite.c.Write(dataToRec)
+	suite.NoError(err)
+	suite.Equal(len(dataToRec), n)
+
+	rec := record.AcquireRecord()
+	defer record.ReleaseRecord(rec)
+
+	buf := &bytes.Buffer{}
+
+	for {
+		if err := rec.Read(&suite.connMock.writeBuffer); err != nil {
+			break
+		}
+
+		suite.Equal(record.TypeApplicationData, rec.Type)
+		suite.Equal(record.Version12, rec.Version)
+		rec.Payload.WriteTo(buf) // nolint: errcheck
+	}
+
+	suite.Equal(dataToRec, buf.Bytes())
+}
+
+func TestConn(t *testing.T) {
+	t.Parallel()
+	suite.Run(t, &ConnTestSuite{})
+}