[PATCH 6/7] kernel32/tests: Call UpdateProcThreadAttribute via function address

[Hugh McMaster]

From: Hugh McMaster

---
 dlls/kernel32/tests/console.c | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/dlls/kernel32/tests/console.c b/dlls/kernel32/tests/console.c
index 2a071900f3c..c7f2218e051 100644
--- a/dlls/kernel32/tests/console.c
+++ b/dlls/kernel32/tests/console.c
@@ -40,6 +40,8 @@ static HANDLE (WINAPI *pOpenConsoleW)(LPCWSTR,DWORD,BOOL,DWORD);
 static BOOL (WINAPI *pSetConsoleInputExeNameA)(LPCSTR);
 static BOOL (WINAPI *pSetConsoleScreenBufferInfoEx)(HANDLE, CONSOLE_SCREEN_BUFFER_INFOEX *);
 static BOOL (WINAPI *pSetCurrentConsoleFontEx)(HANDLE, BOOL, CONSOLE_FONT_INFOEX *);
+static BOOL (WINAPI *pUpdateProcThreadAttribute)(struct _PROC_THREAD_ATTRIBUTE_LIST *, DWORD,
+                                                 DWORD_PTR, void *, SIZE_T, void *, SIZE_T *);
 static BOOL (WINAPI *pVerifyConsoleIoHandle)(HANDLE handle);

static BOOL skip_nt;
@@ -92,6 +94,7 @@ static void init_function_pointers(void)
     KERNEL32_GET_PROC(SetConsoleInputExeNameA);
     KERNEL32_GET_PROC(SetConsoleScreenBufferInfoEx);
     KERNEL32_GET_PROC(SetCurrentConsoleFontEx);
+    KERNEL32_GET_PROC(UpdateProcThreadAttribute);
     KERNEL32_GET_PROC(VerifyConsoleIoHandle);

#undef KERNEL32_GET_PROC
@@ -4743,9 +4746,9 @@ static void test_pseudo_console(void)
     BOOL ret;
     HRESULT hres;

-    if (!pCreatePseudoConsole)
+    if (!pCreatePseudoConsole || !pUpdateProcThreadAttribute)
     {
-        win_skip("CreatePseudoConsole not available\n");
+        win_skip("CreatePseudoConsole or UpdateProcThreadAttribute not available.\n");
         return;
     }

@@ -4779,7 +4782,7 @@ static void test_pseudo_console(void)
     InitializeProcThreadAttributeList(NULL, 1, 0, &attr_size);
     startup.lpAttributeList = HeapAlloc(GetProcessHeap(), 0, attr_size);
     InitializeProcThreadAttributeList(startup.lpAttributeList, 1, 0, &attr_size);
-    UpdateProcThreadAttribute(startup.lpAttributeList, 0, PROC_THREAD_ATTRIBUTE_PSEUDOCONSOLE, pseudo_console,
+    pUpdateProcThreadAttribute(startup.lpAttributeList, 0, PROC_THREAD_ATTRIBUTE_PSEUDOCONSOLE, pseudo_console,
                               sizeof(pseudo_console), NULL, NULL);

winetest_get_mainargs(&argv);