flexnbd-c/tests/unit/check_ioutil.c

#include "ioutil.h"

#include <check.h>

START_TEST( test_read_until_newline_returns_line_length_plus_null )
{
	int fds[2];
	int nread;
	char buf[5] = {0};
	pipe(fds);

	write( fds[1], "1234\n", 5 );
	
	nread = read_until_newline( fds[0], buf, 5 );

	ck_assert_int_eq( 5, nread );
}
END_TEST


START_TEST( test_read_until_newline_inserts_null )
{
	int fds[2];
	int nread;
	char buf[5] = {0};
	pipe(fds);

	write( fds[1], "1234\n", 5 );
	
	nread = read_until_newline( fds[0], buf, 5 );

	ck_assert_int_eq( '\0', buf[4] );

}
END_TEST


START_TEST( test_read_empty_line_inserts_null )
{
	int fds[2];
	int nread;
	char buf[5] = {0};
	pipe(fds);

	write( fds[1], "\n", 1 );
	nread = read_until_newline( fds[0], buf, 1 );

	ck_assert_int_eq( '\0', buf[0] );
	ck_assert_int_eq( 1, nread ); 
}
END_TEST


START_TEST( test_read_eof_returns_err )
{
	int fds[2];
	int nread;
	char buf[5] = {0};
	pipe( fds );

	close( fds[1] );
	nread = read_until_newline( fds[0], buf, 5 );

	ck_assert_int_eq( -1, nread );
}
END_TEST


START_TEST( test_read_eof_fills_line )
{
	int fds[2];
	int nread;
	char buf[5] = {0};
	pipe(fds);

	write( fds[1], "1234", 4 );
	close( fds[1] );
	nread = read_until_newline( fds[0], buf, 5 );

	ck_assert_int_eq( -1, nread );
	ck_assert_int_eq( '4', buf[3] );
}
END_TEST


START_TEST( test_read_lines_until_blankline )
{
	char **lines = NULL;
	int fds[2];
	int nlines;
	pipe( fds );

	write( fds[1], "a\nb\nc\n\n", 7 );

	nlines = read_lines_until_blankline( fds[0], 256, &lines );

	ck_assert_int_eq( 3, nlines );
}
END_TEST


Suite *ioutil_suite(void)
{
	Suite *s = suite_create("ioutil");

	TCase *tc_read_until_newline = tcase_create("read_until_newline");
	TCase *tc_read_lines_until_blankline = tcase_create("read_lines_until_blankline");

	tcase_add_test(tc_read_until_newline, test_read_until_newline_returns_line_length_plus_null);
	tcase_add_test(tc_read_until_newline, test_read_until_newline_inserts_null);
	tcase_add_test(tc_read_until_newline, test_read_empty_line_inserts_null);
	tcase_add_test(tc_read_until_newline, test_read_eof_returns_err);
	tcase_add_test(tc_read_until_newline, test_read_eof_fills_line );

	tcase_add_test(tc_read_lines_until_blankline, test_read_lines_until_blankline );

	suite_add_tcase(s, tc_read_until_newline);
	suite_add_tcase(s, tc_read_lines_until_blankline);

	return s;
}

int main(void)
{
	int number_failed;
	
	Suite *s = ioutil_suite();
	SRunner *sr = srunner_create(s);
	srunner_run_all(sr, CK_NORMAL);
	number_failed = srunner_ntests_failed(sr);
	srunner_free(sr);
	return (number_failed == 0) ? 0 : 1;
}