diff options
Diffstat (limited to 'term-utils')
| -rw-r--r-- | term-utils/Makemodule.am | 5 | ||||
| -rw-r--r-- | term-utils/script.c | 21 |
2 files changed, 24 insertions, 2 deletions
diff --git a/term-utils/Makemodule.am b/term-utils/Makemodule.am index e7ac707fe..de62098aa 100644 --- a/term-utils/Makemodule.am +++ b/term-utils/Makemodule.am @@ -9,6 +9,11 @@ endif if HAVE_UTEMPTER script_LDADD += -lutempter endif + +check_PROGRAMS += test_script +test_script_SOURCES = $(script_SOURCES) +test_script_LDADD = $(script_LDADD) +test_script_CFLAGS = -DTEST_SCRIPT $(logger_CFLAGS) endif # BUILD_SCRIPT if BUILD_SCRIPTREPLAY diff --git a/term-utils/script.c b/term-utils/script.c index e0083f246..dd3b4b592 100644 --- a/term-utils/script.c +++ b/term-utils/script.c @@ -123,6 +123,23 @@ sigset_t block_mask, unblock_mask; int die; int resized; +/* + * For tests we want to be able to control time output + */ +#ifdef TEST_SCRIPT +static inline time_t script_time(time_t *t) +{ + const char *str = getenv("SCRIPT_TEST_SECOND_SINCE_EPOCH"); + time_t sec; + + if (str && sscanf(str, "%ld", &sec) == 1) + return sec; + return time(t); +} +#else /* !TEST_SCRIPT */ +# define script_time(x) time(x) +#endif + static void die_if_link(char *fn) { struct stat s; @@ -446,7 +463,7 @@ dooutput(void) { timingfd = fdopen(STDERR_FILENO, "w"); if (!qflg) { - time_t tvec = time((time_t *)NULL); + time_t tvec = script_time((time_t *)NULL); my_strftime(obuf, sizeof obuf, "%c\n", localtime(&tvec)); fprintf(fscript, _("Script started on %s"), obuf); } @@ -588,7 +605,7 @@ done(void) { if (fscript) { if (!qflg) { char buf[BUFSIZ]; - tvec = time((time_t *)NULL); + tvec = script_time((time_t *)NULL); my_strftime(buf, sizeof buf, "%c\n", localtime(&tvec)); fprintf(fscript, _("\nScript done on %s"), buf); } |