From: Thomas Faber <thomas.faber@reactos.org>
Date: Sun, 22 Nov 2015 13:51:19 +0000 (+0000)
Subject: [CRT_APITEST]
X-Git-Tag: ReactOS-0.4.0~60^2~85
X-Git-Url: https://git.reactos.org/?p=reactos.git;a=commitdiff_plain;h=3aee20e0d8d8b2872445bf95dcd23f4ebb468197

[CRT_APITEST]
- Fix dll loading in static_construct test
CORE-10562

svn path=/trunk/; revision=70022
---

diff --git a/rostests/apitests/crt/static_construct.cpp b/rostests/apitests/crt/static_construct.cpp
index 6db9c04fc5e..cdc27eaf2e5 100644
--- a/rostests/apitests/crt/static_construct.cpp
+++ b/rostests/apitests/crt/static_construct.cpp
@@ -6,6 +6,7 @@
  */
 
 #include <apitest.h>
+#include <strsafe.h>
 #include "dll_startup.h"
 
 extern "C"
@@ -35,7 +36,9 @@ static struct init_static
     }
 } init_static;
 
-START_TEST(static_construct)
+static
+VOID
+TestInitStatic(VOID)
 {
     ok(static_init_counter_at_startup == 123, "static_init_counter at startup: %d\n", static_init_counter_at_startup);
     ok(static_construct_counter_at_startup == 789, "static_construct_counter at startup: %d\n", static_construct_counter_at_startup);
@@ -46,15 +49,25 @@ START_TEST(static_construct)
     ok(static_construct_counter == 790, "static_construct_counter: %d\n", static_construct_counter);
     ok(init_static.m_counter == 2, "init_static.m_counter: %d\n", init_static.m_counter);
     ok(init_static.m_uninit == 1, "init_static.m_uninit: %d\n", init_static.m_uninit);
+}
 
-    counter_values values;
+static
+VOID
+TestDllStartup(VOID)
+{
 #if defined(TEST_MSVCRT)
-    HMODULE hDll = LoadLibraryW(L"msvcrt_crt_dll_startup.dll");
+    const PCWSTR DllName = L"msvcrt_crt_dll_startup.dll";
 #elif defined(TEST_STATIC_CRT)
-    HMODULE hDll = LoadLibraryW(L"static_crt_dll_startup.dll");
+    const PCWSTR DllName = L"static_crt_dll_startup.dll";
 #else
 #error This test only makes sense for static CRT and msvcrt.dll
 #endif
+    WCHAR DllPath[MAX_PATH];
+    GetModuleFileNameW(NULL, DllPath, _countof(DllPath));
+    wcsrchr(DllPath, L'\\')[1] = UNICODE_NULL;
+    StringCchCatW(DllPath, _countof(DllPath), DllName);
+
+    HMODULE hDll = LoadLibraryW(DllPath);
     if (hDll == NULL)
     {
         skip("Helper dll not found\n");
@@ -67,6 +80,7 @@ START_TEST(static_construct)
         FreeLibrary(hDll);
         return;
     }
+    counter_values values;
     pSetCounterValuesPointer(&values);
     ok(values.m_uninit_at_startup == 0, "m_uninit_at_startup = %d\n", values.m_uninit_at_startup);
     ok(values.m_uninit == 1, "m_uninit = %d\n", values.m_uninit);
@@ -86,3 +100,9 @@ START_TEST(static_construct)
     ok(values.dtor_counter_at_detach == 7878, "dtor_counter_at_detach = %d\n", values.dtor_counter_at_detach);
     ok(values.dtor_counter == 7879, "dtor_counter = %d\n", values.dtor_counter);
 }
+
+START_TEST(static_construct)
+{
+    TestInitStatic();
+    TestDllStartup();
+}