wsdapi/tests: Add tests for Get/SetTransportAddress.

Signed-off-by: Owen Rudge <orudge@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Owen Rudge 2017-06-03 10:34:40 +01:00 committed by Alexandre Julliard
parent 30b3563373
commit a4ee1b2675
2 changed files with 62 additions and 1 deletions

View File

@ -1,5 +1,5 @@
TESTDLL = wsdapi.dll TESTDLL = wsdapi.dll
IMPORTS = wsdapi ole32 user32 IMPORTS = wsdapi ole32 user32 ws2_32
C_SRCS = \ C_SRCS = \
address.c \ address.c \

View File

@ -21,6 +21,7 @@
#define COBJMACROS #define COBJMACROS
#include <winsock2.h>
#include <windows.h> #include <windows.h>
#include "wine/test.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); 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) START_TEST(address)
{ {
CoInitialize(NULL); CoInitialize(NULL);
CreateUdpAddress_tests(); CreateUdpAddress_tests();
GetSetTransportAddress_udp_tests();
CoUninitialize(); CoUninitialize();
} }