quartz/tests: Add test for EnumMatchingFilters.

dlls/quartz/tests/Makefile.in

@@ -3,10 +3,11 @@ TOPOBJDIR = ../../..
 SRCDIR    = @srcdir@
 VPATH     = @srcdir@
 TESTDLL   = quartz.dll
-IMPORTS   = strmiids uuid ole32 kernel32
+IMPORTS   = strmiids uuid oleaut32 ole32 kernel32
 
 CTESTS = \
 	filtergraph.c \
+	filtermapper.c \
 	memallocator.c \
 	misc.c \
 	referenceclock.c \

dlls/quartz/tests/filtermapper.c

+/*
+ * Filtermapper unit tests for Quartz
+ *
+ * Copyright (C) 2008 Alexander Dorofeyev
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#define COBJMACROS
+
+#include "wine/test.h"
+#include "winbase.h"
+#include "dshow.h"
+
+/* Helper function, checks if filter with given name was enumerated. */
+static BOOL enum_find_filter(const WCHAR *wszFilterName, IEnumMoniker *pEnum)
+{
+    IMoniker *pMoniker = NULL;
+    BOOL found = FALSE;
+    ULONG nb;
+    HRESULT hr;
+    static const WCHAR wszFriendlyName[] = {'F','r','i','e','n','d','l','y','N','a','m','e',0};
+
+    while(!found && IEnumMoniker_Next(pEnum, 1, &pMoniker, &nb) == S_OK)
+    {
+        IPropertyBag * pPropBagCat = NULL;
+        VARIANT var;
+
+        VariantInit(&var);
+
+        hr = IMoniker_BindToStorage(pMoniker, NULL, NULL, &IID_IPropertyBag, (LPVOID*)&pPropBagCat);
+        ok(SUCCEEDED(hr), "IMoniker_BindToStorage failed with %x\n", hr);
+        if (FAILED(hr) || !pPropBagCat)
+        {
+            VariantClear(&var);
+            IMoniker_Release(pMoniker);
+            continue;
+        }
+
+        hr = IPropertyBag_Read(pPropBagCat, wszFriendlyName, &var, NULL);
+        ok(SUCCEEDED(hr), "IPropertyBag_Read failed with %x\n", hr);
+
+        if (SUCCEEDED(hr))
+        {
+            if (!lstrcmpW((WCHAR*)V_UNION(&var, bstrVal), wszFilterName)) found = TRUE;
+        }
+
+        IPropertyBag_Release(pPropBagCat);
+        IMoniker_Release(pMoniker);
+        VariantClear(&var);
+    }
+
+    return found;
+}
+
+static void test_fm2_enummatchingfilters(void)
+{
+    IFilterMapper2 *pMapper = NULL;
+    HRESULT hr;
+    REGFILTER2 rgf2;
+    REGFILTERPINS2 rgPins2[2];
+    REGPINTYPES rgPinType;
+    static const WCHAR wszFilterName1[] = {'T', 'e', 's', 't', 'f', 'i', 'l', 't', 'e', 'r', '1', 0 };
+    static const WCHAR wszFilterName2[] = {'T', 'e', 's', 't', 'f', 'i', 'l', 't', 'e', 'r', '2', 0 };
+    CLSID clsidFilter1;
+    CLSID clsidFilter2;
+    IEnumMoniker *pEnum = NULL;
+    BOOL found;
+
+    ZeroMemory(&rgf2, sizeof(rgf2));
+
+    hr = CoCreateInstance(&CLSID_FilterMapper2, NULL, CLSCTX_INPROC_SERVER,
+            &IID_IFilterMapper2, (LPVOID*)&pMapper);
+    ok(hr == S_OK, "CoCreateInstance failed with %x\n", hr);
+    if (FAILED(hr)) goto out;
+
+    hr = CoCreateGuid(&clsidFilter1);
+    ok(hr == S_OK, "CoCreateGuid failed with %x\n", hr);
+    hr = CoCreateGuid(&clsidFilter2);
+    ok(hr == S_OK, "CoCreateGuid failed with %x\n", hr);
+
+    /* Test that a test renderer filter is returned when enumerating filters with bRender=FALSE */
+    rgf2.dwVersion = 2;
+    rgf2.dwMerit = MERIT_UNLIKELY;
+    S1(U(rgf2)).cPins2 = 1;
+    S1(U(rgf2)).rgPins2 = rgPins2;
+
+    rgPins2[0].dwFlags = REG_PINFLAG_B_RENDERER;
+    rgPins2[0].cInstances = 1;
+    rgPins2[0].nMediaTypes = 1;
+    rgPins2[0].lpMediaType = &rgPinType;
+    rgPins2[0].nMediums = 0;
+    rgPins2[0].lpMedium = NULL;
+    rgPins2[0].clsPinCategory = NULL;
+
+    rgPinType.clsMajorType = &GUID_NULL;
+    rgPinType.clsMinorType = &GUID_NULL;
+
+    hr = IFilterMapper2_RegisterFilter(pMapper, &clsidFilter1, wszFilterName1, NULL,
+                    &CLSID_LegacyAmFilterCategory, NULL, &rgf2);
+    ok(hr == S_OK, "IFilterMapper2_RegisterFilter failed with %x\n", hr);
+
+    rgPins2[0].dwFlags = 0;
+
+    rgPins2[1].dwFlags = REG_PINFLAG_B_OUTPUT;
+    rgPins2[1].cInstances = 1;
+    rgPins2[1].nMediaTypes = 1;
+    rgPins2[1].lpMediaType = &rgPinType;
+    rgPins2[1].nMediums = 0;
+    rgPins2[1].lpMedium = NULL;
+    rgPins2[1].clsPinCategory = NULL;
+
+    S1(U(rgf2)).cPins2 = 2;
+
+    hr = IFilterMapper2_RegisterFilter(pMapper, &clsidFilter2, wszFilterName2, NULL,
+                    &CLSID_LegacyAmFilterCategory, NULL, &rgf2);
+    ok(hr == S_OK, "IFilterMapper2_RegisterFilter failed with %x\n", hr);
+
+    hr = IFilterMapper2_EnumMatchingFilters(pMapper, &pEnum, 0, TRUE, MERIT_UNLIKELY, TRUE,
+                0, NULL, NULL, &GUID_NULL, FALSE, FALSE, 0, NULL, NULL, &GUID_NULL);
+    ok(hr == S_OK, "IFilterMapper2_EnumMatchingFilters failed with %x\n", hr);
+    if (SUCCEEDED(hr) && pEnum)
+    {
+        found = enum_find_filter(wszFilterName1, pEnum);
+        ok(found, "EnumMatchingFilters failed to return the test filter 1\n");
+        found = enum_find_filter(wszFilterName2, pEnum);
+        ok(found, "EnumMatchingFilters failed to return the test filter 2\n");
+    }
+
+    if (pEnum) IEnumMoniker_Release(pEnum);
+    pEnum = NULL;
+
+    /* Non renderer must not be returned with bRender=TRUE */
+
+    hr = IFilterMapper2_EnumMatchingFilters(pMapper, &pEnum, 0, TRUE, MERIT_UNLIKELY, TRUE,
+                0, NULL, NULL, &GUID_NULL, TRUE, FALSE, 0, NULL, NULL, &GUID_NULL);
+    ok(hr == S_OK, "IFilterMapper2_EnumMatchingFilters failed with %x\n", hr);
+
+    if (SUCCEEDED(hr) && pEnum)
+    {
+        found = enum_find_filter(wszFilterName1, pEnum);
+        ok(found, "EnumMatchingFilters failed to return the test filter 1\n");
+        found = enum_find_filter(wszFilterName2, pEnum);
+        ok(!found, "EnumMatchingFilters should not return the test filter 2\n");
+    }
+
+    hr = IFilterMapper2_UnregisterFilter(pMapper, &CLSID_LegacyAmFilterCategory, NULL,
+            &clsidFilter1);
+    ok(SUCCEEDED(hr), "IFilterMapper2_UnregisterFilter failed with %x\n", hr);
+
+    hr = IFilterMapper2_UnregisterFilter(pMapper, &CLSID_LegacyAmFilterCategory, NULL,
+            &clsidFilter2);
+    ok(SUCCEEDED(hr), "IFilterMapper2_UnregisterFilter failed with %x\n", hr);
+
+    out:
+
+    if (pEnum) IEnumMoniker_Release(pEnum);
+    if (pMapper) IFilterMapper2_Release(pMapper);
+}
+
+START_TEST(filtermapper)
+{
+    CoInitialize(NULL);
+
+    test_fm2_enummatchingfilters();
+
+    CoUninitialize();
+}