mirror of https://github.com/djcb/mu.git
150 lines
3.1 KiB
C++
150 lines
3.1 KiB
C++
/*
|
|
** Copyright (C) 2008-2023 Dirk-Jan C. Binnema <djcb@djcbsoftware.nl>
|
|
**
|
|
** This program is free software; you can redistribute it and/or modify it
|
|
** under the terms of the GNU General Public License as published by the
|
|
** Free Software Foundation; either version 3, or (at your option) any
|
|
** later version.
|
|
**
|
|
** 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 General Public License for more details.
|
|
**
|
|
** You should have received a copy of the GNU General Public License
|
|
** along with this program; if not, write to the Free Software Foundation,
|
|
** Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
**
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <glib.h>
|
|
#include <glib/gstdio.h>
|
|
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
|
|
#include <langinfo.h>
|
|
#include <locale.h>
|
|
|
|
#include "utils/mu-utils.hh"
|
|
#include "utils/mu-test-utils.hh"
|
|
#include "utils/mu-utils-file.hh"
|
|
#include "utils/mu-error.hh"
|
|
|
|
|
|
using namespace Mu;
|
|
|
|
std::string
|
|
Mu::test_random_tmpdir()
|
|
{
|
|
auto&& dir = mu_format("{}{}mu-test-{}{}test-{:x}",
|
|
g_get_tmp_dir(),
|
|
G_DIR_SEPARATOR,
|
|
getuid(),
|
|
G_DIR_SEPARATOR,
|
|
::random() * getpid() * ::time({}));
|
|
|
|
auto res = g_mkdir_with_parents(dir.c_str(), 0700);
|
|
g_assert(res != -1);
|
|
|
|
return dir;
|
|
}
|
|
|
|
const char*
|
|
Mu::set_tz(const char* tz)
|
|
{
|
|
static const char* oldtz;
|
|
|
|
oldtz = getenv("TZ");
|
|
if (tz)
|
|
setenv("TZ", tz, 1);
|
|
else
|
|
unsetenv("TZ");
|
|
|
|
tzset();
|
|
return oldtz;
|
|
}
|
|
|
|
bool
|
|
Mu::set_en_us_utf8_locale()
|
|
{
|
|
setenv("LC_ALL", "en_US.UTF-8", 1);
|
|
setlocale(LC_ALL, "en_US.UTF-8");
|
|
|
|
if (strcmp(nl_langinfo(CODESET), "UTF-8") != 0) {
|
|
mu_println("Note: Unit tests require the en_US.utf8 locale. "
|
|
"Ignoring test cases.");
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
static void
|
|
black_hole(void)
|
|
{
|
|
return; /* do nothing */
|
|
}
|
|
|
|
void
|
|
Mu::mu_test_init(int *argc, char ***argv)
|
|
{
|
|
const auto tmpdir{test_random_tmpdir()};
|
|
|
|
g_setenv("MU_TEST", "yes", TRUE);
|
|
g_setenv("XDG_CACHE_HOME", tmpdir.c_str(), TRUE);
|
|
|
|
g_test_init(argc, argv, NULL);
|
|
|
|
g_test_bug_base("https://github.com/djcb/mu/issues/");
|
|
|
|
if (!g_test_verbose())
|
|
g_log_set_handler(
|
|
NULL,
|
|
(GLogLevelFlags)(G_LOG_LEVEL_MASK |
|
|
G_LOG_FLAG_FATAL | G_LOG_FLAG_RECURSION),
|
|
(GLogFunc)black_hole, NULL);
|
|
}
|
|
|
|
void
|
|
Mu::allow_warnings()
|
|
{
|
|
g_test_log_set_fatal_handler(
|
|
[](const char*, GLogLevelFlags, const char*, gpointer) { return FALSE; },
|
|
{});
|
|
}
|
|
|
|
Mu::TempDir::TempDir(bool autodelete): autodelete_{autodelete} {
|
|
|
|
if (auto res{make_temp_dir()}; !res)
|
|
throw res.error();
|
|
else
|
|
path_ = std::move(*res);
|
|
|
|
mu_debug("created '{}'", path_);
|
|
}
|
|
|
|
Mu::TempDir::~TempDir()
|
|
{
|
|
if (::access(path_.c_str(), F_OK) != 0)
|
|
return; /* nothing to do */
|
|
|
|
if (!autodelete_) {
|
|
mu_debug("_not_ deleting {}", path_);
|
|
return;
|
|
}
|
|
|
|
/* ugly */
|
|
GError *err{};
|
|
const auto cmd{mu_format("/bin/rm -rf '{}'", path_)};
|
|
if (!g_spawn_command_line_sync(cmd.c_str(), NULL,
|
|
NULL, NULL, &err)) {
|
|
mu_warning("error: {}", err ? err->message : "?");
|
|
g_clear_error(&err);
|
|
} else
|
|
mu_debug("removed '{}'", path_);
|
|
}
|