git.oblomov.eu Git - wine/blob - dlls/quartz/tests/misc.c

   1 /*
   2  * Misc unit tests for Quartz
   3  *
   4  * Copyright (C) 2007 Google (Lei Zhang)
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #define COBJMACROS
  22
  23 #include "wine/test.h"
  24 #include "dshow.h"
  25
  26 #define QI_SUCCEED(iface, riid, ppv) hr = IUnknown_QueryInterface(iface, &riid, (LPVOID*)&ppv); \
  27     ok(hr == S_OK, "IUnknown_QueryInterface returned %x\n", hr); \
  28     ok(ppv != NULL, "Pointer is NULL\n");
  29
  30 #define QI_FAIL(iface, riid, ppv) hr = IUnknown_QueryInterface(iface, &riid, (LPVOID*)&ppv); \
  31     ok(hr == E_NOINTERFACE, "IUnknown_QueryInterface returned %x\n", hr); \
  32     ok(ppv == NULL, "Pointer is %p\n", ppv);
  33
  34 #define ADDREF_EXPECT(iface, num) if (iface) { \
  35     refCount = IUnknown_AddRef(iface); \
  36     ok(refCount == num, "IUnknown_AddRef should return %d, got %d\n", num, refCount); \
  37 }
  38
  39 #define RELEASE_EXPECT(iface, num) if (iface) { \
  40     refCount = IUnknown_Release(iface); \
  41     ok(refCount == num, "IUnknown_Release should return %d, got %d\n", num, refCount); \
  42 }
  43
  44 static void test_aggregation(const CLSID clsidOuter, const CLSID clsidInner,
  45                              const IID iidOuter, const IID iidInner)
  46 {
  47     HRESULT hr;
  48     ULONG refCount;
  49     IUnknown *pUnkOuter = NULL;
  50     IUnknown *pUnkInner = NULL;
  51     IUnknown *pUnkInnerFail = NULL;
  52     IUnknown *pUnkOuterTest = NULL;
  53     IUnknown *pUnkInnerTest = NULL;
  54     IUnknown *pUnkAggregatee = NULL;
  55     IUnknown *pUnkAggregator = NULL;
  56     IUnknown *pUnkTest = NULL;
  57
  58     hr = CoCreateInstance(&clsidOuter, NULL, CLSCTX_INPROC_SERVER,
  59                           &IID_IUnknown, (LPVOID*)&pUnkOuter);
  60     ok(hr == S_OK, "CoCreateInstance failed with %x\n", hr);
  61     ok(pUnkOuter != NULL, "pUnkOuter is NULL\n");
  62
  63     if (!pUnkOuter)
  64     {
  65         skip("pUnkOuter is NULL\n");
  66         return;
  67     }
  68
  69     /* for aggregation, we should only be able to request IUnknown */
  70     hr = CoCreateInstance(&clsidInner, pUnkOuter, CLSCTX_INPROC_SERVER,
  71                           &iidInner, (LPVOID*)&pUnkInnerFail);
  72     ok(hr == E_NOINTERFACE, "CoCreateInstance returned %x\n", hr);
  73     ok(pUnkInnerFail == NULL, "pUnkInnerFail is not NULL\n");
  74
  75     /* aggregation, request IUnknown */
  76     hr = CoCreateInstance(&clsidInner, pUnkOuter, CLSCTX_INPROC_SERVER,
  77                           &IID_IUnknown, (LPVOID*)&pUnkInner);
  78     ok(hr == S_OK, "CoCreateInstance returned %x\n", hr);
  79     ok(pUnkInner != NULL, "pUnkInner is NULL\n");
  80
  81     if (!pUnkInner)
  82     {
  83         skip("pUnkInner is NULL\n");
  84         return;
  85     }
  86
  87     ADDREF_EXPECT(pUnkOuter, 2);
  88     ADDREF_EXPECT(pUnkInner, 2);
  89     RELEASE_EXPECT(pUnkOuter, 1);
  90     RELEASE_EXPECT(pUnkInner, 1);
  91
  92     QI_FAIL(pUnkOuter, iidInner, pUnkAggregatee);
  93     QI_FAIL(pUnkInner, iidOuter, pUnkAggregator);
  94
  95     /* these QueryInterface calls should work */
  96     QI_SUCCEED(pUnkOuter, iidOuter, pUnkAggregator);
  97     QI_SUCCEED(pUnkOuter, IID_IUnknown, pUnkOuterTest);
  98     QI_SUCCEED(pUnkInner, iidInner, pUnkAggregatee);
  99     QI_SUCCEED(pUnkInner, IID_IUnknown, pUnkInnerTest);
 100
 101     if (!pUnkAggregator || !pUnkOuterTest || !pUnkAggregatee
 102                     || !pUnkInnerTest)
 103     {
 104         skip("One of the required interfaces is NULL\n");
 105         return;
 106     }
 107
 108     ADDREF_EXPECT(pUnkAggregator, 5);
 109     ADDREF_EXPECT(pUnkOuterTest, 6);
 110     ADDREF_EXPECT(pUnkAggregatee, 7);
 111     ADDREF_EXPECT(pUnkInnerTest, 3);
 112     RELEASE_EXPECT(pUnkAggregator, 6);
 113     RELEASE_EXPECT(pUnkOuterTest, 5);
 114     RELEASE_EXPECT(pUnkAggregatee, 4);
 115     RELEASE_EXPECT(pUnkInnerTest, 2);
 116
 117     QI_SUCCEED(pUnkAggregator, IID_IUnknown, pUnkTest);
 118     QI_SUCCEED(pUnkOuterTest, IID_IUnknown, pUnkTest);
 119     QI_SUCCEED(pUnkAggregatee, IID_IUnknown, pUnkTest);
 120     QI_SUCCEED(pUnkInnerTest, IID_IUnknown, pUnkTest);
 121
 122     QI_FAIL(pUnkAggregator, iidInner, pUnkTest);
 123     QI_FAIL(pUnkOuterTest, iidInner, pUnkTest);
 124     QI_FAIL(pUnkAggregatee, iidInner, pUnkTest);
 125     QI_SUCCEED(pUnkInnerTest, iidInner, pUnkTest);
 126
 127     QI_SUCCEED(pUnkAggregator, iidOuter, pUnkTest);
 128     QI_SUCCEED(pUnkOuterTest, iidOuter, pUnkTest);
 129     QI_SUCCEED(pUnkAggregatee, iidOuter, pUnkTest);
 130     QI_FAIL(pUnkInnerTest, iidOuter, pUnkTest);
 131
 132     RELEASE_EXPECT(pUnkAggregator, 10);
 133     RELEASE_EXPECT(pUnkOuterTest, 9);
 134     RELEASE_EXPECT(pUnkAggregatee, 8);
 135     RELEASE_EXPECT(pUnkInnerTest, 2);
 136     RELEASE_EXPECT(pUnkOuter, 7);
 137     RELEASE_EXPECT(pUnkInner, 1);
 138
 139     do
 140     {
 141         refCount = IUnknown_Release(pUnkInner);
 142     } while (refCount);
 143
 144     do
 145     {
 146         refCount = IUnknown_Release(pUnkOuter);
 147     } while (refCount);
 148 }
 149
 150 static void test_video_renderer_aggregations(void)
 151 {
 152     const IID * iids[] = {
 153         &IID_IMediaFilter, &IID_IBaseFilter, &IID_IBasicVideo, &IID_IVideoWindow
 154     };
 155     int i;
 156
 157     for (i = 0; i < sizeof(iids) / sizeof(iids[0]); i++)
 158     {
 159         test_aggregation(CLSID_SystemClock, CLSID_VideoRenderer,
 160                          IID_IReferenceClock, *iids[i]);
 161     }
 162 }
 163
 164 static void test_filter_graph_aggregations(void)
 165 {
 166     const IID * iids[] = {
 167         &IID_IFilterGraph2, &IID_IMediaControl, &IID_IGraphBuilder,
 168         &IID_IFilterGraph, &IID_IMediaSeeking, &IID_IBasicAudio, &IID_IBasicVideo,
 169         &IID_IVideoWindow, &IID_IMediaEventEx, &IID_IMediaFilter,
 170         &IID_IMediaEventSink, &IID_IGraphConfig, &IID_IMediaPosition
 171     };
 172     int i;
 173
 174     for (i = 0; i < sizeof(iids) / sizeof(iids[0]); i++)
 175     {
 176         test_aggregation(CLSID_SystemClock, CLSID_FilterGraph,
 177                          IID_IReferenceClock, *iids[i]);
 178     }
 179 }
 180
 181 static void test_filter_mapper_aggregations(void)
 182 {
 183     const IID * iids[] = {
 184         &IID_IFilterMapper2, &IID_IFilterMapper
 185     };
 186     int i;
 187
 188     for (i = 0; i < sizeof(iids) / sizeof(iids[0]); i++)
 189     {
 190         test_aggregation(CLSID_SystemClock, CLSID_FilterMapper2,
 191                          IID_IReferenceClock, *iids[i]);
 192     }
 193 }
 194
 195 START_TEST(misc)
 196 {
 197     CoInitialize(NULL);
 198
 199     test_video_renderer_aggregations();
 200     test_filter_graph_aggregations();
 201     test_filter_mapper_aggregations();
 202
 203     CoUninitialize();
 204 }