gdiplus: Add a test for image stream refcount.

This commit is contained in:
Dmitry Timoshkov 2012-06-27 14:27:04 +09:00 committed by Alexandre Julliard
parent 3cd0834a24
commit 55f160db34
1 changed files with 19 additions and 1 deletions

View File

@ -2656,6 +2656,12 @@ static void test_dispose(void)
expect(ObjectBusy, stat); expect(ObjectBusy, stat);
} }
static LONG obj_refcount(void *obj)
{
IUnknown_AddRef((IUnknown *)obj);
return IUnknown_Release((IUnknown *)obj);
}
static GpImage *load_image(const BYTE *image_data, UINT image_size) static GpImage *load_image(const BYTE *image_data, UINT image_size)
{ {
IStream *stream; IStream *stream;
@ -2664,6 +2670,8 @@ static GpImage *load_image(const BYTE *image_data, UINT image_size)
HRESULT hr; HRESULT hr;
GpStatus status; GpStatus status;
GpImage *image = NULL; GpImage *image = NULL;
ImageType image_type;
LONG refcount;
hmem = GlobalAlloc(0, image_size); hmem = GlobalAlloc(0, image_size);
data = GlobalLock(hmem); data = GlobalLock(hmem);
@ -2674,10 +2682,20 @@ static GpImage *load_image(const BYTE *image_data, UINT image_size)
ok(hr == S_OK, "CreateStreamOnHGlobal error %#x\n", hr); ok(hr == S_OK, "CreateStreamOnHGlobal error %#x\n", hr);
if (hr != S_OK) return NULL; if (hr != S_OK) return NULL;
refcount = obj_refcount(stream);
ok(refcount == 1, "expected stream refcount 1, got %d\n", refcount);
status = GdipLoadImageFromStream(stream, &image); status = GdipLoadImageFromStream(stream, &image);
ok(status == Ok, "GdipLoadImageFromStream error %d\n", status); ok(status == Ok, "GdipLoadImageFromStream error %d\n", status);
IStream_Release(stream); status = GdipGetImageType(image, &image_type);
ok(status == Ok, "GdipGetImageType error %d\n", status);
refcount = IStream_Release(stream);
if (image_type == ImageTypeBitmap)
todo_wine ok(refcount >= 1, "expected stream refcount != 0\n");
else
ok(refcount == 0, "expected stream refcount 0, got %d\n", refcount);
return image; return image;
} }