[PATCH 1/2] user32/tests: Test NUMCOLORS for display DCs.

[Zhiyi Zhang]

Signed-off-by: Zhiyi Zhang <zzhang at codeweavers.com>
---
 dlls/user32/tests/monitor.c | 79 ++++++++++++++++++++++++++++++++++++-
 1 file changed, 78 insertions(+), 1 deletion(-)

diff --git a/dlls/user32/tests/monitor.c b/dlls/user32/tests/monitor.c
index d0d4ff1f324..1d88afafb18 100644
--- a/dlls/user32/tests/monitor.c
+++ b/dlls/user32/tests/monitor.c
@@ -2193,8 +2193,10 @@ static void _check_display_dc(INT line, HDC hdc, const DEVMODEA *dm, BOOL allow_
 static void test_display_dc(void)
 {
     static const INT bpps[] = {1, 4, 8, 16, 24, 32};
+    unsigned char buffer[sizeof(BITMAPINFOHEADER) + 256 * sizeof(RGBQUAD)];
+    BITMAPINFO *bmi = (BITMAPINFO *)buffer;
+    INT count, old_count, i, bpp, value;
     HBITMAP hbitmap, old_hbitmap;
-    INT count, old_count, i, bpp;
     DWORD device_idx, mode_idx;
     DEVMODEA dm, dm2, dm3;
     DISPLAY_DEVICEA dd;
@@ -2247,6 +2249,81 @@ static void test_display_dc(void)
         winetest_pop_context();
     }
 
+    /* Test selecting a DIB of a different depth into a display compatible DC */
+    for (i = 0; i < ARRAY_SIZE(bpps); ++i)
+    {
+        winetest_push_context("bpp %d", bpps[i]);
+
+        mem_dc = CreateCompatibleDC(hdc);
+        memset(buffer, 0, sizeof(buffer));
+        bmi->bmiHeader.biSize = sizeof(bmi->bmiHeader);
+        bmi->bmiHeader.biWidth = dm.dmPelsWidth;
+        bmi->bmiHeader.biHeight = dm.dmPelsHeight;
+        bmi->bmiHeader.biBitCount = bpps[i];
+        bmi->bmiHeader.biPlanes = 1;
+        bmi->bmiHeader.biCompression = BI_RGB;
+        hbitmap = CreateDIBSection(hdc, bmi, DIB_RGB_COLORS, NULL, NULL, 0);
+        ok(!!hbitmap, "CreateDIBSection failed, error %ld.\n", GetLastError());
+        old_hbitmap = SelectObject(mem_dc, hbitmap);
+        ok(!!old_hbitmap, "Failed to select bitmap.\n");
+
+        value = GetDeviceCaps(mem_dc, BITSPIXEL);
+        ok(value == 32, "Expected 32, got %d.\n", value);
+        value = GetDeviceCaps(mem_dc, NUMCOLORS);
+        ok(value == -1, "Expected -1, got %d.\n", value);
+
+        SelectObject(mem_dc, old_hbitmap);
+        DeleteObject(hbitmap);
+        DeleteDC(mem_dc);
+
+        winetest_pop_context();
+    }
+
+    /* Test NUMCOLORS for display DCs */
+    for (i = 0; i < ARRAY_SIZE(bpps); ++i)
+    {
+        winetest_push_context("bpp %d", bpps[i]);
+
+        memset(&dm2, 0, sizeof(dm2));
+        dm2.dmSize = sizeof(dm2);
+        for (mode_idx = 0; EnumDisplaySettingsA(NULL, mode_idx, &dm2); ++mode_idx)
+        {
+            if (dm2.dmBitsPerPel == bpps[i])
+                break;
+        }
+        if (dm2.dmBitsPerPel != bpps[i])
+        {
+            skip("%d-bit display mode not found.\n", bpps[i]);
+            winetest_pop_context();
+            continue;
+        }
+
+        res = ChangeDisplaySettingsExA(NULL, &dm2, NULL, CDS_RESET, NULL);
+        /* Win8 TestBots */
+        ok(res == DISP_CHANGE_SUCCESSFUL || broken(res == DISP_CHANGE_FAILED),
+                "ChangeDisplaySettingsExA returned unexpected %ld.\n", res);
+        if (res == DISP_CHANGE_SUCCESSFUL)
+        {
+            value = GetDeviceCaps(hdc, BITSPIXEL);
+            ok(value == (bpps[i] == 4 ? 1 : bpps[i]), "Expected %d, got %d.\n",
+                    (bpps[i] == 4 ? 1 : bpps[i]), value);
+
+            value = GetDeviceCaps(hdc, NUMCOLORS);
+            if (bpps[i] > 8 || (bpps[i] == 8 && LOBYTE(LOWORD(GetVersion())) >= 6))
+                todo_wine_if(bpps[i] == 8 && LOBYTE(LOWORD(GetVersion())) >= 6)
+                ok(value == -1, "Expected -1, got %d.\n", value);
+            else if (bpps[i] == 8 && LOBYTE(LOWORD(GetVersion())) < 6)
+                ok(value > 16 && value <= 256, "Got %d.\n", value);
+            else
+                ok(value == 1 << bpps[i], "Expected %d, got %d.\n", 1 << bpps[i], value);
+
+            res = ChangeDisplaySettingsExA(NULL, NULL, NULL, 0, NULL);
+            ok(res == DISP_CHANGE_SUCCESSFUL, "ChangeDisplaySettingsExA returned unexpected %ld.\n",
+                    res);
+        }
+        winetest_pop_context();
+    }
+
     /* Test selecting a DDB of the same color depth into a display compatible DC */
     mem_dc = CreateCompatibleDC(hdc);
     bpp = GetDeviceCaps(mem_dc, BITSPIXEL);
-- 
2.32.0