Fixed memory leak in test

Griezn · Nov 17, 2024 · 586f3d1 · 586f3d1
1 parent f2bde32
commit 586f3d1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 63 deletions.
diff --git a/benchmark/translator.py b/benchmark/translator.py
diff --git a/tests/dataTests.cpp b/tests/dataTests.cpp
@@ -152,11 +152,17 @@ TEST(DataTests, test_file_source)
     sink_t *gsink = create_generator_sink();
     sink_t *fsink = create_generator_sink();
 
-    gsink->push_next(gsink, gsource->get_next(gsource));
-    fsink->push_next(fsink, fsource->get_next(fsource));
+    data_t *next_gdata = gsource->get_next(gsource);
+    data_t *next_fdata = fsource->get_next(fsource);
+
+    gsink->push_next(gsink, next_gdata);
+    fsink->push_next(fsink, next_fdata);
 
     ASSERT_TRUE(ARR_EQ(gsink->buffer.data, fsink->buffer.data, gsink->buffer.size));
 
+    free(next_gdata);
+    free(next_fdata);
+
     free_generator_source(gsource);
     free_file_source(fsource);
     free(gsink); // not the normal because it is still the array allocated int he lib
@@ -193,6 +199,9 @@ TEST(DataTests, test_file_source_inc)
 
     ASSERT_TRUE(ARR_EQ(gsink->buffer.data + 2*increment, fsink->buffer.data, increment));
 
+    free(next_gdata);
+    free(next_fdata);
+
     free_generator_source(gsource);
     free_file_source(fsource);
     free(gsink); // not the normal because it is still the array allocated int he lib