system/utest/runtests-utils/posix-test-main.cpp

// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// Test main for the runtests-utils test on POSIX systems (e.g., Linux and
// MacOS).

#include "runtests-utils-test-globals.h"

#include <dirent.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include <fbl/auto_call.h>

#include <runtests-utils/posix-run-test.h>
#include <unittest/unittest.h>

namespace runtests {

// Pointer to root of unique subdirectory of TMPDIR or /tmp.
static const fbl::String* TmpDirRoot = nullptr;

const char kScriptShebang[32] = "#!/bin/sh\n\n";
const RunTestFn PlatformRunTest = &PosixRunTest;

const char* TestFsRoot() {
    if (TmpDirRoot == nullptr) {
        char test_fs_template[256];
        sprintf(test_fs_template, "%s/XXXXXX",
                getenv("TMPDIR") ? getenv("TMPDIR") : "/tmp");
        TmpDirRoot = new fbl::String(mkdtemp(test_fs_template));
        printf("Root directory used for testing: %s\n", TmpDirRoot->c_str());
    }
    return TmpDirRoot->c_str();
}

// Removes the directory at |dir_path| and its contents.
void CleanUpDir(const char* dir_path) {
    struct dirent* entry;
    DIR* dp;

    dp = opendir(dir_path);
    if (dp == nullptr) {
        // File found; remove it.
        remove(dir_path);
        closedir(dp);
        return;
    }

    while ((entry = readdir(dp))) {
        // Skip "." and "..".
        if (!strcmp(entry->d_name, ".") || !strcmp(entry->d_name, "..")) {
            continue;
        }
        fbl::String sub_dir_name = JoinPath(dir_path, entry->d_name);
        CleanUpDir(sub_dir_name.c_str());
    }
    closedir(dp);

    // Directory is now empty: remove it.
    rmdir(dir_path);
}

} // namespace runtests

int main(int argc, char** argv) {
    printf("\nRoot directory of the filesystem used for testing: %s\n",
           runtests::TestFsRoot());

    auto auto_test_fs_clean_up = fbl::MakeAutoCall([&] {
        runtests::CleanUpDir(runtests::TestFsRoot());
        delete runtests::TmpDirRoot;
    });
    return unittest_run_all_tests(argc, argv) ? EXIT_SUCCESS : EXIT_FAILURE;
}