jay/go-honeybee
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Consolidated redundant socket mocking function.

Browse Source
This commit is contained in:
Jay
2026-04-20 17:37:30 -04:00
parent 1f38f73b75
commit fd806c0231
8 changed files with 91 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files
inbound/helpers_test.go
+1 -44
View File
@@ -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)
inbound/pool_test.go
+25 -25
View File
@@ -44,7 +44,7 @@ func TestPoolAdd(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Add("peer-1", socket)
assert.NoError(t, err)
})
@@ -53,7 +53,7 @@ func TestPoolAdd(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Add("peer-1", socket)
assert.NoError(t, err)
@@ -64,8 +64,8 @@ func TestPoolAdd(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket1, _, _ := setupTestSocket(t)
socket2, _, _ := setupTestSocket(t)
socket1, _, _ := honeybeetest.SetupTestSocket(t)
socket2, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Add("peer-1", socket1)
assert.NoError(t, err)
@@ -78,7 +78,7 @@ func TestPoolAdd(t *testing.T) {
pool := setupPool(t)
pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Add("peer-1", socket)
assert.ErrorIs(t, err, ErrPoolClosed)
})
@@ -89,8 +89,8 @@ func TestPoolReplace(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket1, _, _ := setupTestSocket(t)
socket2, _, _ := setupTestSocket(t)
socket1, _, _ := honeybeetest.SetupTestSocket(t)
socket2, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Add("peer-1", socket1)
assert.NoError(t, err)
@@ -105,7 +105,7 @@ func TestPoolReplace(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Replace("unknown", socket)
assert.ErrorIs(t, err, ErrPeerNotFound)
})
@@ -114,7 +114,7 @@ func TestPoolReplace(t *testing.T) {
pool := setupPool(t)
pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Replace("peer-1", socket)
assert.ErrorIs(t, err, ErrPoolClosed)
})
@@ -123,8 +123,8 @@ func TestPoolReplace(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket1, _, _ := setupTestSocket(t)
socket2, _, _ := setupTestSocket(t)
socket1, _, _ := honeybeetest.SetupTestSocket(t)
socket2, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Add("peer-1", socket1)
assert.NoError(t, err)
@@ -148,7 +148,7 @@ func TestPoolRemove(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Add("peer-1", socket)
assert.NoError(t, err)
@@ -178,7 +178,7 @@ func TestPoolRemove(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Add("peer-1", socket)
assert.NoError(t, err)
@@ -202,7 +202,7 @@ func TestPoolSend(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, _, outgoing := setupTestSocket(t)
socket, _, outgoing := honeybeetest.SetupTestSocket(t)
err := pool.Add("peer-1", socket)
assert.NoError(t, err)
@@ -244,7 +244,7 @@ func TestPoolClose(t *testing.T) {
pool := setupPool(t)
pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
err := pool.Add("peer-1", socket)
assert.ErrorIs(t, err, ErrPoolClosed)
})
@@ -261,8 +261,8 @@ func TestPoolPeers(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket1, _, _ := setupTestSocket(t)
socket2, _, _ := setupTestSocket(t)
socket1, _, _ := honeybeetest.SetupTestSocket(t)
socket2, _, _ := honeybeetest.SetupTestSocket(t)
pool.Add("peer-1", socket1)
pool.Add("peer-2", socket2)
@@ -276,7 +276,7 @@ func TestPoolPeers(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
pool.Add("peer-1", socket)
pool.Remove("peer-1")
@@ -287,7 +287,7 @@ func TestPoolPeers(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, incoming, _ := setupTestSocket(t)
socket, incoming, _ := honeybeetest.SetupTestSocket(t)
pool.Add("peer-1", socket)
close(incoming)
@@ -303,7 +303,7 @@ func TestPoolEvents(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, incoming, _ := setupTestSocket(t)
socket, incoming, _ := honeybeetest.SetupTestSocket(t)
pool.Add("peer-1", socket)
incoming <- honeybeetest.MockIncomingData{
@@ -321,7 +321,7 @@ func TestPoolEvents(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, incoming, _ := setupTestSocket(t)
socket, incoming, _ := honeybeetest.SetupTestSocket(t)
pool.Add("peer-1", socket)
incoming <- honeybeetest.MockIncomingData{
@@ -345,7 +345,7 @@ func TestPoolEvents(t *testing.T) {
assert.NoError(t, err)
defer pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
pool.Add("peer-1", socket)
expectEvent(t, pool.Events(), "peer-1", EventPeerEvicted)
@@ -359,7 +359,7 @@ func TestPoolEvents(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket, _, _ := setupTestSocket(t)
socket, _, _ := honeybeetest.SetupTestSocket(t)
pool.Add("peer-1", socket)
pool.Remove("peer-1")
@@ -377,8 +377,8 @@ func TestPoolEvents(t *testing.T) {
pool := setupPool(t)
defer pool.Close()
socket1, _, _ := setupTestSocket(t)
socket2, _, _ := setupTestSocket(t)
socket1, _, _ := honeybeetest.SetupTestSocket(t)
socket2, _, _ := honeybeetest.SetupTestSocket(t)
pool.Add("peer-1", socket1)
pool.Replace("peer-1", socket2)