mirror of
https://github.com/pevans/erc-c.git
synced 2024-10-31 13:07:14 +00:00
149 lines
3.1 KiB
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);
|
|
}
|