diff --git a/test/check_csv.c b/test/check_csv.c new file mode 100644 --- /dev/null +++ b/test/check_csv.c @@ -0,0 +1,143 @@ +/* + * Copyright 2010 Nathan Phillip Brink + * + * This file is a part of DistRen. + * + * DistRen is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * DistRen is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with DistRen. If not, see . + */ + +#include "common/csv.h" + +#include +#include +#include +#include +#include +#include + +struct check_correct_vals +{ + size_t row; + size_t num_cols; + char ***correct_vals; +}; + +static void **array_init(size_t num_elems, ...) +{ + va_list args; + void **array; + size_t c; + + array = malloc(sizeof(void *) * (num_elems + 1)); + if (!array) + return NULL; + + va_start(args, num_elems); + + for (c = 0; c < num_elems; c ++) + array[c] = va_arg(args, void *); + array[c] = NULL; + + va_end(args); + + return array; +} + +static int check_correct_vals(struct check_correct_vals *state, csv_row_t row) +{ + size_t c; + + fail_unless (state->num_cols == row->num_cols, + "We asked for %d columns but got %d on row %d", + state->num_cols, row->num_cols, state->row); + + for (c = 0; c < row->num_cols; c ++) + { + fail_if(strcmp(state->correct_vals[state->row][c], + row->cols[c]), + "%d:%d: Expecting %s: got %s", + state->row, c, + state->correct_vals[state->row][c], + row->cols[c] + ); + } + + state->row ++; + if (state->correct_vals[state->row]) + /* there is yet another row to process */ + return TRUE; + return FALSE; +} + +START_TEST(check_csv) +{ + list_t rows; + struct check_correct_vals vals; + + char ***correct_vals; + + correct_vals = + (char ***)array_init(4, + array_init(3, "23", "1" , "0.0"), + array_init(3, "a" , " b", "" ), + array_init(3, "f" , "df", "" ), + array_init(3, "d" , "f" , "ss" )); + fail_if(correct_vals == NULL, "array_init() returned NULL"); + + vals.num_cols = 3; + vals.correct_vals = correct_vals; + vals.row = 0; + + rows = csv_parse("23,1,0.0\n\ +a, b,,d\n\ +f,df\n\ +d,f,ss,e\n", vals.num_cols); + fail_if(rows == NULL, "csv_parse() returned NULL"); + list_traverse(rows, &vals, (list_traverse_func_t)&check_correct_vals, LIST_FRNT|LIST_SAVE|LIST_FORW); + + list_free(rows, (list_dealloc_func_t)&csv_row_free); +} +END_TEST + +static Suite *csv_suite() +{ + Suite *s; + TCase *tc; + + s = suite_create("csv"); + + tc = tcase_create("core"); + tcase_add_test(tc, check_csv); + suite_add_tcase(s, tc); + + return s; +} + +int main(int argc, char *argv[]) +{ + int number_failed; + Suite *s; + SRunner *sr; + + s = csv_suite(); + sr = srunner_create(s); + + srunner_run_all(sr, CK_NORMAL); + number_failed = srunner_ntests_failed(sr); + srunner_free(sr); + + if (number_failed) + return EXIT_FAILURE; + return EXIT_SUCCESS; +}