dlls/msxml3/tests/schema.c

   1 /*
   2  * Schema test
   3  *
   4  * Copyright 2007 Huw Davies
   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 <stdio.h>
  22 #define COBJMACROS
  23
  24 #include "initguid.h"
  25 #include "windows.h"
  26 #include "ole2.h"
  27 #include "xmldom.h"
  28 #include "msxml2.h"
  29 #include "dispex.h"
  30
  31 #include "wine/test.h"
  32
  33 static const WCHAR schema_uri[] = {'x','-','s','c','h','e','m','a',':','t','e','s','t','.','x','m','l',0};
  34
  35 static const WCHAR schema_xml[] = {
  36     '<','S','c','h','e','m','a',' ','x','m','l','n','s','=','\"','u','r','n',':','s','c','h','e','m','a','s','-','m','i','c','r','o','s','o','f','t','-','c','o','m',':','x','m','l','-','d','a','t','a','\"','\n',
  37     'x','m','l','n','s',':','d','t','=','\"','u','r','n',':','s','c','h','e','m','a','s','-','m','i','c','r','o','s','o','f','t','-','c','o','m',':','d','a','t','a','t','y','p','e','s','\"','>','\n',
  38     '<','/','S','c','h','e','m','a','>','\n',0
  39 };
  40
  41 static void test_schema_refs(void)
  42 {
  43     IXMLDOMDocument2 *doc;
  44     IXMLDOMSchemaCollection *schema;
  45     HRESULT r;
  46     LONG ref;
  47     VARIANT v;
  48     VARIANT_BOOL b;
  49     BSTR str;
  50
  51     r = CoCreateInstance( &CLSID_DOMDocument, NULL,
  52         CLSCTX_INPROC_SERVER, &IID_IXMLDOMDocument2, (LPVOID*)&doc );
  53     if( r != S_OK )
  54         return;
  55
  56     r = CoCreateInstance( &CLSID_XMLSchemaCache, NULL,
  57         CLSCTX_INPROC_SERVER, &IID_IXMLDOMSchemaCollection, (LPVOID*)&schema );
  58     if( r != S_OK )
  59     {
  60         IXMLDOMDocument2_Release(doc);
  61         return;
  62     }
  63
  64     str = SysAllocString(schema_xml);
  65     r = IXMLDOMDocument2_loadXML(doc, str, &b);
  66     ok(r == S_OK, "ret %08x\n", r);
  67     ok(b == VARIANT_TRUE, "b %04x\n", b);
  68     SysFreeString(str);
  69
  70     ref = IXMLDOMDocument2_AddRef(doc);
  71     ok(ref == 2, "ref %d\n", ref);
  72     VariantInit(&v);
  73     V_VT(&v) = VT_DISPATCH;
  74     V_DISPATCH(&v) = (IDispatch*)doc;
  75
  76     str = SysAllocString(schema_uri);
  77     r = IXMLDOMSchemaCollection_add(schema, str, v);
  78     ok(r == S_OK, "ret %08x\n", r);
  79
  80     /* IXMLDOMSchemaCollection_add doesn't add a ref on doc */
  81     ref = IXMLDOMDocument2_AddRef(doc);
  82     ok(ref == 3, "ref %d\n", ref);
  83     IXMLDOMDocument2_Release(doc);
  84
  85     SysFreeString(str);
  86     VariantClear(&v);
  87
  88     V_VT(&v) = VT_INT;
  89     r = IXMLDOMDocument2_get_schemas(doc, &v);
  90     ok(r == S_FALSE, "ret %08x\n", r);
  91     ok(V_VT(&v) == VT_NULL, "vt %x\n", V_VT(&v));
  92
  93     ref = IXMLDOMSchemaCollection_AddRef(schema);
  94     ok(ref == 2, "ref %d\n", ref);
  95     V_VT(&v) = VT_DISPATCH;
  96     V_DISPATCH(&v) = (IDispatch*)schema;
  97
  98     /* check that putref_schemas takes a ref */
  99     r = IXMLDOMDocument2_putref_schemas(doc, v);
 100     ok(r == S_OK, "ret %08x\n", r);
 101     ref = IXMLDOMSchemaCollection_AddRef(schema);
 102     ok(ref == 4, "ref %d\n", ref);
 103     IXMLDOMSchemaCollection_Release(schema);
 104     VariantClear(&v);
 105
 106     /* refs now 2 */
 107     V_VT(&v) = VT_INT;
 108     /* check that get_schemas adds a ref */
 109     r = IXMLDOMDocument2_get_schemas(doc, &v);
 110     ok(r == S_OK, "ret %08x\n", r);
 111     ok(V_VT(&v) == VT_DISPATCH, "vt %x\n", V_VT(&v));
 112     ref = IXMLDOMSchemaCollection_AddRef(schema);
 113     ok(ref == 4, "ref %d\n", ref);
 114     IXMLDOMSchemaCollection_Release(schema);
 115
 116     /* refs now 3 */
 117     /* get_schemas doesn't release a ref if passed VT_DISPATCH - ie it doesn't call VariantClear() */
 118     r = IXMLDOMDocument2_get_schemas(doc, &v);
 119     ok(r == S_OK, "ret %08x\n", r);
 120     ok(V_VT(&v) == VT_DISPATCH, "vt %x\n", V_VT(&v));
 121     ref = IXMLDOMSchemaCollection_AddRef(schema);
 122     ok(ref == 5, "ref %d\n", ref);
 123     IXMLDOMSchemaCollection_Release(schema);
 124
 125     /* refs now 4 */
 126     /* release the two refs returned by get_schemas */
 127     IXMLDOMSchemaCollection_Release(schema);
 128     IXMLDOMSchemaCollection_Release(schema);
 129
 130     /* refs now 2 */
 131
 132     /* check that taking another ref on the document doesn't change the schema's ref count */
 133     IXMLDOMDocument2_AddRef(doc);
 134     ref = IXMLDOMSchemaCollection_AddRef(schema);
 135     ok(ref == 3, "ref %d\n", ref);
 136     IXMLDOMSchemaCollection_Release(schema);
 137     IXMLDOMDocument2_Release(doc);
 138
 139
 140     /* refs now 2 */
 141     /* call putref_schema with some odd variants */
 142     V_VT(&v) = VT_INT;
 143     r = IXMLDOMDocument2_putref_schemas(doc, v);
 144     ok(r == E_FAIL, "ret %08x\n", r);
 145     ref = IXMLDOMSchemaCollection_AddRef(schema);
 146     ok(ref == 3, "ref %d\n", ref);
 147     IXMLDOMSchemaCollection_Release(schema);
 148
 149     /* refs now 2 */
 150     /* calling with VT_EMPTY releases the schema */
 151     V_VT(&v) = VT_EMPTY;
 152     r = IXMLDOMDocument2_putref_schemas(doc, v);
 153     ok(r == S_OK, "ret %08x\n", r);
 154     ref = IXMLDOMSchemaCollection_AddRef(schema);
 155     ok(ref == 2, "ref %d\n", ref);
 156     IXMLDOMSchemaCollection_Release(schema);
 157
 158     /* refs now 1 */
 159     /* try setting with VT_UNKNOWN */
 160     IXMLDOMSchemaCollection_AddRef(schema);
 161     V_VT(&v) = VT_UNKNOWN;
 162     V_UNKNOWN(&v) = (IUnknown*)schema;
 163     r = IXMLDOMDocument2_putref_schemas(doc, v);
 164     ok(r == S_OK, "ret %08x\n", r);
 165     ref = IXMLDOMSchemaCollection_AddRef(schema);
 166     ok(ref == 4, "ref %d\n", ref);
 167     IXMLDOMSchemaCollection_Release(schema);
 168     VariantClear(&v);
 169
 170     /* refs now 2 */
 171     /* calling with VT_NULL releases the schema */
 172     V_VT(&v) = VT_NULL;
 173     r = IXMLDOMDocument2_putref_schemas(doc, v);
 174     ok(r == S_OK, "ret %08x\n", r);
 175     ref = IXMLDOMSchemaCollection_AddRef(schema);
 176     ok(ref == 2, "ref %d\n", ref);
 177     IXMLDOMSchemaCollection_Release(schema);
 178
 179     /* refs now 1 */
 180     /* set again */
 181     IXMLDOMSchemaCollection_AddRef(schema);
 182     V_VT(&v) = VT_UNKNOWN;
 183     V_UNKNOWN(&v) = (IUnknown*)schema;
 184     r = IXMLDOMDocument2_putref_schemas(doc, v);
 185     ok(r == S_OK, "ret %08x\n", r);
 186     ref = IXMLDOMSchemaCollection_AddRef(schema);
 187     ok(ref == 4, "ref %d\n", ref);
 188     IXMLDOMSchemaCollection_Release(schema);
 189     VariantClear(&v);
 190
 191     /* refs now 2 */
 192
 193     /* release the final ref on the doc which should release its ref on the schema */
 194     IXMLDOMDocument2_Release(doc);
 195
 196     ref = IXMLDOMSchemaCollection_AddRef(schema);
 197     ok(ref == 2, "ref %d\n", ref);
 198     IXMLDOMSchemaCollection_Release(schema);
 199     IXMLDOMSchemaCollection_Release(schema);
 200 }
 201
 202 START_TEST(schema)
 203 {
 204     HRESULT r;
 205
 206     r = CoInitialize( NULL );
 207     ok( r == S_OK, "failed to init com\n");
 208
 209     test_schema_refs();
 210
 211     CoUninitialize();
 212 }