Criterion/samples/redirect.c

52 lines
1.1 KiB
C

#include <criterion/criterion.h>
#include <criterion/redirect.h>
#include <stdio.h>
#include <ctype.h>
// Testing stdout/stderr
void redirect_all_std(void) {
cr_redirect_stdout();
cr_redirect_stderr();
}
Test(redirect, test_outputs, .init = redirect_all_std) {
fprintf(stdout, "foo");
fflush(stdout);
cr_assert_stdout_eq_str("foo");
fprintf(stderr, "bar");
fflush(stderr);
cr_assert_stderr_eq_str("bar");
}
// Testing general I/O with sample command-line rot13
char rot13_char(char c) {
return isalpha(c) ? (c - 'a' + 13) % 26 + 'a' : c;
}
void rot13_io(void) {
char buf[512];
size_t read;
while ((read = fread(buf, 1, sizeof (buf), stdin)) > 0) {
for (size_t i = 0; i < read; ++i)
buf[i] = rot13_char(buf[i]);
fwrite(buf, 1, read, stdout);
}
fflush(stdout);
}
Test(redirect, rot13, .init = cr_redirect_stdout) {
FILE* f_stdin = cr_get_redirected_stdin();
fprintf(f_stdin, "the quick brown fox jumps over the lazy dog");
fclose(f_stdin);
rot13_io();
cr_assert_stdout_eq_str("gur dhvpx oebja sbk whzcf bire gur ynml qbt");
}