2 * Unit tests for Media Detector
4 * Copyright (C) 2008 Google (Lei Zhang, Dan Hipschman)
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.
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.
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
28 #include "wine/test.h"
33 /* Outer IUnknown for COM aggregation tests */
35 IUnknown IUnknown_iface
;
40 static inline struct unk_impl
*impl_from_IUnknown(IUnknown
*iface
)
42 return CONTAINING_RECORD(iface
, struct unk_impl
, IUnknown_iface
);
45 static HRESULT WINAPI
unk_QueryInterface(IUnknown
*iface
, REFIID riid
, void **ppv
)
47 struct unk_impl
*This
= impl_from_IUnknown(iface
);
49 return IUnknown_QueryInterface(This
->inner_unk
, riid
, ppv
);
52 static ULONG WINAPI
unk_AddRef(IUnknown
*iface
)
54 struct unk_impl
*This
= impl_from_IUnknown(iface
);
56 return InterlockedIncrement(&This
->ref
);
59 static ULONG WINAPI
unk_Release(IUnknown
*iface
)
61 struct unk_impl
*This
= impl_from_IUnknown(iface
);
63 return InterlockedDecrement(&This
->ref
);
66 static const IUnknownVtbl unk_vtbl
=
74 static WCHAR test_avi_filename
[MAX_PATH
];
75 static WCHAR test_sound_avi_filename
[MAX_PATH
];
77 static BOOL
unpack_avi_file(int id
, WCHAR name
[MAX_PATH
])
79 static WCHAR temp_path
[MAX_PATH
];
80 static const WCHAR prefix
[] = {'D','E','S',0};
81 static const WCHAR avi
[] = {'a','v','i',0};
89 res
= FindResourceW(NULL
, MAKEINTRESOURCEW(id
), MAKEINTRESOURCEW(AVI_RES_TYPE
));
93 data
= LoadResource(NULL
, res
);
97 mem
= LockResource(data
);
101 size
= SizeofResource(NULL
, res
);
105 if (!GetTempPathW(MAX_PATH
, temp_path
))
108 /* We might end up relying on the extension here, so .TMP is no good. */
109 if (!GetTempFileNameW(temp_path
, prefix
, 0, name
))
113 lstrcpyW(name
+ lstrlenW(name
) - 3, avi
);
115 fh
= CreateFileW(name
, GENERIC_WRITE
, 0, NULL
, CREATE_NEW
,
116 FILE_ATTRIBUTE_NORMAL
, NULL
);
117 if (fh
== INVALID_HANDLE_VALUE
)
120 ret
= WriteFile(fh
, mem
, size
, &written
, NULL
);
122 return ret
&& written
== size
;
125 static BOOL
init_tests(void)
127 return unpack_avi_file(TEST_AVI_RES
, test_avi_filename
)
128 && unpack_avi_file(TEST_SOUND_AVI_RES
, test_sound_avi_filename
);
131 static void test_mediadet(void)
134 struct unk_impl unk_obj
= {{&unk_vtbl
}, 19, NULL
};
135 IMediaDet
*pM
= NULL
;
137 BSTR filename
= NULL
;
145 /* COM aggregation */
146 hr
= CoCreateInstance(&CLSID_MediaDet
, &unk_obj
.IUnknown_iface
, CLSCTX_INPROC_SERVER
,
147 &IID_IUnknown
, (void**)&unk_obj
.inner_unk
);
148 ok(hr
== S_OK
, "CoCreateInstance failed: %08x\n", hr
);
150 hr
= IUnknown_QueryInterface(unk_obj
.inner_unk
, &IID_IMediaDet
, (void**)&pM
);
151 ok(hr
== S_OK
, "QueryInterface for IID_IMediaDet failed: %08x\n", hr
);
152 refcount
= IMediaDet_AddRef(pM
);
153 ok(refcount
== unk_obj
.ref
, "MediaDet just pretends to support COM aggregation\n");
154 refcount
= IMediaDet_Release(pM
);
155 ok(refcount
== unk_obj
.ref
, "MediaDet just pretends to support COM aggregation\n");
156 refcount
= IMediaDet_Release(pM
);
157 ok(refcount
== 19, "Refcount should be back at 19 but is %u\n", refcount
);
159 IUnknown_Release(unk_obj
.inner_unk
);
161 /* test.avi has one video stream. */
162 hr
= CoCreateInstance(&CLSID_MediaDet
, NULL
, CLSCTX_INPROC_SERVER
,
163 &IID_IMediaDet
, (LPVOID
*)&pM
);
164 ok(hr
== S_OK
, "CoCreateInstance failed with %x\n", hr
);
165 ok(pM
!= NULL
, "pM is NULL\n");
168 hr
= IMediaDet_get_Filename(pM
, &filename
);
169 /* Despite what MSDN claims, this returns S_OK. */
170 ok(hr
== S_OK
, "IMediaDet_get_Filename failed: %08x\n", hr
);
171 ok(filename
== NULL
, "IMediaDet_get_Filename\n");
173 filename
= (BSTR
) -1;
174 hr
= IMediaDet_get_Filename(pM
, &filename
);
175 /* Despite what MSDN claims, this returns S_OK. */
176 ok(hr
== S_OK
, "IMediaDet_get_Filename failed: %08x\n", hr
);
177 ok(filename
== NULL
, "IMediaDet_get_Filename\n");
180 hr
= IMediaDet_get_OutputStreams(pM
, &nstrms
);
181 ok(hr
== E_INVALIDARG
, "IMediaDet_get_OutputStreams failed: %08x\n", hr
);
182 ok(nstrms
== -1, "IMediaDet_get_OutputStreams: nstrms is %i\n", nstrms
);
185 /* The stream defaults to 0, even without a file! */
186 hr
= IMediaDet_get_CurrentStream(pM
, &strm
);
187 ok(hr
== S_OK
, "IMediaDet_get_CurrentStream failed: %08x\n", hr
);
188 ok(strm
== 0, "IMediaDet_get_CurrentStream: strm is %i\n", strm
);
190 hr
= IMediaDet_get_CurrentStream(pM
, NULL
);
191 ok(hr
== E_POINTER
, "IMediaDet_get_CurrentStream failed: %08x\n", hr
);
193 /* But put_CurrentStream doesn't. */
194 hr
= IMediaDet_put_CurrentStream(pM
, 0);
195 ok(hr
== E_INVALIDARG
, "IMediaDet_put_CurrentStream failed: %08x\n", hr
);
197 hr
= IMediaDet_put_CurrentStream(pM
, -1);
198 ok(hr
== E_INVALIDARG
, "IMediaDet_put_CurrentStream failed: %08x\n", hr
);
200 hr
= IMediaDet_get_StreamMediaType(pM
, &mt
);
201 ok(hr
== E_INVALIDARG
, "IMediaDet_get_StreamMediaType failed: %08x\n", hr
);
203 hr
= IMediaDet_get_StreamMediaType(pM
, NULL
);
204 ok(hr
== E_POINTER
, "IMediaDet_get_StreamMediaType failed: %08x\n", hr
);
206 filename
= SysAllocString(test_avi_filename
);
207 hr
= IMediaDet_put_Filename(pM
, filename
);
208 ok(hr
== S_OK
, "IMediaDet_put_Filename failed: %08x\n", hr
);
209 SysFreeString(filename
);
212 /* The stream defaults to 0. */
213 hr
= IMediaDet_get_CurrentStream(pM
, &strm
);
214 ok(hr
== S_OK
, "IMediaDet_get_CurrentStream failed: %08x\n", hr
);
215 ok(strm
== 0, "IMediaDet_get_CurrentStream: strm is %i\n", strm
);
217 ZeroMemory(&mt
, sizeof mt
);
218 hr
= IMediaDet_get_StreamMediaType(pM
, &mt
);
219 ok(hr
== S_OK
, "IMediaDet_get_StreamMediaType failed: %08x\n", hr
);
220 CoTaskMemFree(mt
.pbFormat
);
222 /* Even before get_OutputStreams. */
223 hr
= IMediaDet_put_CurrentStream(pM
, 1);
224 ok(hr
== E_INVALIDARG
, "IMediaDet_put_CurrentStream failed: %08x\n", hr
);
226 hr
= IMediaDet_get_OutputStreams(pM
, &nstrms
);
227 ok(hr
== S_OK
, "IMediaDet_get_OutputStreams failed: %08x\n", hr
);
228 ok(nstrms
== 1, "IMediaDet_get_OutputStreams: nstrms is %i\n", nstrms
);
231 hr
= IMediaDet_get_Filename(pM
, &filename
);
232 ok(hr
== S_OK
, "IMediaDet_get_Filename failed: %08x\n", hr
);
233 ok(lstrcmpW(filename
, test_avi_filename
) == 0,
234 "IMediaDet_get_Filename\n");
235 SysFreeString(filename
);
237 hr
= IMediaDet_get_Filename(pM
, NULL
);
238 ok(hr
== E_POINTER
, "IMediaDet_get_Filename failed: %08x\n", hr
);
241 hr
= IMediaDet_get_CurrentStream(pM
, &strm
);
242 ok(hr
== S_OK
, "IMediaDet_get_CurrentStream failed: %08x\n", hr
);
243 ok(strm
== 0, "IMediaDet_get_CurrentStream: strm is %i\n", strm
);
245 hr
= IMediaDet_get_CurrentStream(pM
, NULL
);
246 ok(hr
== E_POINTER
, "IMediaDet_get_CurrentStream failed: %08x\n", hr
);
248 hr
= IMediaDet_put_CurrentStream(pM
, -1);
249 ok(hr
== E_INVALIDARG
, "IMediaDet_put_CurrentStream failed: %08x\n", hr
);
251 hr
= IMediaDet_put_CurrentStream(pM
, 1);
252 ok(hr
== E_INVALIDARG
, "IMediaDet_put_CurrentStream failed: %08x\n", hr
);
256 hr
= IMediaDet_get_CurrentStream(pM
, &strm
);
257 ok(hr
== S_OK
, "IMediaDet_get_CurrentStream failed: %08x\n", hr
);
258 ok(strm
== 0, "IMediaDet_get_CurrentStream: strm is %i\n", strm
);
260 hr
= IMediaDet_put_CurrentStream(pM
, 0);
261 ok(hr
== S_OK
, "IMediaDet_put_CurrentStream failed: %08x\n", hr
);
264 hr
= IMediaDet_get_CurrentStream(pM
, &strm
);
265 ok(hr
== S_OK
, "IMediaDet_get_CurrentStream failed: %08x\n", hr
);
266 ok(strm
== 0, "IMediaDet_get_CurrentStream: strm is %i\n", strm
);
268 ZeroMemory(&mt
, sizeof mt
);
269 hr
= IMediaDet_get_StreamMediaType(pM
, &mt
);
270 ok(hr
== S_OK
, "IMediaDet_get_StreamMediaType failed: %08x\n", hr
);
271 ok(IsEqualGUID(&mt
.majortype
, &MEDIATYPE_Video
),
272 "IMediaDet_get_StreamMediaType\n");
273 CoTaskMemFree(mt
.pbFormat
);
275 hr
= IMediaDet_get_FrameRate(pM
, NULL
);
276 ok(hr
== E_POINTER
, "IMediaDet_get_FrameRate failed: %08x\n", hr
);
278 hr
= IMediaDet_get_FrameRate(pM
, &fps
);
279 ok(hr
== S_OK
, "IMediaDet_get_FrameRate failed: %08x\n", hr
);
280 ok(fps
== 10.0, "IMediaDet_get_FrameRate: fps is %f\n", fps
);
282 hr
= IMediaDet_Release(pM
);
283 ok(hr
== 0, "IMediaDet_Release returned: %x\n", hr
);
285 DeleteFileW(test_avi_filename
);
287 /* test_sound.avi has one video stream and one audio stream. */
288 hr
= CoCreateInstance(&CLSID_MediaDet
, NULL
, CLSCTX_INPROC_SERVER
,
289 &IID_IMediaDet
, (LPVOID
*)&pM
);
290 ok(hr
== S_OK
, "CoCreateInstance failed with %x\n", hr
);
291 ok(pM
!= NULL
, "pM is NULL\n");
293 filename
= SysAllocString(test_sound_avi_filename
);
294 hr
= IMediaDet_put_Filename(pM
, filename
);
295 ok(hr
== S_OK
, "IMediaDet_put_Filename failed: %08x\n", hr
);
296 SysFreeString(filename
);
298 hr
= IMediaDet_get_OutputStreams(pM
, &nstrms
);
299 ok(hr
== S_OK
, "IMediaDet_get_OutputStreams failed: %08x\n", hr
);
300 ok(nstrms
== 2, "IMediaDet_get_OutputStreams: nstrms is %i\n", nstrms
);
303 hr
= IMediaDet_get_Filename(pM
, &filename
);
304 ok(hr
== S_OK
, "IMediaDet_get_Filename failed: %08x\n", hr
);
305 ok(lstrcmpW(filename
, test_sound_avi_filename
) == 0,
306 "IMediaDet_get_Filename\n");
307 SysFreeString(filename
);
309 /* I don't know if the stream order is deterministic. Just check
310 for both an audio and video stream. */
313 for (i
= 0; i
< 2; ++i
)
315 hr
= IMediaDet_put_CurrentStream(pM
, i
);
316 ok(hr
== S_OK
, "IMediaDet_put_CurrentStream failed: %08x\n", hr
);
319 hr
= IMediaDet_get_CurrentStream(pM
, &strm
);
320 ok(hr
== S_OK
, "IMediaDet_get_CurrentStream failed: %08x\n", hr
);
321 ok(strm
== i
, "IMediaDet_get_CurrentStream: strm is %i\n", strm
);
323 ZeroMemory(&mt
, sizeof mt
);
324 hr
= IMediaDet_get_StreamMediaType(pM
, &mt
);
325 ok(hr
== S_OK
, "IMediaDet_get_StreamMediaType failed: %08x\n", hr
);
326 flags
+= (IsEqualGUID(&mt
.majortype
, &MEDIATYPE_Video
)
328 : (IsEqualGUID(&mt
.majortype
, &MEDIATYPE_Audio
)
332 if (IsEqualGUID(&mt
.majortype
, &MEDIATYPE_Audio
))
334 hr
= IMediaDet_get_FrameRate(pM
, &fps
);
335 ok(hr
== VFW_E_INVALIDMEDIATYPE
, "IMediaDet_get_FrameRate failed: %08x\n", hr
);
338 CoTaskMemFree(mt
.pbFormat
);
340 ok(flags
== 3, "IMediaDet_get_StreamMediaType: flags are %i\n", flags
);
342 hr
= IMediaDet_put_CurrentStream(pM
, 2);
343 ok(hr
== E_INVALIDARG
, "IMediaDet_put_CurrentStream failed: %08x\n", hr
);
346 hr
= IMediaDet_get_CurrentStream(pM
, &strm
);
347 ok(hr
== S_OK
, "IMediaDet_get_CurrentStream failed: %08x\n", hr
);
348 ok(strm
== 1, "IMediaDet_get_CurrentStream: strm is %i\n", strm
);
350 hr
= IMediaDet_Release(pM
);
351 ok(hr
== 0, "IMediaDet_Release returned: %x\n", hr
);
353 DeleteFileW(test_sound_avi_filename
);
356 static void test_samplegrabber(void)
358 struct unk_impl unk_obj
= {{&unk_vtbl
}, 19, NULL
};
367 /* COM aggregation */
368 hr
= CoCreateInstance(&CLSID_SampleGrabber
, &unk_obj
.IUnknown_iface
, CLSCTX_INPROC_SERVER
,
369 &IID_IUnknown
, (void**)&unk_obj
.inner_unk
);
370 ok(hr
== S_OK
, "CoCreateInstance failed: %08x\n", hr
);
372 hr
= IUnknown_QueryInterface(unk_obj
.inner_unk
, &IID_ISampleGrabber
, (void**)&sg
);
373 ok(hr
== S_OK
, "QueryInterface for IID_ISampleGrabber failed: %08x\n", hr
);
374 refcount
= ISampleGrabber_AddRef(sg
);
375 ok(refcount
== unk_obj
.ref
, "SampleGrabber just pretends to support COM aggregation\n");
376 refcount
= ISampleGrabber_Release(sg
);
377 ok(refcount
== unk_obj
.ref
, "SampleGrabber just pretends to support COM aggregation\n");
378 refcount
= ISampleGrabber_Release(sg
);
379 ok(refcount
== 19, "Refcount should be back at 19 but is %u\n", refcount
);
380 IUnknown_Release(unk_obj
.inner_unk
);
383 hr
= CoCreateInstance(&CLSID_SampleGrabber
, NULL
, CLSCTX_INPROC_SERVER
, &IID_IClassFactory
,
385 ok(hr
== E_NOINTERFACE
, "SampleGrabber create failed: %08x, expected E_NOINTERFACE\n", hr
);
387 /* Same refcount for all SampleGrabber interfaces */
388 hr
= CoCreateInstance(&CLSID_SampleGrabber
, NULL
, CLSCTX_INPROC_SERVER
, &IID_ISampleGrabber
,
390 ok(hr
== S_OK
, "SampleGrabber create failed: %08x, expected S_OK\n", hr
);
391 refcount
= ISampleGrabber_AddRef(sg
);
392 ok(refcount
== 2, "refcount == %u, expected 2\n", refcount
);
394 hr
= ISampleGrabber_QueryInterface(sg
, &IID_IBaseFilter
, (void**)&bf
);
395 ok(hr
== S_OK
, "QueryInterface for IID_IBaseFilter failed: %08x\n", hr
);
396 refcount
= IBaseFilter_AddRef(bf
);
397 ok(refcount
== 4, "refcount == %u, expected 4\n", refcount
);
398 refcount
= IBaseFilter_Release(bf
);
400 hr
= ISampleGrabber_QueryInterface(sg
, &IID_IMediaFilter
, (void**)&mf
);
401 ok(hr
== S_OK
, "QueryInterface for IID_IMediaFilter failed: %08x\n", hr
);
402 refcount
= IMediaFilter_AddRef(mf
);
403 ok(refcount
== 5, "refcount == %u, expected 5\n", refcount
);
404 refcount
= IMediaFilter_Release(mf
);
406 hr
= ISampleGrabber_QueryInterface(sg
, &IID_IPersist
, (void**)&persist
);
407 ok(hr
== S_OK
, "QueryInterface for IID_IPersist failed: %08x\n", hr
);
408 refcount
= IPersist_AddRef(persist
);
409 ok(refcount
== 6, "refcount == %u, expected 6\n", refcount
);
410 refcount
= IPersist_Release(persist
);
412 hr
= ISampleGrabber_QueryInterface(sg
, &IID_IUnknown
, (void**)&unk
);
413 ok(hr
== S_OK
, "QueryInterface for IID_IUnknown failed: %08x\n", hr
);
414 refcount
= IUnknown_AddRef(unk
);
415 ok(refcount
== 7, "refcount == %u, expected 7\n", refcount
);
416 refcount
= IUnknown_Release(unk
);
418 /* Interfaces that native does not support */
419 hr
= ISampleGrabber_QueryInterface(sg
, &IID_IMediaPosition
, (void**)&unk
);
420 todo_wine
ok(hr
== E_NOINTERFACE
, "QueryInterface for IID_IMediaPosition failed: %08x\n", hr
);
421 hr
= ISampleGrabber_QueryInterface(sg
, &IID_IMediaSeeking
, (void**)&unk
);
422 todo_wine
ok(hr
== E_NOINTERFACE
, "QueryInterface for IID_IMediaSeeking failed: %08x\n", hr
);
423 hr
= ISampleGrabber_QueryInterface(sg
, &IID_IMemInputPin
, (void**)&unk
);
424 ok(hr
== E_NOINTERFACE
, "QueryInterface for IID_IMemInputPin failed: %08x\n", hr
);
425 hr
= ISampleGrabber_QueryInterface(sg
, &IID_IQualityControl
, (void**)&unk
);
426 ok(hr
== E_NOINTERFACE
, "QueryInterface for IID_IQualityControl failed: %08x\n", hr
);
427 hr
= ISampleGrabber_QueryInterface(sg
, &IID_ISeekingPassThru
, (void**)&unk
);
428 ok(hr
== E_NOINTERFACE
, "QueryInterface for IID_ISeekingPassThru failed: %08x\n", hr
);
430 while (ISampleGrabber_Release(sg
));
433 static void test_COM_sg_enumpins(void)
436 IEnumPins
*pins
, *pins2
;
441 hr
= CoCreateInstance(&CLSID_SampleGrabber
, NULL
, CLSCTX_INPROC_SERVER
, &IID_IBaseFilter
,
443 ok(hr
== S_OK
, "SampleGrabber create failed: %08x, expected S_OK\n", hr
);
444 hr
= IBaseFilter_EnumPins(bf
, &pins
);
445 ok(hr
== S_OK
, "EnumPins create failed: %08x, expected S_OK\n", hr
);
447 /* Same refcount for all EnumPins interfaces */
448 refcount
= IEnumPins_AddRef(pins
);
449 ok(refcount
== 2, "refcount == %u, expected 2\n", refcount
);
450 hr
= IEnumPins_QueryInterface(pins
, &IID_IEnumPins
, (void**)&pins2
);
451 ok(hr
== S_OK
, "QueryInterface for IID_IEnumPins failed: %08x\n", hr
);
452 ok(pins
== pins2
, "QueryInterface for self failed (%p != %p)\n", pins
, pins2
);
453 IEnumPins_Release(pins2
);
455 hr
= IEnumPins_QueryInterface(pins
, &IID_IUnknown
, (void**)&unk
);
456 ok(hr
== S_OK
, "QueryInterface for IID_IUnknown failed: %08x\n", hr
);
457 refcount
= IUnknown_AddRef(unk
);
458 ok(refcount
== 4, "refcount == %u, expected 4\n", refcount
);
459 refcount
= IUnknown_Release(unk
);
461 while (IEnumPins_Release(pins
));
462 IBaseFilter_Release(bf
);
469 skip("Couldn't initialize tests!\n");
475 test_samplegrabber();
476 test_COM_sg_enumpins();