From 1514f509b6f2f1daa620ede667d0b86363a82c66 Mon Sep 17 00:00:00 2001
From: Jeroen Wijenbergh <jeroenwijenbergh@protonmail.com>
Date: Thu, 2 May 2024 11:18:43 +0200
Subject: Types Protocol: Add Test

---
 types/protocol/protocol_test.go | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 types/protocol/protocol_test.go

(limited to 'types')

diff --git a/types/protocol/protocol_test.go b/types/protocol/protocol_test.go
new file mode 100644
index 0000000..481d2c6
--- /dev/null
+++ b/types/protocol/protocol_test.go
@@ -0,0 +1,34 @@
+package protocol
+
+import "testing"
+
+func TestNew(t *testing.T) {
+	cases := []struct {
+		in   string
+		want Protocol
+	}{
+		{
+			in:   "openvpn",
+			want: OpenVPN,
+		},
+		{
+			in:   "wireguard",
+			want: WireGuard,
+		},
+		{
+			in:   "wrong",
+			want: Unknown,
+		},
+		{
+			in:   "",
+			want: Unknown,
+		},
+	}
+
+	for _, c := range cases {
+		got := New(c.in)
+		if got != c.want {
+			t.Fatalf("got: %v, not equal to want: %v", got, c.want)
+		}
+	}
+}
-- 
cgit v1.2.3