[PATCH v2 3/5] quartz/tests: Add tests for IBaseFilter::EnumPins() on the VMR9 filter.

[Zebediah Figura]

Signed-off-by: Zebediah Figura <z.figura12 at gmail.com>
---
 dlls/quartz/tests/vmr9.c | 152 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 152 insertions(+)

diff --git a/dlls/quartz/tests/vmr9.c b/dlls/quartz/tests/vmr9.c
index 7ecf85252a..cbaf081868 100644
--- a/dlls/quartz/tests/vmr9.c
+++ b/dlls/quartz/tests/vmr9.c
@@ -42,6 +42,13 @@ static IBaseFilter *create_vmr9(DWORD mode)
     return filter;
 }
 
+static ULONG get_refcount(void *iface)
+{
+    IUnknown *unknown = iface;
+    IUnknown_AddRef(unknown);
+    return IUnknown_Release(unknown);
+}
+
 static void test_filter_config(void)
 {
     IVMRFilterConfig9 *config;
@@ -235,6 +242,150 @@ static void test_interfaces(void)
     ok(!ref, "Got outstanding refcount %d.\n", ref);
 }
 
+static HRESULT set_mixing_mode(IBaseFilter *filter)
+{
+    IVMRFilterConfig9 *config;
+    HRESULT hr;
+
+    hr = IBaseFilter_QueryInterface(filter, &IID_IVMRFilterConfig9, (void **)&config);
+    ok(hr == S_OK, "Got hr %#x.\n", hr);
+
+    hr = IVMRFilterConfig9_SetNumberOfStreams(config, 2);
+    todo_wine ok(hr == S_OK, "Got hr %#x.\n", hr);
+
+    IVMRFilterConfig9_Release(config);
+    return hr;
+}
+
+static void test_enum_pins(void)
+{
+    IBaseFilter *filter = create_vmr9(0);
+    IEnumPins *enum1, *enum2;
+    ULONG count, ref;
+    IPin *pins[3];
+    HRESULT hr;
+
+    ref = get_refcount(filter);
+    ok(ref == 1, "Got unexpected refcount %d.\n", ref);
+
+    hr = IBaseFilter_EnumPins(filter, NULL);
+    ok(hr == E_POINTER, "Got hr %#x.\n", hr);
+
+    hr = IBaseFilter_EnumPins(filter, &enum1);
+    ok(hr == S_OK, "Got hr %#x.\n", hr);
+    ref = get_refcount(filter);
+    ok(ref == 2, "Got unexpected refcount %d.\n", ref);
+    ref = get_refcount(enum1);
+    ok(ref == 1, "Got unexpected refcount %d.\n", ref);
+
+    hr = IEnumPins_Next(enum1, 1, NULL, NULL);
+    ok(hr == E_POINTER, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Next(enum1, 1, pins, NULL);
+    ok(hr == S_OK, "Got hr %#x.\n", hr);
+    ref = get_refcount(filter);
+    todo_wine ok(ref == 3, "Got unexpected refcount %d.\n", ref);
+    ref = get_refcount(pins[0]);
+    todo_wine ok(ref == 3, "Got unexpected refcount %d.\n", ref);
+    ref = get_refcount(enum1);
+    ok(ref == 1, "Got unexpected refcount %d.\n", ref);
+    IPin_Release(pins[0]);
+    ref = get_refcount(filter);
+    ok(ref == 2, "Got unexpected refcount %d.\n", ref);
+
+    hr = IEnumPins_Next(enum1, 1, pins, NULL);
+    ok(hr == S_FALSE, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Reset(enum1);
+    ok(hr == S_OK, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Next(enum1, 1, pins, &count);
+    ok(hr == S_OK, "Got hr %#x.\n", hr);
+    ok(count == 1, "Got count %u.\n", count);
+    IPin_Release(pins[0]);
+
+    hr = IEnumPins_Next(enum1, 1, pins, &count);
+    ok(hr == S_FALSE, "Got hr %#x.\n", hr);
+    ok(!count, "Got count %u.\n", count);
+
+    hr = IEnumPins_Reset(enum1);
+    ok(hr == S_OK, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Next(enum1, 2, pins, NULL);
+    ok(hr == E_INVALIDARG, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Next(enum1, 2, pins, &count);
+    ok(hr == S_FALSE, "Got hr %#x.\n", hr);
+    ok(count == 1, "Got count %u.\n", count);
+    IPin_Release(pins[0]);
+
+    hr = IEnumPins_Reset(enum1);
+    ok(hr == S_OK, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Clone(enum1, &enum2);
+    ok(hr == S_OK, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Skip(enum1, 2);
+    ok(hr == S_FALSE, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Skip(enum1, 1);
+    ok(hr == S_OK, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Skip(enum1, 1);
+    ok(hr == S_FALSE, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Next(enum1, 1, pins, NULL);
+    ok(hr == S_FALSE, "Got hr %#x.\n", hr);
+
+    hr = IEnumPins_Next(enum2, 1, pins, NULL);
+    ok(hr == S_OK, "Got hr %#x.\n", hr);
+    IPin_Release(pins[0]);
+
+    IEnumPins_Release(enum2);
+
+    if (SUCCEEDED(set_mixing_mode(filter)))
+    {
+        hr = IEnumPins_Next(enum1, 1, pins, NULL);
+        ok(hr == S_FALSE, "Got hr %#x.\n", hr);
+
+        hr = IEnumPins_Reset(enum1);
+        ok(hr == S_OK, "Got hr %#x.\n", hr);
+
+        hr = IEnumPins_Next(enum1, 1, pins, NULL);
+        ok(hr == S_OK, "Got hr %#x.\n", hr);
+        IPin_Release(pins[0]);
+
+        hr = IEnumPins_Next(enum1, 1, pins, NULL);
+        ok(hr == S_OK, "Got hr %#x.\n", hr);
+        IPin_Release(pins[0]);
+
+        hr = IEnumPins_Next(enum1, 1, pins, NULL);
+        ok(hr == S_FALSE, "Got hr %#x.\n", hr);
+
+        hr = IEnumPins_Reset(enum1);
+        ok(hr == S_OK, "Got hr %#x.\n", hr);
+
+        hr = IEnumPins_Next(enum1, 2, pins, &count);
+        ok(hr == S_OK, "Got hr %#x.\n", hr);
+        ok(count == 2, "Got count %u.\n", count);
+        IPin_Release(pins[0]);
+        IPin_Release(pins[1]);
+
+        hr = IEnumPins_Reset(enum1);
+        ok(hr == S_OK, "Got hr %#x.\n", hr);
+
+        hr = IEnumPins_Next(enum1, 3, pins, &count);
+        ok(hr == S_FALSE, "Got hr %#x.\n", hr);
+        ok(count == 2, "Got count %u.\n", count);
+        IPin_Release(pins[0]);
+        IPin_Release(pins[1]);
+    }
+
+    IEnumPins_Release(enum1);
+    ref = IBaseFilter_Release(filter);
+    ok(!ref, "Got outstanding refcount %d.\n", ref);
+}
+
 START_TEST(vmr9)
 {
     IBaseFilter *filter;
@@ -252,6 +403,7 @@ START_TEST(vmr9)
 
     test_filter_config();
     test_interfaces();
+    test_enum_pins();
 
     CoUninitialize();
 }
-- 
2.20.1