[PATCH 2/6] dinput8/tests: Test EnumDevicesBySemantics mouse and keyboard enumeration with flags.

[Lucas Fialho Zawacki]

---
 dlls/dinput8/tests/device.c |   70 ++++++++++++++++++++++++++++++++-----------
 1 files changed, 52 insertions(+), 18 deletions(-)

diff --git a/dlls/dinput8/tests/device.c b/dlls/dinput8/tests/device.c
index c18a83a..256c275 100644
--- a/dlls/dinput8/tests/device.c
+++ b/dlls/dinput8/tests/device.c
@@ -27,13 +27,58 @@
 #include "initguid.h"
 #include "dinput.h"
 
-static BOOL CALLBACK enum_by_semantics(
+struct enum_data {
+    LPDIRECTINPUT8 pDI;
+    LPDIACTIONFORMAT lpdiaf;
+    LPDIRECTINPUTDEVICE8 keyboard;
+    LPDIRECTINPUTDEVICE8 mouse;
+    int ndevices;
+};
+
+/* Dummy GUID */
+const GUID ACTION_MAPPING_GUID = { 0x1, 0x2, 0x3, { 0x4, 0x5, 0x6, 0x7, 0x8, 0x9, 0xa, 0xb } };
+
+DIACTION actionMapping[]=
+{
+  /* axis */
+  { 0, 0x01008A01 /* DIAXIS_DRIVINGR_STEER */ , 0, { "Steer" } },
+  /* button */
+  { 1, 0x01000C01 /* DIBUTTON_DRIVINGR_SHIFTUP */ , 0, { "Upshift" } },
+  /* keyboard key */
+  { 2, DIKEYBOARD_SPACE , 0, { "Missile" } },
+  /* mouse button */
+  { 3, DIMOUSE_BUTTON0 , 0, { "Select" } }
+};
+
+static BOOL CALLBACK enumeration_callback(
     LPCDIDEVICEINSTANCE lpddi,
     LPDIRECTINPUTDEVICE8 lpdid,
     DWORD dwFlags,
     DWORD dwRemaining,
     LPVOID pvRef)
 {
+    struct enum_data *data = pvRef;
+    if (!data) return DIENUM_CONTINUE;
+
+    data->ndevices++;
+
+    /* collect the mouse and keyboard */
+    if ( IsEqualGUID(&lpddi->guidInstance, &GUID_SysKeyboard) )
+    {
+        IDirectInputDevice_AddRef(lpdid);
+        data->keyboard = lpdid;
+
+        ok (dwFlags & DIEDBS_MAPPEDPRI1, "Keyboard should be mapped as pri1 dwFlags=%08x\n",dwFlags);
+    }
+
+    if ( IsEqualGUID(&lpddi->guidInstance, &GUID_SysMouse) )
+    {
+        IDirectInputDevice_AddRef(lpdid);
+        data->mouse = lpdid;
+
+        ok (dwFlags & DIEDBS_MAPPEDPRI1, "Mouse should be mapped as pri1 dwFlags=%08x\n",dwFlags);
+    }
+
     return DIENUM_CONTINUE;
 }
 
@@ -44,17 +89,7 @@ static void test_action_mapping(void)
     HINSTANCE hinst = GetModuleHandle(NULL);
     LPDIRECTINPUT8 pDI = NULL;
     DIACTIONFORMAT af;
-    /* Dummy GUID */
-    const GUID ACTION_MAPPING_GUID = { 0x1, 0x2, 0x3, { 0x4, 0x5, 0x6, 0x7, 0x8, 0x9, 0xa, 0xb } };
-
-    DIACTION actionMapping[]=
-    {
-      /* axis */
-      { 0, 0x01008A01 /* DIAXIS_DRIVINGR_STEER */ , 0, { "Steer" } },
-
-      /* button */
-      { 1, 0x01000C01 /* DIBUTTON_DRIVINGR_SHIFTUP */ , 0, { "Upshift" } }
-    };
+    struct enum_data data = { pDI, &af, NULL, NULL, 0};
 
     hr = CoCreateInstance(&CLSID_DirectInput8, 0, 1, &IID_IDirectInput8A, (LPVOID*)&pDI);
     if (hr == DIERR_OLDDIRECTINPUTVERSION ||
@@ -85,16 +120,15 @@ static void test_action_mapping(void)
     af.guidActionMap = ACTION_MAPPING_GUID;
     af.dwGenre = 0x01000000; /* DIVIRTUAL_DRIVING_RACE */
 
-    hr = IDirectInput8_EnumDevicesBySemantics(pDI,0, &af,
-        enum_by_semantics, 0, 0);
-
+    hr = IDirectInput8_EnumDevicesBySemantics(pDI,0, &af, enumeration_callback, &data, 0);
     ok(SUCCEEDED(hr), "EnumDevicesBySemantics failed: hr=%08x\n",hr);
+    ok( data.ndevices > 0, "EnumDevicesBySemantics did not call the callback hr=%08x\n",hr);
+    ok( data.keyboard != NULL, "EnumDevicesBySemantics should enumerate the keyboard\n");
+    ok( data.mouse != NULL, "EnumDevicesBySemantics should enumerate the mouse\n");
 
     /* The call fails with a zeroed GUID */
     memset(&af.guidActionMap, 0, sizeof(GUID));
-    hr = IDirectInput8_EnumDevicesBySemantics(pDI,0, &af,
-        enum_by_semantics, 0, 0);
-
+    hr = IDirectInput8_EnumDevicesBySemantics(pDI,0, &af, enumeration_callback, 0, 0);
     todo_wine ok(FAILED(hr), "EnumDevicesBySemantics succeeded with invalid GUID hr=%08x\n", hr);
 
 }
-- 
1.7.0.4