dlls/amstream/tests/amstream.c

   1 /*
   2  * Unit tests for MultiMedia Stream functions
   3  *
   4  * Copyright (C) 2009, 2012 Christian Costa
   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 "initguid.h"
  25 #include "amstream.h"
  26 #include "vfwmsgs.h"
  27
  28 static const WCHAR filenameW[] = {'t','e','s','t','.','a','v','i',0};
  29
  30 static IAMMultiMediaStream* pams;
  31 static IDirectDraw7* pdd7;
  32 static IDirectDrawSurface7* pdds7;
  33
  34 static int create_ammultimediastream(void)
  35 {
  36     return S_OK == CoCreateInstance(
  37         &CLSID_AMMultiMediaStream, NULL, CLSCTX_INPROC_SERVER, &IID_IAMMultiMediaStream, (LPVOID*)&pams);
  38 }
  39
  40 static void release_ammultimediastream(void)
  41 {
  42     IAMMultiMediaStream_Release(pams);
  43 }
  44
  45 static int create_directdraw(void)
  46 {
  47     HRESULT hr;
  48     IDirectDraw* pdd = NULL;
  49     DDSURFACEDESC2 ddsd;
  50
  51     hr = DirectDrawCreate(NULL, &pdd, NULL);
  52     ok(hr==DD_OK, "DirectDrawCreate returned: %x\n", hr);
  53     if (hr != DD_OK)
  54        goto error;
  55
  56     hr = IDirectDraw_QueryInterface(pdd, &IID_IDirectDraw7, (LPVOID*)&pdd7);
  57     ok(hr==DD_OK, "QueryInterface returned: %x\n", hr);
  58     if (hr != DD_OK) goto error;
  59
  60     hr = IDirectDraw7_SetCooperativeLevel(pdd7, GetDesktopWindow(), DDSCL_NORMAL);
  61     ok(hr==DD_OK, "SetCooperativeLevel returned: %x\n", hr);
  62
  63     ZeroMemory(&ddsd, sizeof(ddsd));
  64     ddsd.dwSize = sizeof(ddsd);
  65     ddsd.dwFlags = DDSD_CAPS;
  66     ddsd.ddsCaps.dwCaps = DDSCAPS_PRIMARYSURFACE;
  67     hr = IDirectDraw7_CreateSurface(pdd7, &ddsd, &pdds7, NULL);
  68     ok(hr==DD_OK, "CreateSurface returned: %x\n", hr);
  69
  70     return TRUE;
  71
  72 error:
  73     if (pdds7)
  74         IDirectDrawSurface7_Release(pdds7);
  75     if (pdd7)
  76         IDirectDraw7_Release(pdd7);
  77     if (pdd)
  78         IDirectDraw_Release(pdd);
  79
  80     return FALSE;
  81 }
  82
  83 static void release_directdraw(void)
  84 {
  85     IDirectDrawSurface7_Release(pdds7);
  86     IDirectDraw7_Release(pdd7);
  87 }
  88
  89 static void test_openfile(void)
  90 {
  91     HRESULT hr;
  92     IGraphBuilder* pgraph;
  93
  94     if (!create_ammultimediastream())
  95         return;
  96
  97     hr = IAMMultiMediaStream_GetFilterGraph(pams, &pgraph);
  98     ok(hr==S_OK, "IAMMultiMediaStream_GetFilterGraph returned: %x\n", hr);
  99     ok(pgraph==NULL, "Filtergraph should not be created yet\n");
 100
 101     if (pgraph)
 102         IGraphBuilder_Release(pgraph);
 103
 104     hr = IAMMultiMediaStream_OpenFile(pams, filenameW, 0);
 105     ok(hr==S_OK, "IAMMultiMediaStream_OpenFile returned: %x\n", hr);
 106
 107     hr = IAMMultiMediaStream_GetFilterGraph(pams, &pgraph);
 108     ok(hr==S_OK, "IAMMultiMediaStream_GetFilterGraph returned: %x\n", hr);
 109     ok(pgraph!=NULL, "Filtergraph should be created\n");
 110
 111     if (pgraph)
 112         IGraphBuilder_Release(pgraph);
 113
 114     release_ammultimediastream();
 115 }
 116
 117 static void test_renderfile(void)
 118 {
 119     HRESULT hr;
 120     IMediaStream *pvidstream = NULL;
 121     IDirectDrawMediaStream *pddstream = NULL;
 122     IDirectDrawStreamSample *pddsample = NULL;
 123
 124     if (!create_ammultimediastream())
 125         return;
 126     if (!create_directdraw())
 127     {
 128         release_ammultimediastream();
 129         return;
 130     }
 131
 132     hr = IAMMultiMediaStream_Initialize(pams, STREAMTYPE_READ, 0, NULL);
 133     ok(hr==S_OK, "IAMMultiMediaStream_Initialize returned: %x\n", hr);
 134
 135     hr = IAMMultiMediaStream_AddMediaStream(pams, (IUnknown*)pdd7, &MSPID_PrimaryVideo, 0, NULL);
 136     ok(hr==S_OK, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 137
 138     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &MSPID_PrimaryAudio, AMMSF_ADDDEFAULTRENDERER, NULL);
 139     ok(hr==S_OK, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 140
 141     hr = IAMMultiMediaStream_OpenFile(pams, filenameW, 0);
 142     ok(hr==S_OK, "IAMMultiMediaStream_OpenFile returned: %x\n", hr);
 143
 144     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryVideo, &pvidstream);
 145     ok(hr==S_OK, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 146     if (FAILED(hr)) goto error;
 147
 148     hr = IMediaStream_QueryInterface(pvidstream, &IID_IDirectDrawMediaStream, (LPVOID*)&pddstream);
 149     ok(hr==S_OK, "IMediaStream_QueryInterface returned: %x\n", hr);
 150     if (FAILED(hr)) goto error;
 151
 152     hr = IDirectDrawMediaStream_CreateSample(pddstream, NULL, NULL, 0, &pddsample);
 153     todo_wine ok(hr==S_OK, "IDirectDrawMediaStream_CreateSample returned: %x\n", hr);
 154
 155 error:
 156     if (pddsample)
 157         IDirectDrawMediaSample_Release(pddsample);
 158     if (pddstream)
 159         IDirectDrawMediaStream_Release(pddstream);
 160     if (pvidstream)
 161         IMediaStream_Release(pvidstream);
 162
 163     release_directdraw();
 164     release_ammultimediastream();
 165 }
 166
 167 static void test_media_streams(void)
 168 {
 169     HRESULT hr;
 170     IMediaStream *video_stream = NULL;
 171     IMediaStream *audio_stream = NULL;
 172     IMediaStream *dummy_stream;
 173     IDirectDrawMediaStream *ddraw_stream = NULL;
 174     IDirectDrawStreamSample *ddraw_sample = NULL;
 175
 176     if (!create_ammultimediastream())
 177         return;
 178     if (!create_directdraw())
 179     {
 180         release_ammultimediastream();
 181         return;
 182     }
 183
 184     hr = IAMMultiMediaStream_Initialize(pams, STREAMTYPE_READ, 0, NULL);
 185     ok(hr == S_OK, "IAMMultiMediaStream_Initialize returned: %x\n", hr);
 186
 187     /* Verify there is no video media stream */
 188     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryVideo, &video_stream);
 189     ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 190
 191     /* Verify there is no default renderer for video stream */
 192     hr = IAMMultiMediaStream_AddMediaStream(pams, (IUnknown*)pdd7, &MSPID_PrimaryVideo, AMMSF_ADDDEFAULTRENDERER, NULL);
 193     todo_wine ok(hr == MS_E_PURPOSEID, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 194     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryVideo, &video_stream);
 195     todo_wine ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 196     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &MSPID_PrimaryVideo, AMMSF_ADDDEFAULTRENDERER, NULL);
 197     todo_wine ok(hr == MS_E_PURPOSEID, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 198     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryVideo, &video_stream);
 199     todo_wine ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 200
 201     /* Verify normal case for video stream */
 202     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &MSPID_PrimaryVideo, 0, NULL);
 203     ok(hr == S_OK, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 204     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryVideo, &video_stream);
 205     ok(hr == S_OK, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 206
 207     /* Verify there is no audio media stream */
 208     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryAudio, &audio_stream);
 209     ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 210
 211     /* Verify no stream is created when using the default renderer for audio stream */
 212     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &MSPID_PrimaryAudio, AMMSF_ADDDEFAULTRENDERER, NULL);
 213     ok((hr == S_OK) || (hr == VFW_E_NO_AUDIO_HARDWARE), "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 214     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryAudio, &audio_stream);
 215     todo_wine ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 216
 217     /* Verify a stream is created when no default renderer is used */
 218     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &MSPID_PrimaryAudio, 0, NULL);
 219     ok(hr == S_OK, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 220     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryAudio, &audio_stream);
 221     ok(hr == S_OK, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 222
 223     /* Verify behaviour with invalid purpose id */
 224     hr = IAMMultiMediaStream_GetMediaStream(pams, &IID_IUnknown, &dummy_stream);
 225     ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 226     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &IID_IUnknown, 0, NULL);
 227     todo_wine ok(hr == MS_E_PURPOSEID, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 228
 229     if (video_stream)
 230     {
 231         hr = IMediaStream_QueryInterface(video_stream, &IID_IDirectDrawMediaStream, (LPVOID*)&ddraw_stream);
 232         ok(hr == S_OK, "IMediaStream_QueryInterface returned: %x\n", hr);
 233
 234         if (SUCCEEDED(hr))
 235         {
 236             hr = IDirectDrawMediaStream_CreateSample(ddraw_stream, NULL, NULL, 0, &ddraw_sample);
 237             todo_wine ok(hr==S_OK, "IDirectDrawMediaStream_CreateSample returned: %x\n", hr);
 238         }
 239     }
 240
 241     if (ddraw_sample)
 242         IDirectDrawMediaSample_Release(ddraw_sample);
 243     if (ddraw_stream)
 244         IDirectDrawMediaStream_Release(ddraw_stream);
 245     if (video_stream)
 246         IMediaStream_Release(video_stream);
 247     if (audio_stream)
 248         IMediaStream_Release(audio_stream);
 249
 250     release_directdraw();
 251     release_ammultimediastream();
 252 }
 253
 254 START_TEST(amstream)
 255 {
 256     HANDLE file;
 257
 258     CoInitializeEx(NULL, COINIT_MULTITHREADED);
 259
 260     test_media_streams();
 261
 262     file = CreateFileW(filenameW, 0, 0, NULL, OPEN_EXISTING, 0, NULL);
 263     if (file != INVALID_HANDLE_VALUE)
 264     {
 265         CloseHandle(file);
 266
 267         test_openfile();
 268         test_renderfile();
 269     }
 270
 271     CoUninitialize();
 272 }