Merge pull request #14006 from baude/pkgsignalspecgenutilunittests

Unit tests for pkg/specgenutil pkg/signal

1 files changed, 57 insertions, 0 deletions

diff --git a/pkg/specgenutil/ports_test.go b/pkg/specgenutil/ports_test.go
new file mode 100644
index 000000000..3f62c619c
--- /dev/null
+++ b/pkg/specgenutil/ports_test.go
@@ -0,0 +1,57 @@
+package specgenutil
+
+import "testing"
+
+func Test_verifyExpose(t *testing.T) {
+	type args struct {
+		expose []string
+	}
+	tests := []struct {
+		name    string
+		args    args
+		wantErr bool
+	}{
+		{
+			name: "single port with tcp",
+			args: args{
+				expose: []string{"53/tcp"},
+			},
+			wantErr: false,
+		},
+		{
+			name: "single port with udp",
+			args: args{
+				expose: []string{"53/udp"},
+			},
+			wantErr: false,
+		},
+		{
+			name: "good port range",
+			args: args{
+				expose: []string{"100-133"},
+			},
+			wantErr: false,
+		},
+		{
+			name: "high to low should fail",
+			args: args{
+				expose: []string{"100-99"},
+			},
+			wantErr: true,
+		},
+		{
+			name: "range with protocol",
+			args: args{
+				expose: []string{"53/tcp-55/tcp"},
+			},
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if err := verifyExpose(tt.args.expose); (err != nil) != tt.wantErr {
+				t.Errorf("verifyExpose() error = %v, wantErr %v", err, tt.wantErr)
+			}
+		})
+	}
+}