From 4e3e1b9104efcd788013255a25fafca7d36005e3 Mon Sep 17 00:00:00 2001
From: Sudo-Ivan <ivan@quad4.io>
Date: Fri, 2 Jan 2026 17:57:29 -0600
Subject: [PATCH] feat(transport): add SetTransportInstance function to allow
 setting the transport instance

---
 pkg/transport/transport.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pkg/transport/transport.go b/pkg/transport/transport.go
index de5eda7..e983377 100644
--- a/pkg/transport/transport.go
+++ b/pkg/transport/transport.go
@@ -323,6 +323,12 @@ func GetTransportInstance() *Transport {
 	return transportInstance
 }
 
+func SetTransportInstance(t *Transport) {
+	transportMutex.Lock()
+	defer transportMutex.Unlock()
+	transportInstance = t
+}
+
 func (t *Transport) RegisterInterface(name string, iface common.NetworkInterface) error {
 	t.mutex.Lock()
 	defer t.mutex.Unlock()