Add primitive relay tests

mtglib/internal/relay/relay_test.go

@@ -0,0 +1,56 @@
+package relay_test
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/9seconds/mtg/v2/mtglib/internal/relay"
+	"github.com/stretchr/testify/suite"
+)
+
+type RelayTestSuite struct {
+	suite.Suite
+
+	ctx       context.Context
+	ctxCancel context.CancelFunc
+	r         *relay.Relay
+}
+
+func (suite *RelayTestSuite) SetupTest() {
+	suite.ctx, suite.ctxCancel = context.WithCancel(context.Background())
+	suite.r = relay.AcquireRelay(suite.ctx, loggerMock{}, 4096, time.Second)
+}
+
+func (suite *RelayTestSuite) TearDownTest() {
+	suite.ctxCancel()
+	relay.ReleaseRelay(suite.r)
+	suite.r = nil
+}
+
+func (suite *RelayTestSuite) TestCancelled() {
+	suite.ctxCancel()
+
+	eastConn := &rwcMock{}
+	eastConn.Write([]byte{1, 2, 3, 4, 5})
+
+	westConn := &rwcMock{}
+	westConn.Write([]byte{100, 101, 102})
+
+	suite.Nil(suite.r.Process(eastConn, westConn))
+}
+
+func (suite *RelayTestSuite) TestCopyFine() {
+	eastConn := &rwcMock{}
+	eastConn.Write([]byte{1, 2, 3, 4, 5})
+
+	westConn := &rwcMock{}
+	westConn.Write([]byte{100, 101, 102})
+
+	suite.NotNil(suite.r.Process(eastConn, westConn))
+}
+
+func TestRelay(t *testing.T) {
+	t.Parallel()
+	suite.Run(t, &RelayTestSuite{})
+}