add test for add file to media

diff --git a/libs/spandisc/tests/splitter-test.c b/libs/spandisc/tests/splitter-test.c
index 53882f7032041e2b11dd03f81bbda0c4c9f2d8c8..477e9e06936df870f1af789d29bd701b5c746385 100644 (file)
--- a/libs/spandisc/tests/splitter-test.c
+++ b/libs/spandisc/tests/splitter-test.c
@@ -11,11 +11,11 @@
 void ignore_media_tests()
 {
        medium_t* srcdata = init_medium();
-       srcdata->freespace = 500;
+       srcdata->freespace = 2048;
        srcdata->filearray = malloc(sizeof(file_size_t*));
        *srcdata->filearray = malloc(sizeof(file_size_t));
        srcdata->filearray[0]->name = "too-big-file";
-       srcdata->filearray[0]->fsize = 600;
+       srcdata->filearray[0]->fsize = 4096;
        srcdata->length = 1;
 
        medium_t* ignore_medium = ignore(srcdata);
@@ -24,15 +24,43 @@ void ignore_media_tests()
        assert_int_equal(ignore_medium->length, 1);
 
        free_medium(srcdata);
+       srcdata = NULL;
        free(ignore_medium->filearray[0]);
        ignore_medium->filearray[0] = NULL;
        free_medium(ignore_medium);
+       ignore_medium = NULL;
+}
+
+void add_media_tests()
+{
+       medium_t* srcdata = init_medium();
+       srcdata->freespace = 2048;
+       srcdata->filearray = malloc(sizeof(file_size_t*));
+       *srcdata->filearray = malloc(sizeof(file_size_t));
+       srcdata->filearray[0]->name = "file";
+       srcdata->filearray[0]->fsize = 2000;
+       srcdata->filearray[0]->type = 20;
+       srcdata->length = 1;
+
+       medium_t* media = splitter(srcdata);
+
+       assert_string_equal(media->filearray[0]->name, "file");
+       assert_int_equal(media->length, 1);
+       assert_int_equal(media->freespace, 0);
+
+       free_medium(srcdata);
+       srcdata = NULL;
+       free(media->filearray[0]);
+       media->filearray[0] = NULL;
+       free_medium(media);
+       media = NULL;
 }
 
 int main()
 {
        const struct CMUnitTest tests[] = {
-               cmocka_unit_test(ignore_media_tests)
+               cmocka_unit_test(ignore_media_tests),
+               cmocka_unit_test(add_media_tests)
        };
 
        return  cmocka_run_group_tests(tests, NULL, NULL);