diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/Makefile.am | 8 | ||||
| -rw-r--r-- | tests/main.c | 215 |
2 files changed, 223 insertions, 0 deletions
diff --git a/tests/Makefile.am b/tests/Makefile.am new file mode 100644 index 0000000..2b70ec5 --- /dev/null +++ b/tests/Makefile.am @@ -0,0 +1,8 @@ +AM_CPPFLAGS = -I$(top_srcdir)/src + +bin_PROGRAMS = test + +test_LDFLAGS = -lcheck +test_SOURCES = ../src/csv_parser.c \ + ../src/csv_parser.h \ + main.c diff --git a/tests/main.c b/tests/main.c new file mode 100644 index 0000000..685842f --- /dev/null +++ b/tests/main.c @@ -0,0 +1,215 @@ +#include <check.h> +#include <stdlib.h> +#include <stdio.h> +#include <string.h> +#include <csv_parser.h> + + +struct test_results { + int test_case_number; + int *results; +}; + + +// 5 items per row, 3 rows per test case: 15 * 6 slots +static int expected_test_results[] = { + // "" + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + // ", " + 0, 1, -1, -1, -1, + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + // "," + 0, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + // "\n" + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + // "\r\n", + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + // "a,b,,c," + 1, 2, 0, 1, -1, + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + // "a,b,,c, " + 1, 2, 0, 1, 1, + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + // "a,b,\na,b," + 1, 2, 0, -1, -1, + 1, 1, -1, -1, -1, + -1, -1, -1, -1, -1, + // " bc " + 4, -1, -1, -1, -1, + -1, -1, -1, -1, -1, + -1, -1, -1, -1, -1, +}; + + +static const int results_size = sizeof(expected_test_results); +static const int result_slots = sizeof(expected_test_results)/sizeof(int); + + +static const char *test_data[] = { + "", + ", ", + ",", + "\n", + "\r\n", + "a,bb,,c,", + "a,bb,,c, ", + "a,bb,\na,b,", + " bc ", +}; + + +static int test_data_end_states[] = { + csvps_line_start, + csvps_field_value, + csvps_field_start, + csvps_line_end, + csvps_line_end, + csvps_field_start, + csvps_field_value, + csvps_field_start, + csvps_field_value, +}; + +static int num_test_cases = sizeof(test_data)/(sizeof(char*)); + + +void init_test_results(int *data, int len) +{ + int i; + for (i = 0; i < len; i++) { + data[i] = -1; + } +} + + +void print_results(int *data, int len) +{ + int i; + for (i = 0; i < len; i++) { + if (i != 0 && i % 5 == 0) { + printf("\n"); + } + printf("%4d ", data[i]); + } + + printf("\n"); +} + + +int field_cb(csv_parser_t *parser, const char *data, size_t length, int row, int col) +{ + struct test_results *results = parser->data; + int slot = results->test_case_number * 15 + row * 5 + col; + + if (results->results[slot] == -1) { + results->results[slot] = length; + } else { + results->results[slot] += length; + } + return 0; +} + + +START_TEST(test_parser_full) +{ + csv_parser_t parser; + csv_parser_settings_t settings; + struct test_results results; + int i; + + settings.delimiter = ','; + settings.field_cb = field_cb; + + results.results = malloc(results_size); + init_test_results(results.results, result_slots); + for (i = 0; i < num_test_cases; i++) { + const char *str = test_data[i]; + int len = strlen(str); + + results.test_case_number = i; + + csv_parser_init(&parser); + parser.data = &results; + + int nread = csv_parser_execute(&parser, &settings, str, len); + ck_assert_int_eq(len, nread); + ck_assert_int_eq(parser.state, test_data_end_states[i]); + } + + printf("Test results:\n"); + print_results(results.results, result_slots); + ck_assert_int_eq(0, memcmp(results.results, expected_test_results, results_size)); + free(results.results); + +} +END_TEST + + +START_TEST(test_parser_chunked) +{ + csv_parser_t parser; + csv_parser_settings_t settings; + struct test_results results; + int i, offset; + + settings.delimiter = ','; + settings.field_cb = field_cb; + + results.results = malloc(results_size); + init_test_results(results.results, result_slots); + for (i = 0; i < num_test_cases; i++) { + const char *str = test_data[i]; + int len = strlen(str); + + results.test_case_number = i; + + csv_parser_init(&parser); + parser.data = &results; + + // parse string one char at a time + for (offset = 0; offset < len; offset++) { + char c = *(str+offset); + int nread = csv_parser_execute(&parser, &settings, &c, 1); + ck_assert_int_eq(1, nread); + } + ck_assert_int_eq(parser.state, test_data_end_states[i]); + } + + printf("Test results:\n"); + print_results(results.results, result_slots); + ck_assert_int_eq(0, memcmp(results.results, expected_test_results, results_size)); + free(results.results); + +} +END_TEST + + +int main(int argc, const char *argv[]) +{ + Suite *s = suite_create("csv_parser"); + TCase *tc = tcase_create("parser"); + tcase_add_test(tc, test_parser_full); + tcase_add_test(tc, test_parser_chunked); + + suite_add_tcase(s, tc); + SRunner *sr = srunner_create(s); + + printf("Expected results:\n"); + print_results(expected_test_results, result_slots); + + srunner_run_all(sr, CK_VERBOSE); + srunner_free(sr); + + return 0; +} |