Viewing file:  CxxTestMain.cpp (5.88 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
#include <TestSupport.h>
#include "../tut/tut_reporter.h"
#include <oxt/initialize.hpp>
#include <oxt/system_calls.hpp>
#include <string>
#include <map>
#include <vector>
#include <signal.h>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <unistd.h>
#include <Shared/Fundamentals/Initialization.h>
#include <Shared/Fundamentals/AbortHandler.h>
#include <Shared/Fundamentals/Utils.h>
#include <ConfigKit/ConfigKit.h>
#include <LoggingKit/LoggingKit.h>
#include <FileTools/PathManip.h>
#include <Utils.h>
#include <SystemTools/SystemTime.h>
#include <StrIntTools/StrIntUtils.h>
#include <jsoncpp/json.h>
using namespace std;
using namespace Passenger;
using namespace TestSupport;
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. */
enum RunMode { RUN_ALL_GROUPS, RUN_SPECIFIED_GROUPS };
static RunMode runMode = RUN_ALL_GROUPS;
/** The test groups and test numbers that the user wants to run.
* Only meaningful if runMode == RUN_SPECIFIED_GROUPS.
*/
static map< string, vector<int> > groupsToRun;
static ConfigKit::Schema *
createSchema() {
using namespace ConfigKit;
ConfigKit::Schema *schema = new ConfigKit::Schema();
schema->add("passenger_root", STRING_TYPE, REQUIRED);
schema->finalize();
return schema;
}
static void
usage(int exitCode) {
printf("Usage: ./CxxTestMain [options]\n");
printf("Runs the cxx unit tests.\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(" -l LEVEL Log level to use (default: %s). Available levels:\n", DEFAULT_LOG_LEVEL_NAME);
printf(" crit,error,warn,notice,info,debug,debug2,debug3\n");
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
parseGroupSpec(const char *spec, string &groupName, vector<int> &testNumbers) {
testNumbers.clear();
if (*spec == '\0') {
groupName = "";
return;
}
vector<string> components;
split(spec, ':', components);
groupName = components[0];
if (components.size() > 1) {
string testNumbersSpec = components[1];
components.clear();
split(testNumbersSpec, ',', components);
vector<string>::const_iterator it;
for (it = components.begin(); it != components.end(); it++) {
testNumbers.push_back(atoi(*it));
}
}
}
static void
parseOptions(int argc, const char *argv[], ConfigKit::Store &config) {
Json::Value updates;
char path[PATH_MAX + 1];
getcwd(path, PATH_MAX);
updates["passenger_root"] = extractDirName(path);
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);
}
string groupName;
vector<int> testNumbers;
parseGroupSpec(argv[i + 1], groupName, testNumbers);
if (!groupExists(groupName)) {
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[groupName] = testNumbers;
i++;
}
} else if (strcmp(argv[i], "-l") == 0) {
if (argv[i + 1] == NULL) {
fprintf(stderr, "*** ERROR: A -l option must be followed by a log level name.\n");
exit(1);
}
TestSupport::defaultLogLevel = LoggingKit::parseLevel(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);
}
}
vector<ConfigKit::Error> errors;
if (!config.update(updates, errors)) {
P_BUG("Unable to set initial configuration: " <<
ConfigKit::toString(errors));
}
}
static void
loadConfigFile() {
Json::Reader reader;
if (!reader.parse(unsafeReadFile("config.json"), testConfig)) {
fprintf(stderr, "Cannot parse config.json: %s\n",
reader.getFormattedErrorMessages().c_str());
exit(1);
}
}
int
main(int argc, char *argv[]) {
using namespace Agent::Fundamentals;
setenv("RAILS_ENV", "production", 1);
setenv("TESTING_PASSENGER", "1", 1);
setenv("PYTHONDONTWRITEBYTECODE", "1", 1);
unsetenv("PASSENGER_USE_FEEDBACK_FD");
unsetenv("TMPDIR");
if (getEnvBool("GDB", false) || getEnvBool("LLDB", false)) {
setenv("PASSENGER_ABORT_HANDLER", "false", 1);
}
tut::reporter reporter;
tut::runner.get().set_callback(&reporter);
allGroups = tut::runner.get().list_groups();
ConfigKit::Schema *schema = createSchema();
ConfigKit::Store *config = new ConfigKit::Store(*schema);
initializeAgent(argc, &argv, "CxxTestMain", *config,
ConfigKit::DummyTranslator(), parseOptions);
resourceLocator = Agent::Fundamentals::context->resourceLocator;
loadConfigFile();
bool all_ok = true;
if (runMode == RUN_ALL_GROUPS) {
tut::runner.get().run_tests();
} else {
try {
tut::runner.get().run_tests(groupsToRun);
} catch (const tut::no_such_test &) {
fprintf(stderr, "ERROR: test not found.\n");
all_ok = false;
}
}
all_ok = reporter.all_ok();
shutdownAgent(schema, config);
if (all_ok) {
return 0;
} else {
return 1;
}
}
|