0
0
mirror of https://github.com/mongodb/mongo.git synced 2024-11-30 00:56:44 +01:00
mongodb/util/unittest.h
2009-02-18 13:42:32 -05:00

61 lines
1.8 KiB
C++

// unittest.h
/**
* Copyright (C) 2008 10gen Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License, version 3,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
namespace mongo {
/* The idea here is to let all initialization of global variables (classes inheriting from UnitTest)
complete before we run the tests -- otherwise order of initilization being arbitrary may mess
us up. The app's main() function should call runTests().
To define a unit test, inherit from this and implement run. instantiate one object for the new class
as a global.
*/
struct UnitTest {
UnitTest() {
registerTest(this);
}
virtual ~UnitTest() {}
// assert if fails
virtual void run() = 0;
static bool testsInProgress() { return running; }
private:
static vector<UnitTest*> *tests;
static bool running;
public:
static void registerTest(UnitTest *t) {
if ( tests == 0 )
tests = new vector<UnitTest*>();
tests->push_back(t);
}
static void runTests() {
running = true;
for ( vector<UnitTest*>::iterator i = tests->begin(); i != tests->end(); i++ ) {
(*i)->run();
}
running = false;
}
};
} // namespace mongo