Owen Rudge : wsdapi/tests: Add tests for Get/SetTransportAddress.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: a4ee1b2675074b795054c6174b27a3ae11c2c42f
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=a4ee1b2675074b795054c6174b27a3ae11c2c42f

Author: Owen Rudge <orudge at codeweavers.com>
Date:   Sat Jun  3 10:34:40 2017 +0100

wsdapi/tests: Add tests for Get/SetTransportAddress.

Signed-off-by: Owen Rudge <orudge at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/wsdapi/tests/Makefile.in |  2 +-
 dlls/wsdapi/tests/address.c   | 61 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 62 insertions(+), 1 deletion(-)

diff --git a/dlls/wsdapi/tests/Makefile.in b/dlls/wsdapi/tests/Makefile.in
index a4a2821..e2f6642 100644
--- a/dlls/wsdapi/tests/Makefile.in
+++ b/dlls/wsdapi/tests/Makefile.in
@@ -1,5 +1,5 @@
 TESTDLL   = wsdapi.dll
-IMPORTS   = wsdapi ole32 user32
+IMPORTS   = wsdapi ole32 user32 ws2_32
 
 C_SRCS = \
 	address.c \
diff --git a/dlls/wsdapi/tests/address.c b/dlls/wsdapi/tests/address.c
index 8b16b06..c1bf3ba 100644
--- a/dlls/wsdapi/tests/address.c
+++ b/dlls/wsdapi/tests/address.c
@@ -21,6 +21,7 @@
 
 #define COBJMACROS
 
+#include <winsock2.h>
 #include <windows.h>
 
 #include "wine/test.h"
@@ -72,11 +73,71 @@ static void CreateUdpAddress_tests(void)
     ok(ref == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ref);
 }
 
+static void GetSetTransportAddress_udp_tests(void)
+{
+    IWSDUdpAddress *udpAddress = NULL;
+    const WCHAR ipv4Address[] = {'1','0','.','2','0','.','3','0','.','4','0',0};
+    const WCHAR ipv6Address[] = {'a','a','b','b',':','c','d',':',':','a','b','c',0};
+    const WCHAR invalidAddress[] = {'n','o','t','/','v','a','l','i','d',0};
+    LPCWSTR returnedAddress = NULL;
+    WSADATA wsaData;
+    HRESULT rc;
+    int ret;
+
+    ret = WSAStartup(MAKEWORD(2, 2), &wsaData);
+    ok(ret == 0, "WSAStartup failed: %d\n", ret);
+
+    rc = WSDCreateUdpAddress(&udpAddress);
+    ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
+    ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");
+
+    rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
+    todo_wine ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
+    ok(returnedAddress == NULL, "GetTransportAddress returned unexpected address: %08x\n", rc);
+
+    /* Try setting a null address */
+    rc = IWSDUdpAddress_SetTransportAddress(udpAddress, NULL);
+    todo_wine ok(rc == E_INVALIDARG, "SetTransportAddress(NULL) returned unexpected result: %08x\n", rc);
+
+    /* Try setting an invalid address */
+    rc = IWSDUdpAddress_SetTransportAddress(udpAddress, invalidAddress);
+    todo_wine ok(rc == HRESULT_FROM_WIN32(WSAHOST_NOT_FOUND), "SetTransportAddress(invalidAddress) returned unexpected result: %08x\n", rc);
+
+    /* Try setting an IPv4 address */
+    rc = IWSDUdpAddress_SetTransportAddress(udpAddress, ipv4Address);
+    todo_wine ok(rc == S_OK, "SetTransportAddress(ipv4Address) failed: %08x\n", rc);
+
+    rc = IWSDUdpAddress_GetTransportAddress(udpAddress, NULL);
+    todo_wine ok(rc == E_POINTER, "GetTransportAddress(NULL) returned unexpected result: %08x\n", rc);
+
+    rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
+    todo_wine ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
+    todo_wine ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: '%s'\n", wine_dbgstr_w(returnedAddress));
+    todo_wine ok(lstrcmpW(returnedAddress, ipv4Address) == 0, "Returned address != ipv4Address (%s)\n", wine_dbgstr_w(returnedAddress));
+
+    /* Try setting an IPv6 address */
+    rc = IWSDUdpAddress_SetTransportAddress(udpAddress, ipv6Address);
+    todo_wine ok(rc == S_OK, "SetTransportAddress(ipv6Address) failed: %08x\n", rc);
+
+    rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
+    todo_wine ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
+    todo_wine ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: '%s'\n", wine_dbgstr_w(returnedAddress));
+    todo_wine ok(lstrcmpW(returnedAddress, ipv6Address) == 0, "Returned address != ipv6Address (%s)\n", wine_dbgstr_w(returnedAddress));
+
+    /* Release the object */
+    ret = IWSDUdpAddress_Release(udpAddress);
+    ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);
+
+    ret = WSACleanup();
+    ok(ret == 0, "WSACleanup failed: %d\n", ret);
+}
+
 START_TEST(address)
 {
     CoInitialize(NULL);
 
     CreateUdpAddress_tests();
+    GetSetTransportAddress_udp_tests();
 
     CoUninitialize();
 }