[PATCH 4/4] concrt140/tests: Add _Context::_CurrentContext tests.

[Daniel Lehman]

Signed-off-by: Daniel Lehman <dlehman25 at gmail.com>
---
 configure.ac                     |  1 +
 dlls/concrt140/tests/Makefile.in |  4 ++
 dlls/concrt140/tests/concrt140.c | 82 ++++++++++++++++++++++++++++++++
 3 files changed, 87 insertions(+)
 create mode 100644 dlls/concrt140/tests/Makefile.in
 create mode 100644 dlls/concrt140/tests/concrt140.c

diff --git a/configure.ac b/configure.ac
index 2f81fd22610..06031352866 100644
--- a/configure.ac
+++ b/configure.ac
@@ -3106,6 +3106,7 @@ WINE_CONFIG_MAKEFILE(dlls/compstui)
 WINE_CONFIG_MAKEFILE(dlls/comsvcs)
 WINE_CONFIG_MAKEFILE(dlls/comsvcs/tests)
 WINE_CONFIG_MAKEFILE(dlls/concrt140)
+WINE_CONFIG_MAKEFILE(dlls/concrt140/tests)
 WINE_CONFIG_MAKEFILE(dlls/connect)
 WINE_CONFIG_MAKEFILE(dlls/credui)
 WINE_CONFIG_MAKEFILE(dlls/credui/tests)
diff --git a/dlls/concrt140/tests/Makefile.in b/dlls/concrt140/tests/Makefile.in
new file mode 100644
index 00000000000..e36fc37eb86
--- /dev/null
+++ b/dlls/concrt140/tests/Makefile.in
@@ -0,0 +1,4 @@
+TESTDLL   = concrt140.dll
+
+C_SRCS = \
+	concrt140.c
diff --git a/dlls/concrt140/tests/concrt140.c b/dlls/concrt140/tests/concrt140.c
new file mode 100644
index 00000000000..e450024adee
--- /dev/null
+++ b/dlls/concrt140/tests/concrt140.c
@@ -0,0 +1,82 @@
+/*
+ * Copyright 2020 Daniel Lehman
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#include "wine/test.h"
+
+typedef void (*vtable_ptr)(void);
+
+typedef struct {
+    const vtable_ptr *vtable;
+} Context;
+
+typedef struct {
+    Context *ctx;
+} _Context;
+
+static Context* (__cdecl *p_Context_CurrentContext)(void);
+static _Context* (__cdecl *p__Context__CurrentContext)(_Context*);
+
+#define SETNOFAIL(x,y) x = (void*)GetProcAddress(module,y)
+#define SET(x,y) do { SETNOFAIL(x,y); ok(x != NULL, "Export '%s' not found\n", y); } while(0)
+
+static BOOL init(void)
+{
+    HMODULE module;
+
+    module = LoadLibraryA("concrt140.dll");
+    if (!module)
+    {
+        win_skip("concrt140.dll not installed\n");
+        return FALSE;
+    }
+
+    SET(p__Context__CurrentContext,
+        "?_CurrentContext at _Context@details at Concurrency@@SA?AV123 at XZ");
+    if(sizeof(void*) == 8) { /* 64-bit initialization */
+        SET(p_Context_CurrentContext,
+                "?CurrentContext at Context@Concurrency@@SAPEAV12 at XZ");
+    } else {
+        SET(p_Context_CurrentContext,
+                "?CurrentContext at Context@Concurrency@@SAPAV12 at XZ");
+    }
+
+    return TRUE;
+}
+
+static void test_CurrentContext(void)
+{
+    _Context _ctx, *_pctx;
+    Context *ctx;
+
+    ctx = p_Context_CurrentContext();
+    ok(!!ctx, "got NULL\n");
+
+    if (0) /* crash Windows */
+        p__Context__CurrentContext(NULL);
+
+    memset(&_ctx, 0xcc, sizeof(_ctx));
+    _pctx = p__Context__CurrentContext(&_ctx);
+    ok(_ctx.ctx == ctx, "expected %p, got %p\n", ctx, _ctx.ctx);
+    ok(_pctx == &_ctx, "expected %p, got %p\n", &_ctx, _pctx);
+}
+
+START_TEST(concrt140)
+{
+    if (!init()) return;
+    test_CurrentContext();
+}
-- 
2.25.1