Viewing file:  TestSupport.cpp (2.99 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
#include <TestSupport.h>
#include "../support/valgrind.h"
#include <oxt/backtrace.hpp>
#include <sys/types.h>
#include <dirent.h>
#include <unistd.h>
#include <fcntl.h>
#include <pwd.h>
#include <grp.h>
#include <cassert>
#include <FileTools/FileManip.h>
#include <SystemTools/UserDatabase.h>
#include <Utils/ScopeGuard.h>
#include <jsoncpp/json.h>
namespace TestSupport {
LoggingKit::Level defaultLogLevel = (LoggingKit::Level) DEFAULT_LOG_LEVEL;
ResourceLocator *resourceLocator = NULL;
Json::Value testConfig;
void
createInstanceDir(InstanceDirectoryPtr &instanceDir) {
InstanceDirectory::CreationOptions options;
OsUser osUser;
if (!lookupSystemUserByName("nobody", osUser)) {
throw RuntimeException("OS user account 'nobody' does not exist");
}
options.prefix = "passenger-test";
options.userSwitching = geteuid() == 0;
options.defaultUid = osUser.pwd.pw_uid;
options.defaultGid = osUser.pwd.pw_gid;
instanceDir = boost::make_shared<InstanceDirectory>(options);
}
void
writeUntilFull(int fd) {
int flags, ret;
char buf[1024];
memset(buf, 0, sizeof(buf));
flags = fcntl(fd, F_GETFL);
fcntl(fd, F_SETFL, flags | O_NONBLOCK);
while (true) {
ret = write(fd, buf, sizeof(buf));
if (ret == -1) {
int e = errno;
if (e == EAGAIN) {
break;
} else {
throw SystemException("write() failed", e);
}
}
}
while (true) {
ret = write(fd, buf, 50);
if (ret == -1) {
int e = errno;
if (e == EAGAIN) {
break;
} else {
throw SystemException("write() failed", e);
}
}
}
while (true) {
ret = write(fd, buf, 1);
if (ret == -1) {
int e = errno;
if (e == EAGAIN) {
break;
} else {
throw SystemException("write() failed", e);
}
}
}
fcntl(fd, F_SETFL, flags);
}
bool
containsSubstring(const StaticString &str, const StaticString &substr) {
return str.find(substr) != string::npos;
}
void
writeFile(const string &filename, const string &contents) {
createFile(filename, contents);
}
void
touchFile(const char *filename, time_t timestamp) {
FILE *f = fopen(filename, "a");
if (f != NULL) {
fclose(f);
} else if (errno != EISDIR) {
int e = errno;
string message = "Cannot touch file '";
message.append(filename);
message.append("'");
throw FileSystemException(message, e, filename);
}
if (timestamp != (time_t) -1) {
struct utimbuf times;
times.actime = timestamp;
times.modtime = timestamp;
utime(filename, ×);
}
}
string
getPrimaryGroupName(const string &username) {
OsUser osUser;
if (lookupSystemUserByName(username, osUser)) {
OsGroup osGroup;
if (lookupSystemGroupByGid(osUser.pwd.pw_gid, osGroup)) {
return osGroup.grp.gr_name;
} else {
throw RuntimeException("OS group account with GID "
+ toString(osUser.pwd.pw_gid) + " does not exist");
}
} else {
throw RuntimeException("OS user account " + username + " does not exist");
}
}
// Shared test setup code.
TestBase::TestBase() {
if (LoggingKit::getLevel() != defaultLogLevel) {
LoggingKit::setLevel(defaultLogLevel);
}
}
} // namespace TestSupport
|