From 6a474c92dc3e23556f06180f4f10ce11d5f97c36 Mon Sep 17 00:00:00 2001
From: Derrick Hammer <derrick@derrickhammer.com>
Date: Mon, 8 Jan 2024 01:16:20 -0500
Subject: [PATCH] test: fix mock generation

---
 protocol/base/base.go               | 2 +-
 protocol/base/encodeable_message.go | 2 ++
 protocol/base/incoming_message.go   | 2 ++
 protocol/base/signed.go             | 2 ++
 4 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/protocol/base/base.go b/protocol/base/base.go
index d1ac677..e041d22 100644
--- a/protocol/base/base.go
+++ b/protocol/base/base.go
@@ -6,7 +6,7 @@ import (
 	"github.com/vmihailenco/msgpack/v5"
 )
 
-//go:generate mockgen -source=incoming_message.go -destination=../mocks/base/incoming_message.go -package=base
+//go:generate mockgen -source=base.go -destination=../mocks/base/base.go -package=base
 
 type IncomingMessage interface {
 	HandleMessage(node interfaces.Node, peer net.Peer, verifyId bool) error
diff --git a/protocol/base/encodeable_message.go b/protocol/base/encodeable_message.go
index 5fdf84e..c2e79b1 100644
--- a/protocol/base/encodeable_message.go
+++ b/protocol/base/encodeable_message.go
@@ -6,6 +6,8 @@ var (
 	_ EncodeableMessage = (*EncodeableMessageImpl)(nil)
 )
 
+//go:generate mockgen -source=encodeable_message.go -destination=../mocks/base/encodeable_message.go -package=base
+
 type EncodeableMessage interface {
 	ToMessage() (message []byte, err error)
 	msgpack.CustomEncoder
diff --git a/protocol/base/incoming_message.go b/protocol/base/incoming_message.go
index d481d14..100b484 100644
--- a/protocol/base/incoming_message.go
+++ b/protocol/base/incoming_message.go
@@ -10,6 +10,8 @@ import (
 	"net/url"
 )
 
+//go:generate mockgen -source=incoming_message.go -destination=../../mocks/base/incoming_message.go -package=base
+
 var _ msgpack.CustomDecoder = (*IncomingMessageImpl)(nil)
 var _ IncomingMessage = (*IncomingMessageImpl)(nil)
 var _ IncomingMessageTyped = (*IncomingMessageImpl)(nil)
diff --git a/protocol/base/signed.go b/protocol/base/signed.go
index 9013e86..3432ceb 100644
--- a/protocol/base/signed.go
+++ b/protocol/base/signed.go
@@ -1,5 +1,7 @@
 package base
 
+//go:generate mockgen -source=signed.go -destination=../../mocks/base/signed.go -package=base -aux_files=git.lumeweb.com/LumeWeb/libs5-go/protocol/base=base.go
+
 type SignedIncomingMessage interface {
 	IncomingMessage
 }