From 798b23d9711e767da00157cc89c87e3011084ad5 Mon Sep 17 00:00:00 2001
From: Zebediah Figura <zfigura@codeweavers.com>
Date: Tue, 20 Dec 2022 17:43:48 -0600
Subject: [PATCH] d3d8/tests: Test uploading the whole mipmap chain in one map.

Signed-off-by: Zebediah Figura <zfigura@codeweavers.com>
---
 dlls/d3d8/tests/visual.c | 74 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)

diff --git a/dlls/d3d8/tests/visual.c b/dlls/d3d8/tests/visual.c
index ea2d96378f7..c8822b4c0ca 100644
--- a/dlls/d3d8/tests/visual.c
+++ b/dlls/d3d8/tests/visual.c
@@ -250,6 +250,7 @@ static void check_rect(struct surface_readback *rb, RECT r, const char *message)
 
 #define check_rt_color(a, b) check_rt_color_(__LINE__, a, b, false)
 #define check_rt_color_todo(a, b) check_rt_color_(__LINE__, a, b, true)
+#define check_rt_color_todo_if(a, b, c) check_rt_color_(__LINE__, a, b, c)
 static void check_rt_color_(unsigned int line, IDirect3DSurface8 *rt, D3DCOLOR expected_color, bool todo)
 {
     unsigned int color = 0xdeadbeef;
@@ -12126,6 +12127,78 @@ static void test_managed_reset(void)
     release_test_context(&context);
 }
 
+/* Some applications lock a mipmapped texture at level 0, write every level at
+ * once, and expect it to be uploaded. */
+static void test_mipmap_upload(void)
+{
+    unsigned int i, j, width, level_count;
+    struct d3d8_test_context context;
+    IDirect3DTexture8 *texture;
+    D3DLOCKED_RECT locked_rect;
+    IDirect3DDevice8 *device;
+    unsigned int *mem;
+    HRESULT hr;
+
+    static const D3DPOOL pools[] =
+    {
+        D3DPOOL_MANAGED,
+        D3DPOOL_SYSTEMMEM,
+    };
+
+    if (!init_test_context(&context))
+        return;
+    device = context.device;
+
+    for (i = 0; i < ARRAY_SIZE(pools); ++i)
+    {
+        winetest_push_context("pool %#x", pools[i]);
+
+        hr = IDirect3DDevice8_CreateTexture(device, 32, 32, 0, 0,
+                D3DFMT_A8R8G8B8, pools[i], &texture);
+        ok(hr == S_OK, "Got hr %#lx.\n", hr);
+
+        level_count = IDirect3DBaseTexture8_GetLevelCount(texture);
+
+        hr = IDirect3DTexture8_LockRect(texture, 0, &locked_rect, NULL, 0);
+        ok(hr == S_OK, "Got hr %#lx.\n", hr);
+
+        mem = locked_rect.pBits;
+
+        for (j = 0; j < level_count; ++j)
+        {
+            width = 32 >> j;
+            memset(mem, 0x11 * (j + 1), width * width * 4);
+            mem += width * width;
+        }
+
+        hr = IDirect3DTexture8_UnlockRect(texture, 0);
+        ok(hr == S_OK, "Got hr %#lx.\n", hr);
+
+        for (j = 0; j < level_count; ++j)
+        {
+            winetest_push_context("level %u", j);
+
+            hr = IDirect3DDevice8_Clear(device, 0, NULL, D3DCLEAR_TARGET, 0xffff0000, 0.0, 0);
+            ok(hr == S_OK, "Got hr %#lx.\n", hr);
+
+            hr = IDirect3DDevice8_SetTextureStageState(device, 0, D3DTSS_MIPFILTER, D3DTEXF_POINT);
+            ok(hr == S_OK, "Got hr %#lx.\n", hr);
+            hr = IDirect3DDevice8_SetTextureStageState(device, 0, D3DTSS_MAXMIPLEVEL, j);
+            ok(hr == S_OK, "Got hr %#lx.\n", hr);
+
+            draw_textured_quad(&context, texture);
+            check_rt_color_todo_if(context.backbuffer, 0x00111111 * (j + 1), j > 0);
+
+            winetest_pop_context();
+        }
+
+        IDirect3DTexture8_Release(texture);
+
+        winetest_pop_context();
+    }
+    release_test_context(&context);
+}
+
 START_TEST(visual)
 {
     D3DADAPTER_IDENTIFIER8 identifier;
@@ -12208,4 +12281,5 @@ START_TEST(visual)
     test_dynamic_map_synchronization();
     test_filling_convention();
     test_managed_reset();
+    test_mipmap_upload();
 }
-- 
2.11.4.GIT