Dmitry Timoshkov : ole32/tests: Add more tests for CoIncrementMTAUsage/CoDecrementMTAUsage.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 251ab17af7d6162fed951afea417dd11b34d8a26
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=251ab17af7d6162fed951afea417dd11b34d8a26

Author: Dmitry Timoshkov <dmitry at baikal.ru>
Date:   Wed Jul 28 16:43:49 2021 +0300

ole32/tests: Add more tests for CoIncrementMTAUsage/CoDecrementMTAUsage.

Signed-off-by: Dmitry Timoshkov <dmitry at baikal.ru>
Signed-off-by: Huw Davies <huw at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/ole32/tests/compobj.c | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/dlls/ole32/tests/compobj.c b/dlls/ole32/tests/compobj.c
index d1e947e393f..1d46a5b9c13 100644
--- a/dlls/ole32/tests/compobj.c
+++ b/dlls/ole32/tests/compobj.c
@@ -4098,6 +4098,25 @@ static void test_mta_usage(void)
     ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
 
     test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE);
+
+    hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
+    ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
+
+    test_apt_type(APTTYPE_MAINSTA, APTTYPEQUALIFIER_NONE);
+
+    cookie = 0;
+    hr = pCoIncrementMTAUsage(&cookie);
+    ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
+    ok(cookie != NULL, "Unexpected cookie %p.\n", cookie);
+
+    test_apt_type(APTTYPE_MAINSTA, APTTYPEQUALIFIER_NONE);
+
+    hr = pCoDecrementMTAUsage(cookie);
+    ok(hr == S_OK, "Unexpected hr %#x.\n", hr);
+
+    CoUninitialize();
+
+    test_apt_type(APTTYPE_CURRENT, APTTYPEQUALIFIER_NONE);
 }
 
 static void test_CoCreateInstanceFromApp(void)