From b6105fc5d025d3cafb9b3c41e8aea917a5d72718 Mon Sep 17 00:00:00 2001
From: Thom Carlin <tcarlin@redhat.com>
Date: Wed, 16 Oct 2024 12:12:25 -0400
Subject: [PATCH] Add udp test (#1171)

---
 pkg/backends/udp_test.go | 55 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 pkg/backends/udp_test.go

diff --git a/pkg/backends/udp_test.go b/pkg/backends/udp_test.go
new file mode 100644
index 000000000..3db0d9750
--- /dev/null
+++ b/pkg/backends/udp_test.go
@@ -0,0 +1,55 @@
+package backends
+
+import (
+	"net"
+	"sync"
+	"testing"
+
+	"github.com/ansible/receptor/pkg/logger"
+)
+
+func TestNewUDPListener(t *testing.T) {
+	type args struct {
+		address string
+		logger  *logger.ReceptorLogger
+	}
+
+	goodLogger := logger.NewReceptorLogger("UDPtest")
+	goodUDPListener := &UDPListener{
+		laddr:           &net.UDPAddr{},
+		conn:            &net.UDPConn{},
+		sessChan:        make(chan *UDPListenerSession),
+		sessRegLock:     sync.RWMutex{},
+		sessionRegistry: make(map[string]*UDPListenerSession),
+		logger:          goodLogger,
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    *UDPListener
+		wantErr bool
+	}{
+		{
+			name: "Positive",
+			args: args{
+				address: "127.0.0.1:9998",
+				logger:  goodLogger,
+			},
+			want:    goodUDPListener,
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := NewUDPListener(tt.args.address, tt.args.logger)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("NewUDPListener() error = %+v, wantErr %+v", err, tt.wantErr)
+
+				return
+			}
+			if got == nil {
+				t.Errorf("NewUDPListener(): want UDP Listener, got nil")
+			}
+		})
+	}
+}