NetBurner 3.1
MemoryReporterPlugin.h
1 /*
2  * Copyright (c) 2007, Michael Feathers, James Grenning and Bas Vodde
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are met:
7  * * Redistributions of source code must retain the above copyright
8  * notice, this list of conditions and the following disclaimer.
9  * * Redistributions in binary form must reproduce the above copyright
10  * notice, this list of conditions and the following disclaimer in the
11  * documentation and/or other materials provided with the distribution.
12  * * Neither the name of the <organization> nor the
13  * names of its contributors may be used to endorse or promote products
14  * derived from this software without specific prior written permission.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE EARLIER MENTIONED AUTHORS ``AS IS'' AND ANY
17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19  * DISCLAIMED. IN NO EVENT SHALL <copyright holder> BE LIABLE FOR ANY
20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27 
28 #ifndef D_MemoryReporterPlugin_h
29 #define D_MemoryReporterPlugin_h
30 
31 #include "CppUTest/TestPlugin.h"
32 #include "CppUTestExt/MemoryReportAllocator.h"
33 
34 class MemoryReportFormatter;
35 
36 class MemoryReporterPlugin : public TestPlugin
37 {
38  MemoryReportFormatter* formatter_;
39 
40  MemoryReportAllocator mallocAllocator;
41  MemoryReportAllocator newAllocator;
42  MemoryReportAllocator newArrayAllocator;
43 
44  SimpleString currentTestGroup_;
45 public:
46  MemoryReporterPlugin();
47  virtual ~MemoryReporterPlugin();
48 
49  virtual void preTestAction(UtestShell & test, TestResult & result) _override;
50  virtual void postTestAction(UtestShell & test, TestResult & result) _override;
51  virtual bool parseArguments(int, const char**, int) _override;
52 
53 protected:
54  virtual MemoryReportFormatter* createMemoryFormatter(const SimpleString& type);
55 
56 private:
57  void destroyMemoryFormatter(MemoryReportFormatter* formatter);
58 
59  void setGlobalMemoryReportAllocators();
60  void removeGlobalMemoryReportAllocators();
61 
62  void initializeAllocator(MemoryReportAllocator* allocator, TestResult & result);
63 };
64 
65 #endif