1
0
mirror of https://github.com/pevans/erc-c.git synced 2024-10-31 13:07:14 +00:00
erc-c/tests/option.c

149 lines
3.1 KiB
C

#include <criterion/criterion.h>
#include <unistd.h>
#include "option.h"
static void
setup()
{
option_set_error("");
option_set_input(1, NULL);
option_set_input(2, NULL);
}
static void
teardown()
{
FILE *stream;
for (int i = 1; i <= OPTION_MAX_DISKS; i++) {
stream = option_get_input(i);
if (stream
&& stream != stdout
&& stream != stderr
&& stream != stdin
) {
fclose(stream);
}
}
}
TestSuite(options, .init = setup, .fini = teardown);
// No need to do this one...
/* Test(option, print_help) */
/* Test(option, get_error) */
/* Test(option, set_error) */
Test(option, error)
{
char *str = "hahaha FUN";
cr_assert_str_empty(option_get_error());
option_set_error(str);
cr_assert_str_eq(option_get_error(), str);
}
/* Test(option, get_input) */
/* Test(option, set_input) */
Test(option, input)
{
cr_assert_eq(option_get_input(1), NULL);
cr_assert_eq(option_get_input(2), NULL);
option_set_input(2, stdout);
cr_assert_eq(option_get_input(2), stdout);
option_set_input(3, stderr);
cr_assert_eq(option_get_input(3), NULL);
}
Test(option, read_file)
{
char *str = "so much FUN";
char *bad_file = "/tmp/BLEH";
char *file = "/tmp/erc-test.txt";
char buf[256];
cr_assert_eq(option_get_input(1), NULL);
// Maybe we should use sterror(ENOENT)?
cr_assert_eq(option_read_file(1, bad_file), 0);
cr_assert_str_eq(option_get_error(), "--disk1: No such file or directory");
option_set_error("");
FILE *stream;
stream = fopen(file, "w");
cr_assert_neq(stream, NULL);
fwrite(str, sizeof(char), strlen(str), stream);
fclose(stream);
option_read_file(1, file);
fread(buf, sizeof(char), 255, option_get_input(1));
cr_assert_str_eq(buf, str);
unlink(file);
}
/*
* This test is really imperfect... that's because option_parse() does
* a ton of stuff, and is quite complex. I'm punting a lot on the
* complexity here, while pushing as much of the logic as I can into
* other functions that are more easily testable.
*/
Test(option, parse)
{
int argc = 2;
char *argv[] = {
"prog_name",
"--disk1=etc",
};
cr_assert_eq(option_parse(argc, argv), 0);
}
/*
* The get_width and get_height tests also implicitly test the
* option_set_size() function (which is called by option_parse()).
* Test(option, set_size)
*/
Test(option, get_width)
{
int argc = 2;
char *argv[] = {
"prog_name",
"--size=875x600",
};
cr_assert_eq(option_parse(argc, argv), 1);
cr_assert_eq(option_get_width(), 875);
}
Test(option, get_height)
{
int argc = 2;
char *argv[] = {
"prog_name",
"--size=875x600",
};
cr_assert_eq(option_parse(argc, argv), 1);
cr_assert_eq(option_get_height(), 600);
}
Test(option, flag)
{
int argc = 2;
char *argv[] = {
"prog_name",
"--flash",
};
cr_assert_eq(option_flag(OPTION_FLASH), false);
cr_assert_eq(option_parse(argc, argv), 1);
cr_assert_eq(option_flag(OPTION_FLASH), true);
}