sal/qa/systools/test_comtools.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  * This file incorporates work covered by the following license notice:
  10  *
  11  *   Licensed to the Apache Software Foundation (ASF) under one or more
  12  *   contributor license agreements. See the NOTICE file distributed
  13  *   with this work for additional information regarding copyright
  14  *   ownership. The ASF licenses this file to you under the Apache
  15  *   License, Version 2.0 (the "License"); you may not use this file
  16  *   except in compliance with the License. You may obtain a copy of
  17  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
  18  */
  19
  20 // autogenerated file with codegen.pl
  21
  22 #include <testshl/simpleheader.hxx>
  23 #include <systools/win32/comtools.hxx>
  24
  25 class COMObject : public IUnknown
  26 {
  27 public:
  28     COMObject() : ref_count_(0)
  29     {
  30     }
  31
  32     ~COMObject()
  33     {
  34     }
  35
  36     ULONG __stdcall AddRef()
  37     {
  38         ref_count_++;
  39         return ref_count_;
  40     }
  41
  42     ULONG __stdcall Release()
  43     {
  44         ULONG cnt = --ref_count_;
  45         if (cnt == 0)
  46             delete this;
  47         return cnt;
  48     }
  49
  50     HRESULT __stdcall QueryInterface(REFIID riid, LPVOID* ppv)
  51     {
  52         if (riid == IID_IUnknown)
  53         {
  54             AddRef();
  55             *ppv = this;
  56             return S_OK;
  57         }
  58         return E_NOINTERFACE;
  59     }
  60
  61     ULONG GetRefCount() const
  62     {
  63         return ref_count_;
  64     }
  65
  66 private:
  67     ULONG ref_count_;
  68 };
  69
  70 sal::systools::COMReference<IUnknown> comObjectSource()
  71 {
  72     return sal::systools::COMReference<IUnknown>(new COMObject);
  73 }
  74
  75 bool comObjectSink(sal::systools::COMReference<IUnknown> r, ULONG expectedRefCountOnReturn)
  76 {
  77     r = sal::systools::COMReference<IUnknown>();
  78     COMObject* p = reinterpret_cast<COMObject*>(r.get());
  79     if (p)
  80         return (p->GetRefCount() == expectedRefCountOnReturn);
  81     else
  82         return (0 == expectedRefCountOnReturn);
  83 }
  84
  85 void comObjectSource2(LPVOID* ppv)
  86 {
  87     COMObject* p = new COMObject;
  88     p->AddRef();
  89     *ppv = p;
  90 }
  91
  92 namespace test_comtools
  93 {
  94
  95     class test_COMReference : public CppUnit::TestFixture
  96     {
  97
  98     public:
  99         /// test of COMReference<IUnknown> r;
 100         void default_ctor()
 101         {
 102             sal::systools::COMReference<IUnknown> r;
 103             CPPUNIT_ASSERT_MESSAGE("COMReference should be empty", r.get() == NULL);
 104         }
 105
 106         void test_ctor_manual_AddRef()
 107         {
 108             COMObject* p = new COMObject;
 109             p->AddRef();
 110             sal::systools::COMReference<IUnknown> r(p, false);
 111             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 112         }
 113
 114         void test_copy_ctor()
 115         {
 116             sal::systools::COMReference<IUnknown> r(comObjectSource());
 117             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 118         }
 119
 120         void test_copy_assignment()
 121         {
 122             sal::systools::COMReference<IUnknown> r;
 123             CPPUNIT_ASSERT_MESSAGE("COMReference should be empty", r.get() == NULL);
 124
 125             r = comObjectSource();
 126             CPPUNIT_ASSERT_MESSAGE("COMReference should be empty", r.get() != NULL);
 127             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 128         }
 129
 130         void test_ref_to_ref_assignment()
 131         {
 132             sal::systools::COMReference<IUnknown> r1 = comObjectSource();
 133             sal::systools::COMReference<IUnknown> r2 = r1;
 134             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 2 is expected", reinterpret_cast<COMObject*>(r2.get())->GetRefCount() == 2);
 135         }
 136
 137         void test_pointer_to_ref_assignment()
 138         {
 139             sal::systools::COMReference<IUnknown> r;
 140             r = new COMObject;
 141             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 142         }
 143
 144         void test_pointer_to_ref_assignment2()
 145         {
 146             sal::systools::COMReference<IUnknown> r = comObjectSource();
 147             r = new COMObject;
 148             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 149         }
 150
 151         void test_source_sink()
 152         {
 153             CPPUNIT_ASSERT_MESSAGE("Wrong reference count, 0 is expected", comObjectSink(comObjectSource(), 0));
 154         }
 155
 156         void test_address_operator()
 157         {
 158             sal::systools::COMReference<IUnknown> r;
 159             comObjectSource2(reinterpret_cast<LPVOID*>(&r));
 160             CPPUNIT_ASSERT_MESSAGE("Wrong reference count, 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 161         }
 162
 163         void test_address_operator2()
 164         {
 165             sal::systools::COMReference<IUnknown> r1 = comObjectSource();
 166             sal::systools::COMReference<IUnknown> r2 = r1;
 167             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 2 is expected", reinterpret_cast<COMObject*>(r2.get())->GetRefCount() == 2);
 168             comObjectSource2(reinterpret_cast<LPVOID*>(&r1));
 169             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r1.get())->GetRefCount() == 1);
 170             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r2.get())->GetRefCount() == 1);
 171         }
 172
 173         void test_clear()
 174         {
 175             sal::systools::COMReference<IUnknown> r = comObjectSource();
 176             CPPUNIT_ASSERT_MESSAGE("Wrong reference count 1 is expected", reinterpret_cast<COMObject*>(r.get())->GetRefCount() == 1);
 177             r.clear();
 178             CPPUNIT_ASSERT_MESSAGE("Expect reference to be empty", !r.is());
 179         }
 180
 181         void test_query_interface()
 182         {
 183             try
 184             {
 185                 sal::systools::COMReference<IUnknown> r1 = comObjectSource();
 186                 sal::systools::COMReference<IUnknown> r2 = r1.QueryInterface<IUnknown>(IID_IUnknown);
 187                 CPPUNIT_ASSERT_MESSAGE("Wrong reference count, 2 is expected", reinterpret_cast<COMObject*>(r2.get())->GetRefCount() == 2);
 188             }
 189             catch(const sal::systools::ComError& ex)
 190             {
 191                 CPPUNIT_ASSERT_MESSAGE("Exception should not have been thrown", false);
 192             }
 193         }
 194
 195         void test_query_interface_throw()
 196         {
 197             try
 198             {
 199                 sal::systools::COMReference<IUnknown> r1 = comObjectSource();
 200                 sal::systools::COMReference<IPersistFile> r2 = r1.QueryInterface<IPersistFile>(IID_IPersistFile);
 201             }
 202             catch(const sal::systools::ComError& ex)
 203             {
 204                 return;
 205             }
 206             CPPUNIT_ASSERT_MESSAGE("Exception should have been thrown", false);
 207         }
 208
 209         // Change the following lines only, if you add, remove or rename
 210         // member functions of the current class,
 211         // because these macros are need by auto register mechanism.
 212
 213         CPPUNIT_TEST_SUITE(test_COMReference);
 214         CPPUNIT_TEST(default_ctor);
 215         CPPUNIT_TEST(test_ctor_manual_AddRef);
 216         CPPUNIT_TEST(test_copy_ctor);
 217         CPPUNIT_TEST(test_copy_assignment);
 218         CPPUNIT_TEST(test_ref_to_ref_assignment);
 219         CPPUNIT_TEST(test_pointer_to_ref_assignment);
 220         CPPUNIT_TEST(test_pointer_to_ref_assignment2);
 221         CPPUNIT_TEST(test_source_sink);
 222         CPPUNIT_TEST(test_address_operator);
 223         CPPUNIT_TEST(test_address_operator2);
 224         CPPUNIT_TEST(test_clear);
 225         CPPUNIT_TEST(test_query_interface);
 226         CPPUNIT_TEST(test_query_interface_throw);
 227         CPPUNIT_TEST_SUITE_END();
 228     };
 229
 230 CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(test_comtools::test_COMReference, "test_comtools");
 231
 232 } // namespace rtl_OUString
 233
 234 // this macro creates an empty function, which will called by the RegisterAllFunctions()
 235 // to let the user the possibility to also register some functions by hand.
 236 NOADDITIONAL;
 237
 238 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */