Piotr Caban : winmm: Add RIFF creation tests.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: f6b46db1c7b0e21b4c02deafee2c015a6fbbb4ed
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=f6b46db1c7b0e21b4c02deafee2c015a6fbbb4ed

Author: Piotr Caban <piotr at codeweavers.com>
Date:   Fri Feb 20 15:42:19 2015 +0100

winmm: Add RIFF creation tests.

---

 dlls/winmm/tests/mmio.c | 83 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)

diff --git a/dlls/winmm/tests/mmio.c b/dlls/winmm/tests/mmio.c
index cf02c0d..89146ef 100644
--- a/dlls/winmm/tests/mmio.c
+++ b/dlls/winmm/tests/mmio.c
@@ -906,6 +906,88 @@ static void test_mmio_buffer_pointer(void)
     DeleteFileA(test_file);
 }
 
+static void test_riff_write(void)
+{
+    static const DWORD test_write_data[] =
+    {
+        FOURCC_RIFF, 0x28, mmioFOURCC('W','A','V','E'),  mmioFOURCC('d','a','t','a'),
+        0x1b, 0xdededede, 0xdededede, 0xefefefef,
+        0xefefefef, 0xbabababa, 0xbabababa, 0xefefef
+    };
+
+    char name[] = "test_write.wav";
+    char buf[256];
+    MMCKINFO chunk_info[2];
+    MMIOINFO info;
+    HMMIO mmio;
+    MMRESULT ret;
+    LONG written;
+    DWORD read;
+    HANDLE h;
+
+    memset(chunk_info, 0, sizeof(chunk_info));
+
+    mmio = mmioOpenA(name, NULL, MMIO_ALLOCBUF|MMIO_CREATE|MMIO_READWRITE);
+    ok(mmio != NULL, "mmioOpen failed\n");
+
+    chunk_info[0].fccType = mmioFOURCC('W','A','V','E');
+    ret = mmioCreateChunk(mmio, chunk_info, MMIO_CREATERIFF);
+    ok(ret == MMSYSERR_NOERROR, "mmioCreateChunk failed %x\n", ret);
+    ok(chunk_info[0].ckid == FOURCC_RIFF, "chunk_info[0].ckid = %x\n", chunk_info[0].ckid);
+    ok(chunk_info[0].cksize == 0, "chunk_info[0].cksize = %d\n", chunk_info[0].cksize);
+    ok(chunk_info[0].dwDataOffset == 8, "chunk_info[0].dwDataOffset = %d\n", chunk_info[0].dwDataOffset);
+    ok(chunk_info[0].dwFlags == MMIO_DIRTY, "chunk_info[0].dwFlags = %x\n", chunk_info[0].dwFlags);
+
+    chunk_info[1].ckid = mmioFOURCC('d','a','t','a');
+    ret = mmioCreateChunk(mmio, chunk_info+1, 0);
+    ok(ret == MMSYSERR_NOERROR, "mmioCreateChunk failed %x\n", ret);
+    ok(chunk_info[1].ckid == mmioFOURCC('d','a','t','a'), "chunk_info[1].ckid = %x\n", chunk_info[1].ckid);
+    ok(chunk_info[1].cksize == 0, "chunk_info[1].cksize = %d\n", chunk_info[1].cksize);
+    ok(chunk_info[1].dwDataOffset == 20, "chunk_info[1].dwDataOffset = %d\n", chunk_info[1].dwDataOffset);
+    ok(chunk_info[1].dwFlags == MMIO_DIRTY, "chunk_info[1].dwFlags = %x\n", chunk_info[1].dwFlags);
+
+    memset(buf, 0xde, sizeof(buf));
+    written = mmioWrite(mmio, buf, 8);
+    ok(written == 8, "mmioWrite failed %x\n", ret);
+
+    ret = mmioGetInfo(mmio, &info, 0);
+    ok(ret == MMSYSERR_NOERROR, "mmioGetInfo failed %x\n", ret);
+
+    memset(info.pchNext, 0xef, 8);
+    info.pchNext += 8;
+    ret = mmioAdvance(mmio, &info, 1);
+    ok(ret == MMSYSERR_NOERROR, "mmioAdvance failed %x\n", ret);
+    ok(info.lBufOffset == 36, "info.lBufOffset = %d\n", info.lBufOffset);
+
+    info.dwFlags |= MMIO_DIRTY;
+    memset(info.pchNext, 0xba, 8);
+    info.pchNext += 8;
+    ret = mmioAdvance(mmio, &info, 1);
+    ok(ret == MMSYSERR_NOERROR, "mmioAdvance failed %x\n", ret);
+    ok(info.lBufOffset == 44, "info.lBufOffset = %d\n", info.lBufOffset);
+
+    info.dwFlags |= MMIO_DIRTY;
+    memset(info.pchNext, 0xef, 3);
+    info.pchNext += 3;
+    ret = mmioSetInfo(mmio, &info, 0);
+    ok(ret == MMSYSERR_NOERROR, "mmioSetInfo failed %x\n", ret);
+
+    ret = mmioAscend(mmio, chunk_info+1, 0);
+    ok(ret == MMSYSERR_NOERROR, "mmioAscend failed %x\n", ret);
+    ret = mmioAscend(mmio, chunk_info, 0);
+    ok(ret == MMSYSERR_NOERROR, "mmioAscend failed %x\n", ret);
+    ret = mmioClose(mmio, 0);
+    ok(ret == MMSYSERR_NOERROR, "mmioClose failed %x\n", ret);
+
+    h = CreateFileA("test_write.wav", GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
+    ok(h != INVALID_HANDLE_VALUE, "CreateFile failed\n");
+    ok(ReadFile(h, buf, sizeof(buf), &read, NULL), "ReadFile failed\n");
+    CloseHandle(h);
+    ok(!memcmp(buf, test_write_data, sizeof(test_write_data)), "created file is incorrect\n");
+
+    DeleteFileA("test_write.wav");
+}
+
 START_TEST(mmio)
 {
     /* Make it possible to run the tests against a specific AVI file in
@@ -924,4 +1006,5 @@ START_TEST(mmio)
     test_mmioSeek();
     test_mmio_end_of_file();
     test_mmio_buffer_pointer();
+    test_riff_write();
 }