[PATCH v2 5/5] mf/tests: Add some WMA decoder ProcessOutput tests.

[Rémi Bernon]

Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=51931
Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=52391
Signed-off-by: Rémi Bernon <rbernon at codeweavers.com>
---
 dlls/mf/tests/mf.c | 121 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 121 insertions(+)

diff --git a/dlls/mf/tests/mf.c b/dlls/mf/tests/mf.c
index 6289bfbd993..72cd2e6860a 100644
--- a/dlls/mf/tests/mf.c
+++ b/dlls/mf/tests/mf.c
@@ -5858,11 +5858,13 @@ static void test_wma_decoder(void)
     MFT_REGISTER_TYPE_INFO output_type = {MFMediaType_Audio, MFAudioFormat_Float};
     MFT_OUTPUT_STREAM_INFO output_info;
     MFT_INPUT_STREAM_INFO input_info;
+    MFT_OUTPUT_DATA_BUFFER output;
     IMFMediaType *media_type;
     IMFTransform *transform;
     IMFSample *sample;
     GUID class_id;
     ULONG i, ret;
+    DWORD status;
     HRESULT hr;
 
     hr = CoInitialize(NULL);
@@ -6072,6 +6074,125 @@ static void test_wma_decoder(void)
     todo_wine
     ok(ret == 1, "Release returned %u\n", ret);
 
+    /* As output_info.dwFlags doesn't have MFT_OUTPUT_STREAM_CAN_PROVIDE_SAMPLES
+     * IMFTransform_ProcessOutput needs a sample or returns MF_E_TRANSFORM_NEED_MORE_INPUT */
+
+    status = 0xdeadbeef;
+    memset(&output, 0, sizeof(output));
+    hr = IMFTransform_ProcessOutput(transform, 0, 1, &output, &status);
+    todo_wine
+    ok(hr == MF_E_TRANSFORM_NEED_MORE_INPUT, "ProcessOutput returned %#x\n", hr);
+    ok(output.dwStreamID == 0, "got dwStreamID %u\n", output.dwStreamID);
+    ok(!output.pSample, "got pSample %p\n", output.pSample);
+    todo_wine
+    ok(output.dwStatus == MFT_OUTPUT_DATA_BUFFER_NO_SAMPLE,
+            "got dwStatus %#x\n", output.dwStatus);
+    ok(!output.pEvents, "got pEvents %p\n", output.pEvents);
+    todo_wine
+    ok(status == 0, "got status %#x\n", status);
+
+    sample = create_sample(wma_encoded_data[0], wma_block_size);
+    hr = IMFTransform_ProcessInput(transform, 0, sample, 0);
+    todo_wine
+    ok(hr == MF_E_NOTACCEPTING, "ProcessInput returned %#x\n", hr);
+    ret = IMFSample_Release(sample);
+    ok(ret == 0, "Release returned %u\n", ret);
+
+    status = 0xdeadbeef;
+    memset(&output, 0, sizeof(output));
+    hr = IMFTransform_ProcessOutput(transform, 0, 1, &output, &status);
+    todo_wine
+    ok(hr == MF_E_TRANSFORM_NEED_MORE_INPUT, "ProcessOutput returned %#x\n", hr);
+    ok(!output.pSample, "got pSample %p\n", output.pSample);
+    todo_wine
+    ok(output.dwStatus == MFT_OUTPUT_DATA_BUFFER_NO_SAMPLE,
+            "got dwStatus %#x\n", output.dwStatus);
+    todo_wine
+    ok(status == 0, "got status %#x\n", status);
+
+    status = 0xdeadbeef;
+    output_info.cbSize = sizeof(wma_decoded_data);
+    sample = create_sample(NULL, output_info.cbSize);
+    memset(&output, 0, sizeof(output));
+    output.pSample = sample;
+    hr = IMFTransform_ProcessOutput(transform, 0, 1, &output, &status);
+    while (hr == MF_E_TRANSFORM_NEED_MORE_INPUT)
+    {
+        ok(hr == MF_E_TRANSFORM_NEED_MORE_INPUT, "ProcessOutput returned %#x\n", hr);
+        ok(output.pSample == sample, "got pSample %p\n", output.pSample);
+        ok(output.dwStatus == 0, "got dwStatus %#x\n", output.dwStatus);
+        ok(status == 0, "got status %#x\n", status);
+        check_sample(sample, NULL, 0);
+        ret = IMFSample_Release(sample);
+        ok(ret == 0, "Release returned %u\n", ret);
+
+        sample = create_sample(wma_encoded_data[0], wma_block_size);
+        hr = IMFTransform_ProcessInput(transform, 0, sample, 0);
+        ok(hr == S_OK, "ProcessInput returned %#x\n", hr);
+        ret = IMFSample_Release(sample);
+        ok(ret == 1, "Release returned %u\n", ret);
+
+        status = 0xdeadbeef;
+        sample = create_sample(NULL, output_info.cbSize);
+        memset(&output, 0, sizeof(output));
+        output.pSample = sample;
+        hr = IMFTransform_ProcessOutput(transform, 0, 1, &output, &status);
+    }
+
+    todo_wine
+    ok(hr == S_OK, "ProcessOutput returned %#x\n", hr);
+    ok(output.pSample == sample, "got pSample %p\n", output.pSample);
+
+    i = 0;
+    while (output.dwStatus == MFT_OUTPUT_DATA_BUFFER_INCOMPLETE)
+    {
+        ok(hr == S_OK, "ProcessOutput returned %#x\n", hr);
+        ok(output.pSample == sample, "got pSample %p\n", output.pSample);
+        ok(output.dwStatus == MFT_OUTPUT_DATA_BUFFER_INCOMPLETE, "got dwStatus %#x\n",
+                output.dwStatus);
+        ok(status == 0, "got status %#x\n", status);
+        check_sample(sample, wma_decoded_data, sizeof(wma_decoded_data));
+        i += sizeof(wma_decoded_data);
+        ret = IMFSample_Release(sample);
+        ok(ret == 0, "Release returned %u\n", ret);
+
+        status = 0xdeadbeef;
+        sample = create_sample(NULL, output_info.cbSize);
+        memset(&output, 0, sizeof(output));
+        output.pSample = sample;
+        hr = IMFTransform_ProcessOutput(transform, 0, 1, &output, &status);
+    }
+
+    todo_wine
+    ok(hr == S_OK, "ProcessOutput returned %#x\n", hr);
+    ok(output.pSample == sample, "got pSample %p\n", output.pSample);
+    ok(output.dwStatus == 0, "got dwStatus %#x\n", output.dwStatus);
+    todo_wine
+    ok(status == 0, "got status %#x\n", status);
+    if (hr == S_OK)
+    {
+        check_sample(sample, wma_decoded_data, sizeof(wma_decoded_data) / 2);
+        i += sizeof(wma_decoded_data) / 2;
+        ok(i == 0xe000, "ProcessOutput produced %#x bytes\n", i);
+    }
+    ret = IMFSample_Release(sample);
+    ok(ret == 0, "Release returned %u\n", ret);
+
+    status = 0xdeadbeef;
+    sample = create_sample(NULL, output_info.cbSize);
+    memset(&output, 0, sizeof(output));
+    output.pSample = sample;
+    hr = IMFTransform_ProcessOutput(transform, 0, 1, &output, &status);
+    todo_wine
+    ok(hr == MF_E_TRANSFORM_NEED_MORE_INPUT, "ProcessOutput returned %#x\n", hr);
+    ok(output.pSample == sample, "got pSample %p\n", output.pSample);
+    ok(output.dwStatus == 0, "got dwStatus %#x\n", output.dwStatus);
+    todo_wine
+    ok(status == 0, "got status %#x\n", status);
+    check_sample(sample, NULL, 0);
+    ret = IMFSample_Release(sample);
+    ok(ret == 0, "Release returned %u\n", ret);
+
     sample = create_sample(wma_encoded_data[0], wma_block_size);
     hr = IMFTransform_ProcessInput(transform, 0, sample, 0);
     todo_wine
-- 
2.34.1