1 /*
2  *  Copyright (C) 2011  Regents of the University of Michigan
3  *
4  *   This program is free software: you can redistribute it and/or modify
5  *   it under the terms of the GNU General Public License as published by
6  *   the Free Software Foundation, either version 3 of the License, or
7  *   (at your option) any later version.
8  *
9  *   This program is distributed in the hope that it will be useful,
10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  *   GNU General Public License for more details.
13  *
14  *   You should have received a copy of the GNU General Public License
15  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
16  */
17 
18 #ifndef __UNITTEST_H
19 #define __UNITTEST_H
20 
21 #include <string>
22 #include <iostream>
23 
24 class UnitTest
25 {
26 protected:
27     std::string m_title;
28     int m_failures;
29     int m_testNum;
30 
31 public:
UnitTest(const char * title)32     UnitTest(const char *title) : m_title(title), m_failures(0), m_testNum(0) {;};
33     void test();
getPassCount()34     int getPassCount() {return m_testNum - m_failures;}
getFailureCount()35     int getFailureCount() {return m_failures;}
getTitle()36     const std::string getTitle() const {return m_title;}
37 };
38 
39 std::ostream &operator << (std::ostream &stream, UnitTest &test)
40 {
41     stream << test.getTitle() << " PASS: " << test.getPassCount() <<
42         "  FAIL: " << test.getFailureCount() << std::endl;
43     return stream;
44 }
45 
46 #endif
47