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