[PATCH 1/2] comctl32/tests: Add more treeview background tests.

[Zhiyi Zhang]

From: Zhiyi Zhang

Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=56861
---
 dlls/comctl32/tests/Makefile.in |  2 +-
 dlls/comctl32/tests/misc.c      | 23 +++++++++++++++++++++++
 2 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/dlls/comctl32/tests/Makefile.in b/dlls/comctl32/tests/Makefile.in
index c9b0d5c578f..0c96203a2f6 100644
--- a/dlls/comctl32/tests/Makefile.in
+++ b/dlls/comctl32/tests/Makefile.in
@@ -1,5 +1,5 @@
 TESTDLL   = comctl32.dll
-IMPORTS   = ole32 user32 gdi32 advapi32 imm32
+IMPORTS   = ole32 user32 gdi32 advapi32 imm32 uxtheme

SOURCES = \
    animate.c \
diff --git a/dlls/comctl32/tests/misc.c b/dlls/comctl32/tests/misc.c
index ff6336f31af..ed31dfb8719 100644
--- a/dlls/comctl32/tests/misc.c
+++ b/dlls/comctl32/tests/misc.c
@@ -21,6 +21,7 @@
 #include <stdio.h>
 #include <windows.h>
 #include <commctrl.h>
+#include <uxtheme.h>

#include "wine/test.h"
 #include "v6util.h"
@@ -829,6 +830,7 @@ static void test_themed_background(void)
     HMODULE uxtheme;
     COLORREF color;
     WNDCLASSA cls;
+    HRESULT hr;
     HDC hdc;
     int i;

@@ -838,6 +840,7 @@ static void test_themed_background(void)
         DWORD style;
         const struct message *seq;
         BOOL todo;
+        const WCHAR *window_theme;
     }
     tests[] =
     {
@@ -886,6 +889,8 @@ static void test_themed_background(void)
         {TOOLTIPS_CLASSA, 0, empty_seq},
         {TRACKBAR_CLASSA, 0, wm_ctlcolorstatic_seq},
         {WC_TREEVIEWA, 0, treeview_seq},
+        {WC_TREEVIEWA, TVS_HASBUTTONS, treeview_seq},
+        {WC_TREEVIEWA, TVS_HASBUTTONS, treeview_seq, TRUE, L"explorer"},
         {UPDOWN_CLASSA, 0, empty_seq},
         {WC_SCROLLBARA, 0, scrollbar_seq},
         {WC_SCROLLBARA, SBS_SIZEBOX, empty_seq},
@@ -933,12 +938,30 @@ static void test_themed_background(void)
         child = CreateWindowA(tests[i].class_name, "    ", WS_CHILD | WS_VISIBLE | tests[i].style,
                               0, 0, 50, 100, parent, 0, 0, 0);
         ok(child != NULL, "CreateWindowA failed, error %lu.\n", GetLastError());
+
+        /* Extra preparations */
+        if (tests[i].window_theme)
+        {
+            hr = SetWindowTheme(child, tests[i].window_theme, NULL);
+            ok(hr == S_OK, "Got unexpected hr %#lx.\n", hr);
+        }
+
         flush_events();
         flush_sequences(sequences, NUM_MSG_SEQUENCES);

RedrawWindow(child, NULL, NULL, RDW_INVALIDATE | RDW_ERASE | RDW_UPDATENOW | RDW_ERASENOW | RDW_FRAME);
         ok_sequence(sequences, PARENT_SEQ_INDEX, tests[i].seq, "paint background", tests[i].todo);

+        /* Extra background tests */
+        if (!lstrcmpA(tests[i].class_name, WC_TREEVIEWA))
+        {
+            hdc = GetDC(child);
+            color = GetPixel(hdc, 40, 40);
+            todo_wine_if(tests[i].window_theme && !lstrcmpW(tests[i].window_theme, L"explorer"))
+            ok(color == 0xffffff, "Expected color %#x, got %#lx.\n", 0xffffff, color);
+            ReleaseDC(child, hdc);
+        }
+
         /* For message sequences that contain both DrawThemeParentBackground() messages and
          * WM_CTLCOLOR*, do a color test to check which is really in effect for controls that can be
          * tested automatically. For WM_ERASEBKGND from DrawThemeParentBackground(), a red brush is