From d175da71898789dc3122e6c34feb781f31cd1529 Mon Sep 17 00:00:00 2001 From: Pierre Ossman Date: Mon, 11 Jul 2016 10:05:05 +0200 Subject: [PATCH] Add unit tests for getHostAndPort() --- tests/CMakeLists.txt | 3 ++ tests/hostport.cxx | 80 ++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 83 insertions(+) create mode 100644 tests/hostport.cxx diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index e99c825a..bfd69dcd 100644 --- 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 index 00000000..00026e61 --- /dev/null +++ b/tests/hostport.cxx @@ -0,0 +1,80 @@ +/* Copyright 2016 Pierre Ossman 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 + +#include + +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; +} -- 2.39.5