From c56bddf94e737565d9165d4261960e18c07c9e2a Mon Sep 17 00:00:00 2001
From: Daniel Lehman <dlehman25@gmail.com>
Date: Mon, 2 Nov 2020 20:12:10 -0800
Subject: [PATCH] concrt140/tests: Add _Context::_CurrentContext tests.

Signed-off-by: Daniel Lehman <dlehman25@gmail.com>
Signed-off-by: Piotr Caban <piotr@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
---
 configure                        |  1 +
 configure.ac                     |  1 +
 dlls/concrt140/tests/Makefile.in |  4 ++
 dlls/concrt140/tests/concrt140.c | 79 ++++++++++++++++++++++++++++++++
 4 files changed, 85 insertions(+)
 create mode 100644 dlls/concrt140/tests/Makefile.in
 create mode 100644 dlls/concrt140/tests/concrt140.c

diff --git a/configure b/configure
index ac6a7985cc7..4e76dbf9d02 100755
--- a/configure
+++ b/configure
@@ -20363,6 +20363,7 @@ wine_fn_config_makefile dlls/compstui enable_compstui
 wine_fn_config_makefile dlls/comsvcs enable_comsvcs
 wine_fn_config_makefile dlls/comsvcs/tests enable_tests
 wine_fn_config_makefile dlls/concrt140 enable_concrt140
+wine_fn_config_makefile dlls/concrt140/tests enable_tests
 wine_fn_config_makefile dlls/connect enable_connect
 wine_fn_config_makefile dlls/credui enable_credui
 wine_fn_config_makefile dlls/credui/tests enable_tests
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..03e0b2850f2
--- /dev/null
+++ b/dlls/concrt140/tests/concrt140.c
@@ -0,0 +1,79 @@
+/*
+ * 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@_Context@details@Concurrency@@SA?AV123@XZ");
+    if(sizeof(void*) == 8) { /* 64-bit initialization */
+        SET(p_Context_CurrentContext,
+                "?CurrentContext@Context@Concurrency@@SAPEAV12@XZ");
+    } else {
+        SET(p_Context_CurrentContext,
+                "?CurrentContext@Context@Concurrency@@SAPAV12@XZ");
+    }
+
+    return TRUE;
+}
+
+static void test_CurrentContext(void)
+{
+    _Context _ctx, *ret;
+    Context *ctx;
+
+    ctx = p_Context_CurrentContext();
+    ok(!!ctx, "got NULL\n");
+
+    memset(&_ctx, 0xcc, sizeof(_ctx));
+    ret = p__Context__CurrentContext(&_ctx);
+    ok(_ctx.ctx == ctx, "expected %p, got %p\n", ctx, _ctx.ctx);
+    ok(ret == &_ctx, "expected %p, got %p\n", &_ctx, ret);
+}
+
+START_TEST(concrt140)
+{
+    if (!init()) return;
+    test_CurrentContext();
+}