Rémi Bernon : kernel32/tests: Add some GetProcessHeaps tests.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 13a4da538ce9132394d5a6fd16a8fa75febd0b8f
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=13a4da538ce9132394d5a6fd1...

Author: Rémi Bernon 
Date:   Fri Apr  8 16:06:48 2022 +0200

kernel32/tests: Add some GetProcessHeaps tests.

Signed-off-by: Rémi Bernon 
Signed-off-by: Alexandre Julliard

---

dlls/kernel32/tests/heap.c | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/dlls/kernel32/tests/heap.c b/dlls/kernel32/tests/heap.c
index 190e83cc208..80e5d588047 100644
--- a/dlls/kernel32/tests/heap.c
+++ b/dlls/kernel32/tests/heap.c
@@ -80,23 +80,40 @@ static void test_HeapCreate(void)
 {
     static const BYTE buffer[512] = {0};
     SIZE_T alloc_size = 0x8000 * sizeof(void *), size, i;
+    HANDLE heap, heap1, heaps[8];
     BYTE *ptr, *ptr1, *ptrs[128];
-    HANDLE heap, heap1;
+    DWORD heap_count, count;
     UINT_PTR align;
     BOOL ret;

+    heap_count = GetProcessHeaps( 0, NULL );
+    ok( heap_count <= 6, "GetProcessHeaps returned %lu\n", heap_count );
+
     /* check heap alignment */

heap = HeapCreate( 0, 0, 0 );
     ok( !!heap, "HeapCreate failed, error %lu\n", GetLastError() );
     ok( !((ULONG_PTR)heap & 0xffff), "wrong heap alignment\n" );
+    count = GetProcessHeaps( 0, NULL );
+    ok( count == heap_count + 1, "GetProcessHeaps returned %lu\n", count );
     heap1 = HeapCreate( 0, 0, 0 );
     ok( !!heap, "HeapCreate failed, error %lu\n", GetLastError() );
     ok( !((ULONG_PTR)heap1 & 0xffff), "wrong heap alignment\n" );
+    count = GetProcessHeaps( 0, NULL );
+    ok( count == heap_count + 2, "GetProcessHeaps returned %lu\n", count );
+    count = GetProcessHeaps( ARRAY_SIZE(heaps), heaps );
+    ok( count == heap_count + 2, "GetProcessHeaps returned %lu\n", count );
+    ok( heaps[0] == GetProcessHeap(), "got wrong heap\n" );
+    todo_wine
+    ok( heaps[heap_count + 0] == heap, "got wrong heap\n" );
+    todo_wine
+    ok( heaps[heap_count + 1] == heap1, "got wrong heap\n" );
     ret = HeapDestroy( heap1 );
     ok( ret, "HeapDestroy failed, error %lu\n", GetLastError() );
     ret = HeapDestroy( heap );
     ok( ret, "HeapDestroy failed, error %lu\n", GetLastError() );
+    count = GetProcessHeaps( 0, NULL );
+    ok( count == heap_count, "GetProcessHeaps returned %lu\n", count );

/* growable heap */