Giovanni Mascellani : mfplat/tests: Test interaction between Lock() and LockRect().

[Alexandre Julliard]

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

Author: Giovanni Mascellani <gmascellani at codeweavers.com>
Date:   Tue Jun 14 17:15:46 2022 +0200

mfplat/tests: Test interaction between Lock() and LockRect().

Signed-off-by: Giovanni Mascellani <gmascellani at codeweavers.com>
Signed-off-by: Nikolay Sivov <nsivov at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/mfplat/tests/mfplat.c | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/dlls/mfplat/tests/mfplat.c b/dlls/mfplat/tests/mfplat.c
index 28b89de4b1f..2cd27ed1269 100644
--- a/dlls/mfplat/tests/mfplat.c
+++ b/dlls/mfplat/tests/mfplat.c
@@ -6398,6 +6398,7 @@ static void test_MFCreateDXSurfaceBuffer(void)
 {
     IDirect3DSurface9 *backbuffer = NULL, *surface;
     IDirect3DSwapChain9 *swapchain;
+    D3DLOCKED_RECT locked_rect;
     DWORD length, max_length;
     IDirect3DDevice9 *device;
     IMF2DBuffer2 *_2dbuffer2;
@@ -6470,9 +6471,32 @@ static void test_MFCreateDXSurfaceBuffer(void)
     ok(hr == S_OK, "Failed to get length, hr %#lx.\n", hr);
     ok(length == 2 * max_length, "Unexpected length %lu.\n", length);
 
+    hr = IDirect3DSurface9_LockRect(backbuffer, &locked_rect, NULL, 0);
+    ok(hr == S_OK, "Failed to lock back buffer, hr %#lx.\n", hr);
+
+    /* Cannot lock while the surface is locked. */
+    hr = IMFMediaBuffer_Lock(buffer, &data, NULL, &length);
+    ok(hr == D3DERR_INVALIDCALL, "Unexpected hr %#lx.\n", hr);
+
+    hr = IDirect3DSurface9_UnlockRect(backbuffer);
+    ok(hr == S_OK, "Failed to unlock back buffer, hr %#lx.\n", hr);
+
+    hr = IMFMediaBuffer_Lock(buffer, &data, NULL, &length);
+    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
+    ok(length == max_length, "Unexpected length %lu instead of %lu.\n", length, max_length);
+
+    /* You can lock the surface while the media buffer is locked. */
+    hr = IDirect3DSurface9_LockRect(backbuffer, &locked_rect, NULL, 0);
+    ok(hr == S_OK, "Failed to lock back buffer, hr %#lx.\n", hr);
+
     hr = IMFMediaBuffer_Lock(buffer, &data, NULL, &length);
     ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
-    ok(length == max_length, "Unexpected length.\n");
+
+    hr = IMFMediaBuffer_Unlock(buffer);
+    ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
+
+    hr = IDirect3DSurface9_UnlockRect(backbuffer);
+    ok(hr == S_OK, "Failed to unlock back buffer, hr %#lx.\n", hr);
 
     /* Unlock twice. */
     hr = IMFMediaBuffer_Unlock(buffer);