[gadamopoulos] 73884: [COMCTL32] -Fix a couple more button tests.

Author: gadamopoulos
Date: Wed Feb 22 13:24:17 2017
New Revision: 73884

URL: http://svn.reactos.org/svn/reactos?rev=73884&view=rev
Log:
[COMCTL32] -Fix a couple more button tests.

Modified:
    trunk/reactos/dll/win32/comctl32/button.c

Modified: trunk/reactos/dll/win32/comctl32/button.c
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/comctl32/button.c...
==============================================================================
--- trunk/reactos/dll/win32/comctl32/button.c	[iso-8859-1] (original)
+++ trunk/reactos/dll/win32/comctl32/button.c	[iso-8859-1] Wed Feb 22 13:24:17 2017
@@ -626,7 +626,10 @@
     case WM_SETFOCUS:
         TRACE("WM_SETFOCUS %p\n",hWnd);
         set_button_state( hWnd, get_button_state(hWnd) | BST_FOCUS );
-        InvalidateRect(hWnd, NULL, FALSE);
+        if (btn_type == BS_OWNERDRAW)
+            paint_button( hWnd, btn_type, ODA_FOCUS );
+        else
+            InvalidateRect(hWnd, NULL, FALSE);
         if (style & BS_NOTIFY)
             BUTTON_NOTIFY_PARENT(hWnd, BN_SETFOCUS);
         break;
@@ -635,7 +638,6 @@
         TRACE("WM_KILLFOCUS %p\n",hWnd);
         state = get_button_state( hWnd );
         set_button_state( hWnd, state & ~BST_FOCUS );
-	paint_button( hWnd, btn_type, ODA_FOCUS );

if ((state & BUTTON_BTNPRESSED) && GetCapture() == hWnd)
             ReleaseCapture();