[PATCH 2/3] winegstreamer: Implement stream_sink_type_handler_GetMediaTypeCount.

[Ziqing Hui]

From: Ziqing Hui

---
 dlls/mf/tests/mf.c              | 4 ++--
 dlls/winegstreamer/media_sink.c | 8 ++++++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/dlls/mf/tests/mf.c b/dlls/mf/tests/mf.c
index 53f22809dc4..b9363163370 100644
--- a/dlls/mf/tests/mf.c
+++ b/dlls/mf/tests/mf.c
@@ -6024,10 +6024,10 @@ static void test_mpeg4_media_sink(void)
     ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
     ok(IsEqualGUID(&guid, &MFMediaType_Audio), "Unexpected major type.\n");

+    hr = IMFMediaTypeHandler_GetMediaTypeCount(type_handler, NULL);
+    ok(hr == E_POINTER, "Unexpected hr %#lx.\n", hr);
     hr = IMFMediaTypeHandler_GetMediaTypeCount(type_handler, &count);
-    todo_wine
     ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
-    todo_wine
     ok(count == 1, "Unexpected count %lu.\n", count);

hr = IMFMediaTypeHandler_GetCurrentMediaType(type_handler, &media_type);
diff --git a/dlls/winegstreamer/media_sink.c b/dlls/winegstreamer/media_sink.c
index 3ee03d663d0..e8ed90951bf 100644
--- a/dlls/winegstreamer/media_sink.c
+++ b/dlls/winegstreamer/media_sink.c
@@ -422,9 +422,13 @@ static HRESULT WINAPI stream_sink_type_handler_IsMediaTypeSupported(IMFMediaType

static HRESULT WINAPI stream_sink_type_handler_GetMediaTypeCount(IMFMediaTypeHandler *iface, DWORD *count)
 {
-    FIXME("iface %p, count %p.\n", iface, count);
+    TRACE("iface %p, count %p.\n", iface, count);

-    return E_NOTIMPL;
+    if (!count)
+        return E_POINTER;
+
+    *count = 1;
+    return S_OK;
 }

static HRESULT WINAPI stream_sink_type_handler_GetMediaTypeByIndex(IMFMediaTypeHandler *iface, DWORD index,