test: add tests for HandleMessage

This commit is contained in:
Derrick Hammer 2024-01-08 07:15:30 -05:00
parent a488cb806f
commit 1020293b35
Signed by: pcfreak30
GPG Key ID: C997C339BE476FF2
1 changed files with 74 additions and 0 deletions

View File

@ -5,6 +5,7 @@ import (
"encoding/base64" "encoding/base64"
"fmt" "fmt"
"git.lumeweb.com/LumeWeb/libs5-go/types" "git.lumeweb.com/LumeWeb/libs5-go/types"
"github.com/golang/mock/gomock"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/vmihailenco/msgpack/v5" "github.com/vmihailenco/msgpack/v5"
"testing" "testing"
@ -140,3 +141,76 @@ func TestHandshakeOpen_DecodeMessage(t *testing.T) {
}) })
} }
} }
func TestHandshakeOpen_HandleMessage_Success(t *testing.T) {
setup(t)
testResultEncoded := "AsQWZXhhbXBsZSBoYW5kc2hha2UgZGF0YQMA"
testResult, err := base64.StdEncoding.DecodeString(testResultEncoded)
assert.NoError(t, err)
node.EXPECT().Services().Return(services).Times(1)
node.EXPECT().NetworkId().Return("").Times(1)
services.EXPECT().P2P().Return(p2p).Times(1)
p2p.EXPECT().SignMessageSimple(testResult).Return(testResult, nil).Times(1)
peer.EXPECT().SendMessage(testResult).Return(nil).Times(1)
handshake := []byte("example handshake data")
handshakeOpen := NewHandshakeOpen([]byte{}, "")
handshakeOpen.SetHandshake(handshake)
assert.NoError(t, handshakeOpen.HandleMessage(node, peer, false))
}
func TestHandshakeOpen_HandleMessage_DifferentNetworkID(t *testing.T) {
setup(t) // Assuming setup initializes the mocks and any required objects
// Define a network ID that is different from the one in handshakeOpen
differentNetworkID := "differentNetworkID"
// Setup expectations for the mock objects
node.EXPECT().NetworkId().Return(differentNetworkID).Times(1)
// No other method calls are expected after the NetworkId check fails
// Create a HandshakeOpen instance with a specific network ID that differs from `differentNetworkID`
networkIDForHandshakeOpen := "expectedNetworkID"
handshakeOpen := NewHandshakeOpen([]byte{}, networkIDForHandshakeOpen)
handshakeOpen.SetHandshake([]byte("example handshake data"))
// Invoke HandleMessage and expect an error
err := handshakeOpen.HandleMessage(node, peer, false)
assert.Error(t, err)
// Optionally, assert that the error message is as expected
expectedErrorMessage := fmt.Sprintf("Peer is in different network: %s", networkIDForHandshakeOpen)
assert.Equal(t, expectedErrorMessage, err.Error())
}
func TestHandshakeOpen_HandleMessage_MarshalError(t *testing.T) {
setup(t)
node.EXPECT().Services().Return(services).Times(1)
node.EXPECT().NetworkId().Return("").Times(1)
services.EXPECT().P2P().Return(p2p).Times(1)
p2p.EXPECT().SignMessageSimple(gomock.Any()).Return(nil, fmt.Errorf("marshal error")).Times(1)
handshake := []byte("example handshake data")
handshakeOpen := NewHandshakeOpen([]byte{}, "")
handshakeOpen.SetHandshake(handshake)
assert.Error(t, handshakeOpen.HandleMessage(node, peer, false))
}
func TestHandshakeOpen_HandleMessage_SendMessageError(t *testing.T) {
setup(t)
node.EXPECT().Services().Return(services).Times(1)
node.EXPECT().NetworkId().Return("").Times(1)
services.EXPECT().P2P().Return(p2p).Times(1)
p2p.EXPECT().SignMessageSimple(gomock.Any()).Return([]byte{}, nil).Times(1)
peer.EXPECT().SendMessage(gomock.Any()).Return(fmt.Errorf("send message error")).Times(1)
handshake := []byte("example handshake data")
handshakeOpen := NewHandshakeOpen([]byte{}, "")
handshakeOpen.SetHandshake(handshake)
assert.Error(t, handshakeOpen.HandleMessage(node, peer, false))
}