]>
Commit | Line | Data |
---|---|---|
1 | #define _POSIX_C_SOURCE 2 | |
2 | #include <errno.h> | |
3 | #include <stdio.h> | |
4 | #include <stdlib.h> | |
5 | #include <string.h> | |
6 | #include <time.h> | |
7 | ||
8 | typedef struct expectation { | |
9 | time_t a, b; | |
10 | const char *message; | |
11 | } ex_t; | |
12 | ||
13 | const ex_t END = {((time_t)-1), ((time_t)-1), NULL}; | |
14 | static int is_end(ex_t exp) { | |
15 | return exp.a == END.a && exp.b == END.b && exp.message == END.message; | |
16 | } | |
17 | static ex_t expectation(time_t a, time_t b, const char *message) { | |
18 | ex_t exp; | |
19 | exp.a = a; | |
20 | exp.b = b; | |
21 | exp.message = message; | |
22 | return exp; | |
23 | } | |
24 | ||
25 | static void die(const char *message) { | |
26 | fputs(message, stderr); | |
27 | fputc('\n', stderr); | |
28 | exit(1); | |
29 | } | |
30 | ||
31 | static void die_err(const char *message) { | |
32 | perror(message); | |
33 | exit(1); | |
34 | } | |
35 | ||
36 | static void remove_logfile() { | |
37 | if (remove("tl.log") != 0) { | |
38 | if (errno != ENOENT) { | |
39 | die_err("Error removing log file"); | |
40 | } | |
41 | } | |
42 | } | |
43 | ||
44 | static ex_t write_to_tl_append(const char *content) { | |
45 | FILE *p = popen("./tl-append", "w"); | |
46 | if (p == NULL) | |
47 | die_err("Couldn't run tl-append"); | |
48 | time_t start = time(NULL); | |
49 | if (fputs(content, p) == EOF) | |
50 | die("Couldn't write to pipe"); | |
51 | int status = pclose(p); | |
52 | time_t end = time(NULL); | |
53 | if (status < 0) | |
54 | die_err("Error closing pipe"); | |
55 | if (status != 0) | |
56 | die("tl-append exited abnormally"); | |
57 | return expectation(start, end, content); | |
58 | } | |
59 | ||
60 | static void verify_log_contents(ex_t exps[]) { | |
61 | ||
62 | FILE *f = fopen("tl.log", "r"); | |
63 | if (f == NULL) | |
64 | die_err("Error opening log file"); | |
65 | for (size_t i = 0; !is_end(exps[i]); i++) { | |
66 | size_t len = strlen(exps[i].message); | |
67 | char *buf = (char *)malloc(len + 2); | |
68 | if (fgets(buf, len + 1, f) == NULL) | |
69 | die("Error reading log file"); | |
70 | if (ferror(f)) | |
71 | die("Error reading log file"); | |
72 | if (strncmp(exps[i].message, buf, len + 1) != 0) | |
73 | die("Wrong contents in log file"); | |
74 | free(buf); | |
75 | } | |
76 | if (fclose(f) != 0) | |
77 | die_err("Error closing log file"); | |
78 | } | |
79 | ||
80 | static void write_and_read_line() { | |
81 | remove_logfile(); | |
82 | ex_t e = write_to_tl_append("foo\n"); | |
83 | verify_log_contents((ex_t[]){e, END}); | |
84 | } | |
85 | ||
86 | static void write_and_read_two_lines() { | |
87 | remove_logfile(); | |
88 | ex_t e1 = write_to_tl_append("foo\n"); | |
89 | ex_t e2 = write_to_tl_append("bar\n"); | |
90 | verify_log_contents((ex_t[]){e1, e2, END}); | |
91 | } | |
92 | ||
93 | int main() { | |
94 | write_and_read_line(); | |
95 | write_and_read_two_lines(); | |
96 | } |