From bd2fa9d7d8169668dd5ec767ab8469e1bca8f78a Mon Sep 17 00:00:00 2001
From: Joel  Speed <joel.speed@hotmail.co.uk>
Date: Sat, 5 Jun 2021 13:24:04 +0100
Subject: [PATCH] Add tests for split host port

---
 pkg/app/redirect/validator_test.go | 41 ++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/pkg/app/redirect/validator_test.go b/pkg/app/redirect/validator_test.go
index ac08f6dc..f7692bff 100644
--- a/pkg/app/redirect/validator_test.go
+++ b/pkg/app/redirect/validator_test.go
@@ -99,4 +99,45 @@ var _ = Describe("Validator suite", func() {
 			Entry("Partial Subdomain", "evilbar.foo", false),
 		)
 	})
+
+	Context("SplitHostPort", func() {
+		type splitHostPortTableInput struct {
+			hostport     string
+			expectedHost string
+			expectedPort string
+		}
+
+		DescribeTable("Should split the host and port",
+			func(in splitHostPortTableInput) {
+				host, port := splitHostPort(in.hostport)
+				Expect(host).To(Equal(in.expectedHost))
+				Expect(port).To(Equal(in.expectedPort))
+			},
+			Entry("when no port is specified", splitHostPortTableInput{
+				hostport:     "foo.bar",
+				expectedHost: "foo.bar",
+				expectedPort: "",
+			}),
+			Entry("with a valid port specified", splitHostPortTableInput{
+				hostport:     "foo.bar:8080",
+				expectedHost: "foo.bar",
+				expectedPort: "8080",
+			}),
+			Entry("with an invalid port specified", splitHostPortTableInput{
+				hostport:     "foo.bar:808a",
+				expectedHost: "foo.bar:808a",
+				expectedPort: "",
+			}),
+			Entry("with a wildcard port specified", splitHostPortTableInput{
+				hostport:     "foo.bar:*",
+				expectedHost: "foo.bar",
+				expectedPort: "*",
+			}),
+			Entry("when the host is specified with brackets", splitHostPortTableInput{
+				hostport:     "[foo.bar]",
+				expectedHost: "foo.bar",
+				expectedPort: "",
+			}),
+		)
+	})
 })