dlls/qedit/tests/mediadet.c

   1 /*
   2  * Unit tests for Media Detector
   3  *
   4  * Copyright (C) 2008 Google (Lei Zhang, Dan Hipschman)
   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 #include <assert.h>
  22
  23 #define COBJMACROS
  24
  25 #include "initguid.h"
  26 #include "ole2.h"
  27 #include "vfwmsgs.h"
  28 #include "uuids.h"
  29 #include "wine/test.h"
  30 #include "qedit.h"
  31 #include "rc.h"
  32
  33 static WCHAR test_avi_filename[MAX_PATH];
  34 static WCHAR test_sound_avi_filename[MAX_PATH];
  35
  36 static BOOL unpack_avi_file(int id, WCHAR name[MAX_PATH])
  37 {
  38     static WCHAR temp_path[MAX_PATH];
  39     static WCHAR prefix[] = {'D','E','S',0};
  40     static WCHAR avi[] = {'a','v','i',0};
  41     HRSRC res;
  42     HGLOBAL data;
  43     char *mem;
  44     DWORD size, written;
  45     HANDLE fh;
  46
  47     res = FindResource(NULL, MAKEINTRESOURCE(id), MAKEINTRESOURCE(AVI_RES_TYPE));
  48     if (!res)
  49         return FALSE;
  50
  51     data = LoadResource(NULL, res);
  52     if (!data)
  53         return FALSE;
  54
  55     mem = LockResource(data);
  56     if (!mem)
  57         return FALSE;
  58
  59     size = SizeofResource(NULL, res);
  60     if (size == 0)
  61         return FALSE;
  62
  63     if (!GetTempPathW(MAX_PATH, temp_path))
  64         return FALSE;
  65
  66     /* We might end up relying on the extension here, so .TMP is no good.  */
  67     if (!GetTempFileNameW(temp_path, prefix, 0, name))
  68         return FALSE;
  69
  70     DeleteFileW(name);
  71     lstrcpyW(name + lstrlenW(name) - 3, avi);
  72
  73     fh = CreateFileW(name, GENERIC_WRITE, 0, NULL, CREATE_NEW,
  74                      FILE_ATTRIBUTE_NORMAL, NULL);
  75     if (fh == INVALID_HANDLE_VALUE)
  76         return FALSE;
  77
  78     if (!WriteFile(fh, mem, size, &written, NULL) || written != size)
  79         return FALSE;
  80
  81     CloseHandle(fh);
  82
  83     return TRUE;
  84 }
  85
  86 static BOOL init_tests(void)
  87 {
  88     return unpack_avi_file(TEST_AVI_RES, test_avi_filename)
  89         && unpack_avi_file(TEST_SOUND_AVI_RES, test_sound_avi_filename);
  90 }
  91
  92 static void test_mediadet(void)
  93 {
  94     HRESULT hr;
  95     IMediaDet *pM = NULL;
  96     BSTR filename = NULL;
  97     LONG nstrms = 0;
  98     LONG strm;
  99     AM_MEDIA_TYPE mt;
 100     double fps;
 101     int flags;
 102     int i;
 103
 104     /* test.avi has one video stream.  */
 105     hr = CoCreateInstance(&CLSID_MediaDet, NULL, CLSCTX_INPROC_SERVER,
 106             &IID_IMediaDet, (LPVOID*)&pM);
 107     ok(hr == S_OK, "CoCreateInstance failed with %x\n", hr);
 108     ok(pM != NULL, "pM is NULL\n");
 109
 110     filename = NULL;
 111     hr = IMediaDet_get_Filename(pM, &filename);
 112     /* Despite what MSDN claims, this returns S_OK.  */
 113     ok(hr == S_OK, "IMediaDet_get_Filename\n");
 114     ok(filename == NULL, "IMediaDet_get_Filename\n");
 115
 116     filename = (BSTR) -1;
 117     hr = IMediaDet_get_Filename(pM, &filename);
 118     /* Despite what MSDN claims, this returns S_OK.  */
 119     ok(hr == S_OK, "IMediaDet_get_Filename\n");
 120     ok(filename == NULL, "IMediaDet_get_Filename\n");
 121
 122     nstrms = -1;
 123     hr = IMediaDet_get_OutputStreams(pM, &nstrms);
 124     ok(hr == E_INVALIDARG, "IMediaDet_get_OutputStreams\n");
 125     ok(nstrms == -1, "IMediaDet_get_OutputStreams\n");
 126
 127     strm = -1;
 128     /* The stream defaults to 0, even without a file!  */
 129     hr = IMediaDet_get_CurrentStream(pM, &strm);
 130     ok(hr == S_OK, "IMediaDet_get_CurrentStream\n");
 131     ok(strm == 0, "IMediaDet_get_CurrentStream\n");
 132
 133     hr = IMediaDet_get_CurrentStream(pM, NULL);
 134     ok(hr == E_POINTER, "IMediaDet_get_CurrentStream\n");
 135
 136     /* But put_CurrentStream doesn't.  */
 137     hr = IMediaDet_put_CurrentStream(pM, 0);
 138     ok(hr == E_INVALIDARG, "IMediaDet_put_CurrentStream\n");
 139
 140     hr = IMediaDet_put_CurrentStream(pM, -1);
 141     ok(hr == E_INVALIDARG, "IMediaDet_put_CurrentStream\n");
 142
 143     hr = IMediaDet_get_StreamMediaType(pM, &mt);
 144     ok(hr == E_INVALIDARG, "IMediaDet_get_StreamMediaType\n");
 145
 146     hr = IMediaDet_get_StreamMediaType(pM, NULL);
 147     ok(hr == E_POINTER, "IMediaDet_get_StreamMediaType\n");
 148
 149     filename = SysAllocString(test_avi_filename);
 150     hr = IMediaDet_put_Filename(pM, filename);
 151     ok(hr == S_OK, "IMediaDet_put_Filename -> %x\n", hr);
 152     SysFreeString(filename);
 153
 154     strm = -1;
 155     /* The stream defaults to 0.  */
 156     hr = IMediaDet_get_CurrentStream(pM, &strm);
 157     ok(hr == S_OK, "IMediaDet_get_CurrentStream\n");
 158     ok(strm == 0, "IMediaDet_get_CurrentStream\n");
 159
 160     ZeroMemory(&mt, sizeof mt);
 161     hr = IMediaDet_get_StreamMediaType(pM, &mt);
 162     ok(hr == S_OK, "IMediaDet_get_StreamMediaType\n");
 163     CoTaskMemFree(mt.pbFormat);
 164
 165     /* Even before get_OutputStreams.  */
 166     hr = IMediaDet_put_CurrentStream(pM, 1);
 167     ok(hr == E_INVALIDARG, "IMediaDet_put_CurrentStream\n");
 168
 169     hr = IMediaDet_get_OutputStreams(pM, &nstrms);
 170     ok(hr == S_OK, "IMediaDet_get_OutputStreams\n");
 171     ok(nstrms == 1, "IMediaDet_get_OutputStreams\n");
 172
 173     filename = NULL;
 174     hr = IMediaDet_get_Filename(pM, &filename);
 175     ok(hr == S_OK, "IMediaDet_get_Filename\n");
 176     ok(lstrcmpW(filename, test_avi_filename) == 0,
 177        "IMediaDet_get_Filename\n");
 178     SysFreeString(filename);
 179
 180     hr = IMediaDet_get_Filename(pM, NULL);
 181     ok(hr == E_POINTER, "IMediaDet_get_Filename\n");
 182
 183     strm = -1;
 184     hr = IMediaDet_get_CurrentStream(pM, &strm);
 185     ok(hr == S_OK, "IMediaDet_get_CurrentStream\n");
 186     ok(strm == 0, "IMediaDet_get_CurrentStream\n");
 187
 188     hr = IMediaDet_get_CurrentStream(pM, NULL);
 189     ok(hr == E_POINTER, "IMediaDet_get_CurrentStream\n");
 190
 191     hr = IMediaDet_put_CurrentStream(pM, -1);
 192     ok(hr == E_INVALIDARG, "IMediaDet_put_CurrentStream\n");
 193
 194     hr = IMediaDet_put_CurrentStream(pM, 1);
 195     ok(hr == E_INVALIDARG, "IMediaDet_put_CurrentStream\n");
 196
 197     /* Try again.  */
 198     strm = -1;
 199     hr = IMediaDet_get_CurrentStream(pM, &strm);
 200     ok(hr == S_OK, "IMediaDet_get_CurrentStream\n");
 201     ok(strm == 0, "IMediaDet_get_CurrentStream\n");
 202
 203     hr = IMediaDet_put_CurrentStream(pM, 0);
 204     ok(hr == S_OK, "IMediaDet_put_CurrentStream\n");
 205
 206     strm = -1;
 207     hr = IMediaDet_get_CurrentStream(pM, &strm);
 208     ok(hr == S_OK, "IMediaDet_get_CurrentStream\n");
 209     ok(strm == 0, "IMediaDet_get_CurrentStream\n");
 210
 211     ZeroMemory(&mt, sizeof mt);
 212     hr = IMediaDet_get_StreamMediaType(pM, &mt);
 213     ok(hr == S_OK, "IMediaDet_get_StreamMediaType\n");
 214     ok(IsEqualGUID(&mt.majortype, &MEDIATYPE_Video),
 215                  "IMediaDet_get_StreamMediaType\n");
 216     CoTaskMemFree(mt.pbFormat);
 217
 218     hr = IMediaDet_get_FrameRate(pM, NULL);
 219     ok(hr == E_POINTER, "IMediaDet_get_FrameRate\n");
 220
 221     hr = IMediaDet_get_FrameRate(pM, &fps);
 222     ok(hr == S_OK, "IMediaDet_get_FrameRate\n");
 223     ok(fps == 10.0, "IMediaDet_get_FrameRate\n");
 224
 225     hr = IMediaDet_Release(pM);
 226     ok(hr == 0, "IMediaDet_Release returned: %x\n", hr);
 227
 228     DeleteFileW(test_avi_filename);
 229
 230     /* test_sound.avi has one video stream and one audio stream.  */
 231     hr = CoCreateInstance(&CLSID_MediaDet, NULL, CLSCTX_INPROC_SERVER,
 232             &IID_IMediaDet, (LPVOID*)&pM);
 233     ok(hr == S_OK, "CoCreateInstance failed with %x\n", hr);
 234     ok(pM != NULL, "pM is NULL\n");
 235
 236     filename = SysAllocString(test_sound_avi_filename);
 237     hr = IMediaDet_put_Filename(pM, filename);
 238     ok(hr == S_OK, "IMediaDet_put_Filename -> %x\n", hr);
 239     SysFreeString(filename);
 240
 241     hr = IMediaDet_get_OutputStreams(pM, &nstrms);
 242     ok(hr == S_OK, "IMediaDet_get_OutputStreams\n");
 243     ok(nstrms == 2, "IMediaDet_get_OutputStreams\n");
 244
 245     filename = NULL;
 246     hr = IMediaDet_get_Filename(pM, &filename);
 247     ok(hr == S_OK, "IMediaDet_get_Filename\n");
 248     ok(lstrcmpW(filename, test_sound_avi_filename) == 0,
 249        "IMediaDet_get_Filename\n");
 250     SysFreeString(filename);
 251
 252     /* I don't know if the stream order is deterministic.  Just check
 253        for both an audio and video stream.  */
 254     flags = 0;
 255
 256     for (i = 0; i < 2; ++i)
 257     {
 258         hr = IMediaDet_put_CurrentStream(pM, i);
 259         ok(hr == S_OK, "IMediaDet_put_CurrentStream\n");
 260
 261         strm = -1;
 262         hr = IMediaDet_get_CurrentStream(pM, &strm);
 263         ok(hr == S_OK, "IMediaDet_get_CurrentStream\n");
 264         ok(strm == i, "IMediaDet_get_CurrentStream\n");
 265
 266         ZeroMemory(&mt, sizeof mt);
 267         hr = IMediaDet_get_StreamMediaType(pM, &mt);
 268         ok(hr == S_OK, "IMediaDet_get_StreamMediaType\n");
 269         flags += (IsEqualGUID(&mt.majortype, &MEDIATYPE_Video)
 270                   ? 1
 271                   : (IsEqualGUID(&mt.majortype, &MEDIATYPE_Audio)
 272                      ? 2
 273                      : 0));
 274
 275         if (IsEqualGUID(&mt.majortype, &MEDIATYPE_Audio))
 276         {
 277             hr = IMediaDet_get_FrameRate(pM, &fps);
 278             ok(hr == VFW_E_INVALIDMEDIATYPE, "IMediaDet_get_FrameRate\n");
 279         }
 280
 281         CoTaskMemFree(mt.pbFormat);
 282     }
 283     ok(flags == 3, "IMediaDet_get_StreamMediaType\n");
 284
 285     hr = IMediaDet_put_CurrentStream(pM, 2);
 286     ok(hr == E_INVALIDARG, "IMediaDet_put_CurrentStream\n");
 287
 288     strm = -1;
 289     hr = IMediaDet_get_CurrentStream(pM, &strm);
 290     ok(hr == S_OK, "IMediaDet_get_CurrentStream\n");
 291     ok(strm == 1, "IMediaDet_get_CurrentStream\n");
 292
 293     hr = IMediaDet_Release(pM);
 294     ok(hr == 0, "IMediaDet_Release returned: %x\n", hr);
 295
 296     DeleteFileW(test_sound_avi_filename);
 297 }
 298
 299 START_TEST(mediadet)
 300 {
 301     if (!init_tests())
 302     {
 303         skip("Couldn't initialize tests!\n");
 304         return;
 305     }
 306
 307     CoInitialize(NULL);
 308     test_mediadet();
 309     CoUninitialize();
 310 }