test-suite/MemPoolTest.cc

   1 /*
   2  * $Id$
   3  *
   4  * AUTHOR: Robert Collins
   5  *
   6  * SQUID Web Proxy Cache          http://www.squid-cache.org/
   7  * ----------------------------------------------------------
   8  *
   9  *  Squid is the result of efforts by numerous individuals from
  10  *  the Internet community; see the CONTRIBUTORS file for full
  11  *  details.   Many organizations have provided support for Squid's
  12  *  development; see the SPONSORS file for full details.  Squid is
  13  *  Copyrighted (C) 2001 by the Regents of the University of
  14  *  California; see the COPYRIGHT file for full details.  Squid
  15  *  incorporates software developed and/or copyrighted by other
  16  *  sources; see the CREDITS file for full details.
  17  *
  18  *  This program is free software; you can redistribute it and/or modify
  19  *  it under the terms of the GNU General Public License as published by
  20  *  the Free Software Foundation; either version 2 of the License, or
  21  *  (at your option) any later version.
  22  *
  23  *  This program is distributed in the hope that it will be useful,
  24  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  25  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  26  *  GNU General Public License for more details.
  27  *
  28  *  You should have received a copy of the GNU General Public License
  29  *  along with this program; if not, write to the Free Software
  30  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
  31  *
  32  * Copyright (c) 2003, Robert Collins <robertc@squid-cache.org>
  33  */
  34
  35 #include "squid.h"
  36
  37 #if !DISABLE_POOLS
  38
  39 #include "MemPool.h"
  40
  41 #if HAVE_IOSTREAM
  42 #include <iostream>
  43 #endif
  44
  45 /* TODO: put this in a libTest */
  46 void
  47 xassert(const char *msg, const char *file, int line)
  48 {
  49     std::cout << "Assertion failed: (" << msg << ") at " << file << ":" << line << std::endl;
  50     exit (1);
  51 }
  52
  53 class MemPoolTest
  54 {
  55 public:
  56     void run();
  57 private:
  58     class SomethingToAlloc
  59     {
  60     public:
  61         int aValue;
  62     };
  63     static MemAllocator *Pool;
  64 };
  65 MemAllocator *MemPoolTest::Pool = NULL;
  66
  67 void
  68 MemPoolTest::run()
  69 {
  70     assert (Pool == NULL);
  71     Pool = memPoolCreate("Test Pool", sizeof(SomethingToAlloc));
  72     assert (Pool);
  73     SomethingToAlloc *something = static_cast<SomethingToAlloc *>(Pool->alloc());
  74     assert (something);
  75     assert (something->aValue == 0);
  76     something->aValue = 5;
  77     Pool->freeOne(something);
  78     SomethingToAlloc *otherthing = static_cast<SomethingToAlloc *>(Pool->alloc());
  79     assert (otherthing == something);
  80     assert (otherthing->aValue == 0);
  81     Pool->freeOne(otherthing);
  82     delete Pool;
  83 }
  84
  85 #endif /* DISABLE_POOLS */
  86
  87 int
  88 main (int argc, char **argv)
  89 {
  90 #if !DISABLE_POOLS
  91     MemPoolTest aTest;
  92     aTest.run();
  93 #endif
  94     return 0;
  95 }
  96