mirror of
https://github.com/troglobit/libite.git
synced 2026-10-01 13:32:42 +07:00
Update pidfile unit test to cover new use-case
Signed-off-by: Joachim Nilsson <troglobit@gmail.com>
This commit is contained in:
+76
-20
@@ -6,26 +6,30 @@
|
||||
|
||||
#include "check.h"
|
||||
|
||||
extern char *__progname;
|
||||
#define PRINT(args...) if (verbose) printf(args)
|
||||
|
||||
extern char *__pidfile_name;
|
||||
extern char *__pidfile_path;
|
||||
|
||||
static char *ident;
|
||||
static char *pidfile_arg;
|
||||
static char PIDFILE[42];
|
||||
static int verbose = 0;
|
||||
static struct stat before, after;
|
||||
|
||||
static void sigterm_handler(int UNUSED(signo))
|
||||
{
|
||||
if (verbose)
|
||||
printf("Exiting ...\n");
|
||||
PRINT("Exiting ...\n");
|
||||
remove(PIDFILE);
|
||||
}
|
||||
|
||||
static void sigalrm_handler(int UNUSED(signo))
|
||||
{
|
||||
pidfile(NULL);
|
||||
PRINT("Calling pidfile(%s)\n", pidfile_arg ?: "NULL");
|
||||
if (pidfile(pidfile_arg))
|
||||
err(1, "failed creating1 %s", pidfile_arg);
|
||||
if (!fexist(PIDFILE))
|
||||
err(1, "failed creating %s", PIDFILE);
|
||||
err(1, "failed creating2 %s, created %s", PIDFILE, __pidfile_name);
|
||||
stat(PIDFILE, &before);
|
||||
}
|
||||
|
||||
@@ -37,31 +41,40 @@ static int mtime()
|
||||
/* Must sleep a while here otherwise we execute too fast => no mtime change :-( */
|
||||
usleep(10000);
|
||||
|
||||
if (verbose)
|
||||
printf("Calling pidfile() again to update mtime ...\n");
|
||||
pidfile(NULL);
|
||||
PRINT("Calling pidfile() again to update mtime ...\n");
|
||||
if (pidfile(pidfile_arg))
|
||||
err(1, "failed creating3 %s", pidfile_arg);
|
||||
stat(PIDFILE, &after);
|
||||
|
||||
ret = timespec_newer(&after.st_mtim, &before.st_mtim);
|
||||
if (verbose) {
|
||||
printf("Before: %s\n", timespec2str(&before.st_mtim, buf, sizeof(buf)));
|
||||
printf("After : %s\n", timespec2str(&after.st_mtim, buf, sizeof(buf)));
|
||||
}
|
||||
PRINT("Before: %s\n", timespec2str(&before.st_mtim, buf, sizeof(buf)));
|
||||
PRINT("After : %s\n", timespec2str(&after.st_mtim, buf, sizeof(buf)));
|
||||
|
||||
return !ret;
|
||||
}
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
int test_pidfile(char *arg0)
|
||||
{
|
||||
if (argc > 1 && !strcmp(argv[1], "-v"))
|
||||
verbose = 1;
|
||||
/* pidfile() argument */
|
||||
pidfile_arg = arg0;
|
||||
|
||||
/* Override default _PATH_VARRUN in pidfile() */
|
||||
__pidfile_path = _PATH_TMP;
|
||||
snprintf(PIDFILE, sizeof(PIDFILE), "%s%s.pid", __pidfile_path, __progname);
|
||||
/* Expected PID file name */
|
||||
if (!arg0)
|
||||
snprintf(PIDFILE, sizeof(PIDFILE), "%s%s.pid", __pidfile_path, ident);
|
||||
else if (arg0[0] != '/')
|
||||
snprintf(PIDFILE, sizeof(PIDFILE), "%s%s.pid", __pidfile_path, arg0);
|
||||
else
|
||||
snprintf(PIDFILE, sizeof(PIDFILE), "%s", arg0);
|
||||
|
||||
PRINT("Verifying pidfile(%s) ... PID: %d\n", PIDFILE, getpid());
|
||||
|
||||
/* Remove any lingering files from previous test runs */
|
||||
remove(PIDFILE);
|
||||
|
||||
signal(SIGTERM, sigterm_handler);
|
||||
signal(SIGALRM, sigalrm_handler);
|
||||
|
||||
/* For signalling test */
|
||||
alarm(1);
|
||||
|
||||
return test(pidfile_poll(PIDFILE) != getpid(), "Testing pidfile_poll()")
|
||||
@@ -69,7 +82,50 @@ int main(int argc, char *argv[])
|
||||
|| test(mtime(), "Testing mtime update of followup pidfile() call")
|
||||
|| test(pidfile_signal(PIDFILE, SIGTERM), "Testing signalling ourselves")
|
||||
;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static char *progname(char *arg0)
|
||||
{
|
||||
char *nm;
|
||||
|
||||
nm = strrchr(arg0, '/');
|
||||
if (nm)
|
||||
nm++;
|
||||
else
|
||||
nm = arg0;
|
||||
|
||||
return nm;
|
||||
}
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
int ret = 0;
|
||||
size_t i;
|
||||
char *pidfiles[] = {
|
||||
NULL,
|
||||
"pidfile_test1",
|
||||
"/var/tmp/pidfile_test2.pid",
|
||||
};
|
||||
|
||||
if (argc > 1 && !strcmp(argv[1], "-v"))
|
||||
verbose = 1;
|
||||
|
||||
/* Override default _PATH_VARRUN in pidfile() */
|
||||
__pidfile_path = _PATH_TMP;
|
||||
|
||||
/* Find __progname, not all C libraries support this */
|
||||
ident = progname(argv[0]);
|
||||
|
||||
for (i = 0; i < NELEMS(pidfiles); i++)
|
||||
ret += test_pidfile(pidfiles[i]);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
/**
|
||||
* Local Variables:
|
||||
* indent-tabs-mode: t
|
||||
* c-file-style: "linux"
|
||||
* compile-command: "make pidfile && ./pidfile -v"
|
||||
* End:
|
||||
*/
|
||||
|
||||
Reference in New Issue
Block a user