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