JezK
Edit File: CxxTestMain.cpp
#include "tut.h" #include "tut_reporter.h" #include <string> #include <apr_general.h> #include <signal.h> #include <cstdio> #include <cstdlib> #include <cstring> #include "Utils.h" using namespace std; namespace tut { test_runner_singleton runner; } typedef tut::groupnames::const_iterator groupnames_iterator; /** All available groups. */ static tut::groupnames allGroups; /** Whether the user wants to run all test groups, or only the specified test groups. */ static enum { RUN_ALL_GROUPS, RUN_SPECIFIED_GROUPS } runMode = RUN_ALL_GROUPS; /** The test groups the user wants to run. Only meaningful if runMode == RUN_SPECIFIED_GROUPS. */ static tut::groupnames groupsToRun; static void usage(int exitCode) { printf("Usage: ./Apache2ModuleTests [options]\n"); printf("Runs the unit tests for the Apache 2 module.\n\n"); printf("Options:\n"); printf(" -g GROUP_NAME Instead of running all unit tests, only run the test group\n"); printf(" named GROUP_NAME. You can specify -g multiple times, which\n"); printf(" will result in only the specified test groups being run.\n\n"); printf(" Available test groups:\n\n"); for (groupnames_iterator it = allGroups.begin(); it != allGroups.end(); it++) { printf(" %s\n", it->c_str()); } printf("\n"); printf(" -h Print this usage information.\n"); exit(exitCode); } static bool groupExists(const string &name) { for (groupnames_iterator it = allGroups.begin(); it != allGroups.end(); it++) { if (name == *it) { return true; } } return false; } static void parseOptions(int argc, char *argv[]) { for (int i = 1; i < argc; i++) { if (strcmp(argv[i], "-h") == 0) { usage(0); } else if (strcmp(argv[i], "-g") == 0) { if (argv[i + 1] == NULL) { fprintf(stderr, "*** ERROR: A -g option must be followed by a test group name.\n"); exit(1); } else if (!groupExists(argv[i + 1])) { fprintf(stderr, "*** ERROR: Invalid test group '%s'. Available test groups are:\n\n", argv[i + 1]); for (groupnames_iterator it = allGroups.begin(); it != allGroups.end(); it++) { printf("%s\n", it->c_str()); } exit(1); } else { runMode = RUN_SPECIFIED_GROUPS; groupsToRun.push_back(argv[i + 1]); i++; } } else { fprintf(stderr, "*** ERROR: Unknown option: %s\n", argv[i]); fprintf(stderr, "Please pass -h for a list of valid options.\n"); exit(1); } } } /** * Creates a Phusion Passenger temp dir at the beginning, and deletes it * at program exit. */ struct TempDirSetup { TempDirSetup() { char command[1024]; Passenger::createPassengerTempDir("", geteuid() == 0, "nobody", geteuid(), getgid()); snprintf(command, sizeof(command), "chmod -R u=rwx,g=rwx,o=rwx \"%s\"", Passenger::getPassengerTempDir().c_str()); system(command); } ~TempDirSetup() { Passenger::removeDirTree(Passenger::getPassengerTempDir()); } }; int main(int argc, char *argv[]) { apr_initialize(); signal(SIGPIPE, SIG_IGN); setenv("RAILS_ENV", "production", 1); setenv("TESTING_PASSENGER", "1", 1); tut::reporter reporter; tut::runner.get().set_callback(&reporter); allGroups = tut::runner.get().list_groups(); parseOptions(argc, argv); TempDirSetup ts; try { bool all_ok = true; if (runMode == RUN_ALL_GROUPS) { tut::runner.get().run_tests(); all_ok = reporter.all_ok(); } else { all_ok = true; for (groupnames_iterator it = groupsToRun.begin(); it != groupsToRun.end(); it++) { tut::runner.get().run_tests(*it); all_ok = all_ok && reporter.all_ok(); } } if (all_ok) { return 0; } else { return 1; } } catch (const std::exception &ex) { cerr << "*** Exception raised: " << ex.what() << endl; return 2; } }