[PATCH v9 2/2] Added tests for class FILE_INFORMATION_ALLOCATION in setFileInformationByHandle

[Stéphane Bacri]

From: Stéphane Bacri

---
 dlls/ntdll/tests/file.c | 46 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/dlls/ntdll/tests/file.c b/dlls/ntdll/tests/file.c
index cc0afdbdb64..a4fbd3d7e95 100644
--- a/dlls/ntdll/tests/file.c
+++ b/dlls/ntdll/tests/file.c
@@ -1334,6 +1334,49 @@ static void test_file_io_completion(void)
     pNtClose( h );
 }

+static void test_file_setFileAllocationInformation(void) {
+    FILE_ALLOCATION_INFORMATION fas;
+    BY_HANDLE_FILE_INFORMATION info;
+    IO_STATUS_BLOCK io;
+    HANDLE handle;
+    int res;
+    char* buffer;
+    DWORD bytesWritten, bytesRead;
+    DWORD length = 64*1024;
+    LARGE_INTEGER distanceToMove;
+    distanceToMove.QuadPart = 0;
+
+    if( !(handle = create_temp_file(0)) ) return;
+
+    memset( &fas, 0, sizeof(fas) );
+    fas.AllocationSize.QuadPart = length;
+    res = pNtSetInformationFile( handle, &io, &fas, sizeof fas, FileAllocationInformation );
+
+    ok ( res == STATUS_SUCCESS, "file allocation failed, NtSetInformationFile returned %x\n", res );
+    ok ( io.Status == STATUS_SUCCESS, "file allocation failed, io.Status is %lx\n", io.Status );
+
+    memset( &info, 0x22, sizeof(info) );
+    res = GetFileInformationByHandle( handle, &info );
+    ok( res, "GetFileInformationByHandle failed\n" );
+    ok( info.nFileSizeLow == 0, "incorrect file size after allocation: should be 0, got %lx\n", info.nFileSizeLow );
+
+    buffer = malloc(length);
+    memset( buffer, 0xa5, length );
+    res = WriteFile( handle, buffer, length, &bytesWritten, NULL );
+    ok ( res != 0 , "couldn't write to file after allocation\n");
+    ok ( bytesWritten == length, "couldn't write to whole or part of file after allocation: allocated %lx, written %lx\n", length, bytesWritten );
+
+    memset( buffer, 0, length );
+    SetFilePointerEx( handle, distanceToMove, NULL, FILE_BEGIN );
+    res = ReadFile( handle, buffer, length, &bytesRead, NULL );
+    ok ( res != 0 , "couldn't read file after allocation\n" );
+    ok ( bytesRead == length, "couldn't read whole or part of file after allocation: allocated %lx, read %lx\n", length, bytesRead );
+    ok ( *buffer == (char)0xa5, "data read from allocated file doesn't match written pattern: wrote 0xa5, got 0x%x\n", *buffer & 0xff );
+
+    free(buffer);
+    CloseHandle( handle );
+}
+
 static void test_file_full_size_information(void)
 {
     IO_STATUS_BLOCK io;
@@ -6185,6 +6228,9 @@ START_TEST(file)
     test_file_access_information();
     test_file_attribute_tag_information();
     test_file_stat_information();
+#ifndef __APPLE__
+    test_file_setFileAllocationInformation();
+#endif
     test_dotfile_file_attributes();
     test_file_mode();
     test_file_readonly_access();