Consolidated redundant socket mocking function.

inbound/helpers_test.go

@@ -4,52 +4,9 @@ import (
 	"git.wisehodl.dev/jay/go-honeybee/honeybeetest"
 	"git.wisehodl.dev/jay/go-honeybee/transport"
 	"github.com/stretchr/testify/assert"
-	"io"
 	"testing"
 )
 
-func setupTestSocket(t *testing.T) (
-	socket *honeybeetest.MockSocket,
-	incoming chan honeybeetest.MockIncomingData,
-	outgoing chan honeybeetest.MockOutgoingData,
-) {
-	t.Helper()
-
-	incoming = make(chan honeybeetest.MockIncomingData, 10)
-	outgoing = make(chan honeybeetest.MockOutgoingData, 10)
-	socket = honeybeetest.NewMockSocket()
-
-	socket.CloseFunc = func() error {
-		socket.Once.Do(func() { close(socket.Closed) })
-		return nil
-	}
-
-	socket.ReadMessageFunc = func() (int, []byte, error) {
-		select {
-		case data, ok := <-incoming:
-			if !ok {
-				return 0, nil, io.EOF
-			}
-			return data.MsgType, data.Data, data.Err
-		case <-socket.Closed:
-			return 0, nil, io.EOF
-		}
-	}
-
-	socket.WriteMessageFunc = func(msgType int, data []byte) error {
-		select {
-		case outgoing <- honeybeetest.MockOutgoingData{MsgType: msgType, Data: data}:
-			return nil
-		case <-socket.Closed:
-			return io.EOF
-		default:
-			return io.EOF
-		}
-	}
-
-	return
-}
-
 func setupTestConnection(t *testing.T) (
 	conn *transport.Connection,
 	socket *honeybeetest.MockSocket,
@@ -58,7 +15,7 @@ func setupTestConnection(t *testing.T) (
 ) {
 	t.Helper()
 
-	socket, incoming, outgoing = setupTestSocket(t)
+	socket, incoming, outgoing = honeybeetest.SetupTestSocket(t)
 
 	var err error
 	conn, err = transport.NewConnectionFromSocket(socket, nil, nil)