Add unit tests for getHostAndPort()

diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index e99c825a94aa3391a30669659797477c757e28a0..bfd69dcd5d1989421e22e02401658aa99db4602d 100644 (file)
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -13,3 +13,6 @@ target_link_libraries(decperf test_util rfb)
 
 add_executable(encperf encperf.cxx)
 target_link_libraries(encperf test_util rfb)
+
+add_executable(hostport hostport.cxx)
+target_link_libraries(hostport rfb)

diff --git a/tests/hostport.cxx b/tests/hostport.cxx
new file mode 100644 (file)
index 0000000..00026e6
--- /dev/null
+++ b/tests/hostport.cxx
@@ -0,0 +1,80 @@
+/* Copyright 2016 Pierre Ossman <ossman@cendio.se> for Cendio AB
+ *
+ * This is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this software; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,
+ * USA.
+ */
+
+#include <stdio.h>
+
+#include <rfb/Hostname.h>
+
+static void doTest(const char* hostAndPort,
+                   const char* expectedHost, int expectedPort)
+{
+    char* host;
+    int port;
+
+    printf("\"%s\": ", hostAndPort);
+
+    rfb::getHostAndPort(hostAndPort, &host, &port);
+
+    if (strcmp(host, expectedHost) != 0)
+        printf("FAILED (\"%s\" != \"%s\")", host, expectedHost);
+    else if (port != expectedPort)
+        printf("FAILED (%d != %d)", port, expectedPort);
+    else
+        printf("OK");
+    printf("\n");
+    fflush(stdout);
+
+    rfb::strFree(host);
+}
+
+int main(int argc, char** argv)
+{
+    doTest(":5", "localhost", 5905);
+
+    doTest("1.2.3.4", "1.2.3.4", 5900);
+
+    doTest("1.2.3.4:5", "1.2.3.4", 5905);
+    doTest("1.2.3.4:99", "1.2.3.4", 5999);
+    doTest("1.2.3.4:100", "1.2.3.4", 100);
+    doTest("1.2.3.4:5901", "1.2.3.4", 5901);
+
+    doTest("1.2.3.4::5", "1.2.3.4", 5);
+    doTest("1.2.3.4::99", "1.2.3.4", 99);
+    doTest("1.2.3.4::5901", "1.2.3.4", 5901);
+
+    doTest("[1.2.3.4]", "1.2.3.4", 5900);
+    doTest("[1.2.3.4]:5", "1.2.3.4", 5905);
+    doTest("[1.2.3.4]:100", "1.2.3.4", 100);
+    doTest("[1.2.3.4]::5", "1.2.3.4", 5);
+    doTest("[1.2.3.4]::100", "1.2.3.4", 100);
+
+    // Ambigiuous. For now we'll keep the old behaviour...
+    doTest("::1", "localhost", 1);
+
+    doTest("2001:1234::20:1", "2001:1234::20:1", 5900);
+
+    doTest("[::1]", "::1", 5900);
+    doTest("[2001:1234::20:1]", "2001:1234::20:1", 5900);
+
+    doTest("[2001:1234::20:1]:5", "2001:1234::20:1", 5905);
+    doTest("[2001:1234::20:1]:99", "2001:1234::20:1", 5999);
+    doTest("[2001:1234::20:1]:100", "2001:1234::20:1", 100);
+    doTest("[2001:1234::20:1]:5901", "2001:1234::20:1", 5901);
+
+    return 0;
+}