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