Michael Stefaniuc : dmime/tests: Add some COM tests for DirectMusicSegment.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: edcf723bab1d6512bc81ce64b1d3c873e281f0f6
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=edcf723bab1d6512bc81ce64b1d3c873e281f0f6

Author: Michael Stefaniuc <mstefani at redhat.de>
Date:   Fri Apr  6 00:22:49 2012 +0200

dmime/tests: Add some COM tests for DirectMusicSegment.

---

 dlls/dmime/tests/performance.c |   53 ++++++++++++++++++++++++++++++++++++++++
 1 files changed, 53 insertions(+), 0 deletions(-)

diff --git a/dlls/dmime/tests/performance.c b/dlls/dmime/tests/performance.c
index 931478c..684642b 100644
--- a/dlls/dmime/tests/performance.c
+++ b/dlls/dmime/tests/performance.c
@@ -18,6 +18,8 @@
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  */
 
+#define COBJMACROS
+
 #include <stdarg.h>
 #include <windef.h>
 #include <initguid.h>
@@ -239,6 +241,55 @@ static void test_COM(void)
     ok (refcount == 0, "refcount == %u, expected 0\n", refcount);
 }
 
+static void test_dms_COM(void)
+{
+    IDirectMusicSegment8 *dms = (IDirectMusicSegment8*)0xdeadbeef;
+    IDirectMusicObject *dmo;
+    IPersistStream *stream;
+    IUnknown *unk;
+    ULONG refcount;
+    HRESULT hr;
+
+    /* COM aggregation */
+    hr = CoCreateInstance(&CLSID_DirectMusicSegment, (IUnknown*)&dms, CLSCTX_INPROC_SERVER,
+            &IID_IUnknown, (void**)&dms);
+    ok(hr == CLASS_E_NOAGGREGATION,
+            "DirectMusicSegment create failed: %08x, expected CLASS_E_NOAGGREGATION\n", hr);
+    ok(!dms, "dms = %p\n", dms);
+
+    /* Invalid RIID */
+    hr = CoCreateInstance(&CLSID_DirectMusicSegment, NULL, CLSCTX_INPROC_SERVER,
+            &IID_IDirectSound, (void**)&dms);
+    ok(hr == E_NOINTERFACE,
+            "DirectMusicSegment create failed: %08x, expected E_NOINTERFACE\n", hr);
+
+    /* Same refcount */
+    hr = CoCreateInstance(&CLSID_DirectMusicSegment, NULL, CLSCTX_INPROC_SERVER,
+            &IID_IDirectMusicSegment8, (void**)&dms);
+    ok(hr == S_OK, "DirectMusicSegment create failed: %08x, expected S_OK\n", hr);
+    refcount = IDirectMusicSegment8_AddRef(dms);
+    ok (refcount == 2, "refcount == %u, expected 2\n", refcount);
+    hr = IDirectMusicSegment8_QueryInterface(dms, &IID_IDirectMusicObject, (void**)&dmo);
+    ok(hr == S_OK, "QueryInterface for IID_IDirectMusicObject failed: %08x\n", hr);
+    IDirectMusicSegment8_AddRef(dms);
+    refcount = IDirectMusicSegment8_Release(dms);
+    ok (refcount == 3, "refcount == %u, expected 3\n", refcount);
+    hr = IDirectMusicSegment8_QueryInterface(dms, &IID_IPersistStream, (void**)&stream);
+    ok(hr == S_OK, "QueryInterface for IID_IPersistStream failed: %08x\n", hr);
+    refcount = IDirectMusicSegment8_Release(dms);
+    ok (refcount == 3, "refcount == %u, expected 3\n", refcount);
+    hr = IDirectMusicSegment8_QueryInterface(dms, &IID_IUnknown, (void**)&unk);
+    ok(hr == S_OK, "QueryInterface for IID_IUnknown failed: %08x\n", hr);
+    refcount = IUnknown_Release(unk);
+    ok (refcount == 3, "refcount == %u, expected 3\n", refcount);
+    refcount = IDirectMusicObject_Release(dmo);
+    ok (refcount == 2, "refcount == %u, expected 2\n", refcount);
+    refcount = IPersistStream_Release(stream);
+    ok (refcount == 1, "refcount == %u, expected 1\n", refcount);
+    refcount = IDirectMusicSegment8_Release(dms);
+    ok (refcount == 0, "refcount == %u, expected 0\n", refcount);
+}
+
 START_TEST( performance )
 {
     HRESULT hr;
@@ -258,5 +309,7 @@ START_TEST( performance )
     test_COM();
     test_createport();
 
+    test_dms_COM();
+
     CoUninitialize();
 }