1 files changed, 54 insertions, 0 deletions
diff --git a/Tools/Hazel-Networking/Hazel.UnitTests/UPnPTests.cs b/Tools/Hazel-Networking/Hazel.UnitTests/UPnPTests.cs
new file mode 100644
index 0000000..3bee911
--- /dev/null
+++ b/Tools/Hazel-Networking/Hazel.UnitTests/UPnPTests.cs
@@ -0,0 +1,54 @@
+using System;
+using Hazel.UPnP;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+
+namespace Hazel.UnitTests
+{
+    // [TestClass]
+    // TODO: These tests are super flaky because of hardware differences. Not sure what can be done.
+    public class UPnPTests
+    {
+        [TestMethod]
+        public void CanForwardPort()
+        {
+            using (UPnPHelper dut = new UPnPHelper(Logger.Instance))
+            {
+                Assert.IsTrue(dut.ForwardPort(22023, "Hazel Test"));
+            }
+        }
+
+        [TestMethod]
+        public void CanDeletePort()
+        {
+            using (UPnPHelper dut = new UPnPHelper(Logger.Instance))
+            {
+                Assert.IsTrue(dut.DeleteForwardingRule(22023));
+            }
+        }
+    }
+
+    public class Logger : ILogger
+    {
+        public static readonly ILogger Instance = new Logger();
+
+        public void WriteVerbose(string msg)
+        {
+            Console.WriteLine(msg);
+        }
+
+        public void WriteWarning(string msg)
+        {
+            Console.WriteLine(msg);
+        }
+
+        public void WriteError(string msg)
+        {
+            Console.WriteLine(msg);
+        }
+
+        public void WriteInfo(string msg)
+        {
+            Console.WriteLine(msg);
+        }
+    }
+}