1 /*
2  * Unit tests for Direct Show functions
3  *
4  * Copyright (C) 2005 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 "uuids.h"
25 #include "dshow.h"
26 #include "control.h"
27 
28 static void CommitDecommitTest(void)
29 {
30     IMemAllocator* pMemAllocator;
31     HRESULT hr;
32 
33     hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC_SERVER, &IID_IMemAllocator, (LPVOID*)&pMemAllocator);
34     ok(hr==S_OK, "Unable to create memory allocator %x\n", hr);
35 
36     if (hr == S_OK)
37     {
38         ALLOCATOR_PROPERTIES RequestedProps;
39         ALLOCATOR_PROPERTIES ActualProps;
40 
41         IMediaSample *sample = NULL, *sample2 = NULL;
42 
43         RequestedProps.cBuffers = 2;
44         RequestedProps.cbBuffer = 65536;
45         RequestedProps.cbAlign = 1;
46         RequestedProps.cbPrefix = 0;
47 
48 	hr = IMemAllocator_SetProperties(pMemAllocator, &RequestedProps, &ActualProps);
49 	ok(hr==S_OK, "SetProperties returned: %x\n", hr);
50 
51 	hr = IMemAllocator_Commit(pMemAllocator);
52 	ok(hr==S_OK, "Commit returned: %x\n", hr);
53 	hr = IMemAllocator_Commit(pMemAllocator);
54 	ok(hr==S_OK, "Commit returned: %x\n", hr);
55 
56         hr = IMemAllocator_GetBuffer(pMemAllocator, &sample, NULL, NULL, 0);
57         ok(hr==S_OK, "Could not get a buffer: %x\n", hr);
58 
59 	hr = IMemAllocator_Decommit(pMemAllocator);
60 	ok(hr==S_OK, "Decommit returned: %x\n", hr);
61 	hr = IMemAllocator_Decommit(pMemAllocator);
62 	ok(hr==S_OK, "Cecommit returned: %x\n", hr);
63 
64         /* Decommit and recommit while holding a sample */
65         if (sample)
66         {
67             hr = IMemAllocator_Commit(pMemAllocator);
68             ok(hr==S_OK, "Commit returned: %x\n", hr);
69 
70             hr = IMemAllocator_GetBuffer(pMemAllocator, &sample2, NULL, NULL, 0);
71             ok(hr==S_OK, "Could not get a buffer: %x\n", hr);
72             IMediaSample_Release(sample);
73             if (sample2)
74                 IMediaSample_Release(sample2);
75 
76             hr = IMemAllocator_Decommit(pMemAllocator);
77             ok(hr==S_OK, "Cecommit returned: %x\n", hr);
78         }
79         IMemAllocator_Release(pMemAllocator);
80     }
81 }
82 
83 START_TEST(memallocator)
84 {
85     CoInitialize(NULL);
86 
87     CommitDecommitTest();
88 
89     CoUninitialize();
90 }
91