sal/qa/systools/test_comtools.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*************************************************************************
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * Copyright 2000, 2010 Oracle and/or its affiliates.
   7  *
   8  * OpenOffice.org - a multi-platform office productivity suite
   9  *
  10  * This file is part of OpenOffice.org.
  11  *
  12  * OpenOffice.org is free software: you can redistribute it and/or modify
  13  * it under the terms of the GNU Lesser General Public License version 3
  14  * only, as published by the Free Software Foundation.
  15  *
  16  * OpenOffice.org is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  * GNU Lesser General Public License version 3 for more details
  20  * (a copy is included in the LICENSE file that accompanied this code).
  21  *
  22  * You should have received a copy of the GNU Lesser General Public License
  23  * version 3 along with OpenOffice.org.  If not, see
  24  * <http://www.openoffice.org/license.html>
  25  * for a copy of the LGPLv3 License.
  26  *
  27  ************************************************************************/
  28
  29
  30 // autogenerated file with codegen.pl
  31
  32 #include <testshl/simpleheader.hxx>
  33 #include <systools/win32/comtools.hxx>
  34
  35 class COMObject : public IUnknown
  36 {
  37 public:
  38     COMObject() : ref_count_(0)
  39     {
  40     }
  41
  42     ~COMObject()
  43     {
  44     }
  45
  46     ULONG __stdcall AddRef()
  47     {
  48         ref_count_++;
  49         return ref_count_;
  50     }
  51
  52     ULONG __stdcall Release()
  53     {
  54         ULONG cnt = --ref_count_;
  55         if (cnt == 0)
  56             delete this;
  57         return cnt;
  58     }
  59
  60     HRESULT __stdcall QueryInterface(REFIID riid, LPVOID* ppv)
  61     {
  62         if (riid == IID_IUnknown)
  63         {
  64             AddRef();
  65             *ppv = this;
  66             return S_OK;
  67         }
  68         return E_NOINTERFACE;
  69     }
  70
  71     ULONG GetRefCount() const
  72     {
  73         return ref_count_;
  74     }
  75
  76 private:
  77     ULONG ref_count_;
  78 };
  79
  80 sal::systools::COMReference<IUnknown> comObjectSource()
  81 {
  82     return sal::systools::COMReference<IUnknown>(new COMObject);
  83 }
  84
  85 bool comObjectSink(sal::systools::COMReference<IUnknown> r, ULONG expectedRefCountOnReturn)
  86 {
  87     r = sal::systools::COMReference<IUnknown>();
  88     COMObject* p = reinterpret_cast<COMObject*>(r.get());
  89     if (p)
  90         return (p->GetRefCount() == expectedRefCountOnReturn);
  91     else
  92         return (0 == expectedRefCountOnReturn);
  93 }
  94
  95 void comObjectSource2(LPVOID* ppv)
  96 {
  97     COMObject* p = new COMObject;
  98     p->AddRef();
  99     *ppv = p;
 100 }
 101
 102 namespace test_comtools
 103 {
 104
 105     class test_COMReference : public CppUnit::TestFixture
 106     {
 107
 108     public:
 109         /// test of COMReference<IUnknown> r;
 110         void default_ctor()
 111         {
 112             sal::systools::COMReference<IUnknown> r;
 113             CPPUNIT_ASSERT_MESSAGE("COMReference should be empty", r.get() == NULL);
 114         }
 115
 116         void test_ctor_manual_AddRef()
 117         {
 118             COMObject* p = new COMObject;
 119             p->AddRef();
 120             sal::systools::COMReference<IUnknown> r(p, false);
 121             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 122         }
 123
 124         void test_copy_ctor()
 125         {
 126             sal::systools::COMReference<IUnknown> r(comObjectSource());
 127             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 128         }
 129
 130         void test_copy_assignment()
 131         {
 132             sal::systools::COMReference<IUnknown> r;
 133             CPPUNIT_ASSERT_MESSAGE("COMReference should be empty", r.get() == NULL);
 134
 135             r = comObjectSource();
 136             CPPUNIT_ASSERT_MESSAGE("COMReference should be empty", r.get() != NULL);
 137             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 138         }
 139
 140         void test_ref_to_ref_assignment()
 141         {
 142             sal::systools::COMReference<IUnknown> r1 = comObjectSource();
 143             sal::systools::COMReference<IUnknown> r2 = r1;
 144             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 2 is expected", reinterpret_cast<COMObject*>(r2.get())->GetRefCount() == 2);
 145         }
 146
 147         void test_pointer_to_ref_assignment()
 148         {
 149             sal::systools::COMReference<IUnknown> r;
 150             r = new COMObject;
 151             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 152         }
 153
 154         void test_pointer_to_ref_assignment2()
 155         {
 156             sal::systools::COMReference<IUnknown> r = comObjectSource();
 157             r = new COMObject;
 158             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 159         }
 160
 161         void test_source_sink()
 162         {
 163             CPPUNIT_ASSERT_MESSAGE("Wrong reference count, 0 is expected", comObjectSink(comObjectSource(), 0));
 164         }
 165
 166         void test_address_operator()
 167         {
 168             sal::systools::COMReference<IUnknown> r;
 169             comObjectSource2(reinterpret_cast<LPVOID*>(&r));
 170             CPPUNIT_ASSERT_MESSAGE("Wrong reference count, 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 171         }
 172
 173         void test_address_operator2()
 174         {
 175             sal::systools::COMReference<IUnknown> r1 = comObjectSource();
 176             sal::systools::COMReference<IUnknown> r2 = r1;
 177             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 2 is expected", reinterpret_cast<COMObject*>(r2.get())->GetRefCount() == 2);
 178             comObjectSource2(reinterpret_cast<LPVOID*>(&r1));
 179             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r1.get())->GetRefCount() == 1);
 180             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r2.get())->GetRefCount() == 1);
 181         }
 182
 183         void test_clear()
 184         {
 185             sal::systools::COMReference<IUnknown> r = comObjectSource();
 186             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 187             r.clear();
 188             CPPUNIT_ASSERT_MESSAGE("Expect reference to be empty", !r.is());
 189         }
 190
 191         void test_query_interface()
 192         {
 193             try
 194             {
 195                 sal::systools::COMReference<IUnknown> r1 = comObjectSource();
 196                 sal::systools::COMReference<IUnknown> r2 = r1.QueryInterface<IUnknown>(IID_IUnknown);
 197                 CPPUNIT_ASSERT_MESSAGE("Wrong reference count, 2 is expected", reinterpret_cast<COMObject*>(r2.get())->GetRefCount() == 2);
 198             }
 199             catch(const sal::systools::ComError& ex)
 200             {
 201                 CPPUNIT_ASSERT_MESSAGE("Exception should not have been thrown", false);
 202             }
 203         }
 204
 205         void test_query_interface_throw()
 206         {
 207             try
 208             {
 209                 sal::systools::COMReference<IUnknown> r1 = comObjectSource();
 210                 sal::systools::COMReference<IPersistFile> r2 = r1.QueryInterface<IPersistFile>(IID_IPersistFile);
 211             }
 212             catch(const sal::systools::ComError& ex)
 213             {
 214                 return;
 215             }
 216             CPPUNIT_ASSERT_MESSAGE("Exception should have been thrown", false);
 217         }
 218
 219         // Change the following lines only, if you add, remove or rename
 220         // member functions of the current class,
 221         // because these macros are need by auto register mechanism.
 222
 223         CPPUNIT_TEST_SUITE(test_COMReference);
 224         CPPUNIT_TEST(default_ctor);
 225         CPPUNIT_TEST(test_ctor_manual_AddRef);
 226         CPPUNIT_TEST(test_copy_ctor);
 227         CPPUNIT_TEST(test_copy_assignment);
 228         CPPUNIT_TEST(test_ref_to_ref_assignment);
 229         CPPUNIT_TEST(test_pointer_to_ref_assignment);
 230         CPPUNIT_TEST(test_pointer_to_ref_assignment2);
 231         CPPUNIT_TEST(test_source_sink);
 232         CPPUNIT_TEST(test_address_operator);
 233         CPPUNIT_TEST(test_address_operator2);
 234         CPPUNIT_TEST(test_clear);
 235         CPPUNIT_TEST(test_query_interface);
 236         CPPUNIT_TEST(test_query_interface_throw);
 237         CPPUNIT_TEST_SUITE_END();
 238     };
 239
 240 // -----------------------------------------------------------------------------
 241 CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(test_comtools::test_COMReference, "test_comtools");
 242
 243 } // namespace rtl_OUString
 244
 245
 246 // this macro creates an empty function, which will called by the RegisterAllFunctions()
 247 // to let the user the possibility to also register some functions by hand.
 248 NOADDITIONAL;
 249
 250 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */