feat: add keepalive test

Signed-off-by: Ales Verbic <[email protected]>

internal/test/ouroboros_mock/entry.go

@@ -23,10 +23,11 @@ import (
 )
 
 const (
-	MockNetworkMagic       uint32 = 999999
-	MockProtocolVersionNtC uint16 = (14 + protocol.ProtocolVersionNtCOffset)
-	MockProtocolVersionNtN uint16 = 13
-	MockKeepAliveCookie    uint16 = 999
+	MockNetworkMagic         uint32 = 999999
+	MockProtocolVersionNtC   uint16 = (14 + protocol.ProtocolVersionNtCOffset)
+	MockProtocolVersionNtN   uint16 = 13
+	MockKeepAliveCookie      uint16 = 999
+	MockKeepAliveWrongCookie uint16 = 1000
 )
 
 type EntryType int
@@ -109,6 +110,16 @@ var ConversationEntryKeepAliveResponse = ConversationEntry{
 	},
 }
 
+// ConversationEntryKeepAliveResponse is a pre-defined conversation entry for a keep-alive response
+var ConversationEntryKeepAliveResponseWrongResponse = ConversationEntry{
+	Type:       EntryTypeOutput,
+	ProtocolId: keepalive.ProtocolId,
+	IsResponse: true,
+	OutputMessages: []protocol.Message{
+		keepalive.NewMsgKeepAliveResponse(MockKeepAliveWrongCookie),
+	},
+}
+
 // ConversationKeepAlive is a pre-defined conversation with a NtN handshake and repeated keep-alive requests
 // and responses
 var ConversationKeepAlive = []ConversationEntry{
@@ -124,6 +135,15 @@ var ConversationKeepAlive = []ConversationEntry{
 	ConversationEntryKeepAliveResponse,
 }
 
+// ConversationKeepAlive is a pre-defined conversation with a NtN handshake and repeated keep-alive requests
+// and responses
+var ConversationKeepAliveWrongResponse = []ConversationEntry{
+	ConversationEntryHandshakeRequestGeneric,
+	ConversationEntryHandshakeNtNResponse,
+	ConversationEntryKeepAliveRequest,
+	ConversationEntryKeepAliveResponseWrongResponse,
+}
+
 // ConversationKeepAliveClose is a pre-defined conversation with a NtN handshake that will close the connection
 // after receiving a keep-alive request
 var ConversationKeepAliveClose = []ConversationEntry{

protocol/keepalive/serverclient_test.go

@@ -0,0 +1,68 @@
+package keepalive_test
+
+import (
+	"testing"
+
+	ouroboros "github.com/blinklabs-io/gouroboros"
+	"github.com/blinklabs-io/gouroboros/internal/test/ouroboros_mock"
+	"github.com/blinklabs-io/gouroboros/protocol/keepalive"
+)
+
+func TestServerKeepaliveHandling(t *testing.T) {
+
+	keepAliveSentCounter := 0
+
+	keepAliveConfig := keepalive.NewConfig(
+		keepalive.WithKeepAliveFunc(func(protocolID uint16) error {
+			keepAliveSentCounter++
+			return nil
+		}),
+	)
+
+	mockConn := ouroboros_mock.NewConnection(
+		ouroboros_mock.ProtocolRoleNone,
+		ouroboros_mock.ConversationKeepAlive,
+	).(*ouroboros_mock.Connection)
+
+	oConn, err := ouroboros.New(
+		ouroboros.WithConnection(mockConn),
+		ouroboros.WithNetworkMagic(ouroboros_mock.MockNetworkMagic),
+		ouroboros.WithNodeToNode(true),
+		ouroboros.WithKeepAlive(true),
+		ouroboros.WithKeepAliveConfig(keepAliveConfig),
+	)
+	if err != nil {
+		t.Fatalf("unexpected error when creating Connection object: %s", err)
+	}
+
+	// TODO Add test condition
+
+	// Close connection
+	if err := oConn.Close(); err != nil {
+		t.Fatalf("unexpected error when closing Connection object: %s", err)
+	}
+}
+
+func TestServerKeepaliveHandlingWithWrongResponse(t *testing.T) {
+	mockConn := ouroboros_mock.NewConnection(
+		ouroboros_mock.ProtocolRoleNone,
+		ouroboros_mock.ConversationKeepAliveWrongResponse,
+	).(*ouroboros_mock.Connection)
+
+	oConn, err := ouroboros.New(
+		ouroboros.WithConnection(mockConn),
+		ouroboros.WithNetworkMagic(ouroboros_mock.MockNetworkMagic),
+		ouroboros.WithNodeToNode(true),
+		ouroboros.WithKeepAlive(true),
+	)
+	if err != nil {
+		t.Fatalf("unexpected error when creating Connection object: %s", err)
+	}
+
+	// TODO Add test condition
+
+	// Close connection
+	if err := oConn.Close(); err != nil {
+		t.Fatalf("unexpected error when closing Connection object: %s", err)
+	}
+}