From ce4b94ca443909a31a28e4f127dac0348bbedec1 Mon Sep 17 00:00:00 2001
From: Daniel Silverstone <dsilvers@digital-scurf.org>
Date: Sun, 29 Jul 2018 12:53:32 +0100
Subject: Clean up nodes and lists

---
 test/DOMTSHandler.pm | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'test')

diff --git a/test/DOMTSHandler.pm b/test/DOMTSHandler.pm
index 09b657e..0cf3a56 100644
--- a/test/DOMTSHandler.pm
+++ b/test/DOMTSHandler.pm
@@ -1306,6 +1306,7 @@ sub generate_control_statement {
 				print "unsigned int iterator$iterator_index = 0;";
 				print "foreach_initialise_domnodelist($coll, \&iterator$iterator_index);\n";
 				print "while(get_next_domnodelist($coll, \&iterator$iterator_index, \&$member)) {\n";
+                                $self->addto_cleanup($member);
 			}
 
 			if ($self->{"var"}->{$coll} eq "NamedNodeMap") {
@@ -1560,6 +1561,12 @@ sub end_half_assertion {
 ### Enclose an unsed function
 ##############################################################################################
 
+sub cleanup_lists {
+   my ($self, $indent) = @_;
+   foreach my $list (keys %{$self->{list_map}}) {
+      print "${indent}if ($list != NULL)\n${indent}\tlist_destroy($list);\n" if ($list ne '');
+   }
+}
 
 sub cleanup_domvar {
 	my ($self, $indent) = @_;
@@ -1579,6 +1586,7 @@ sub cleanup {
 	my $self = shift;
 
 	print "\n\n";
+        $self->cleanup_lists("\t");
 	$self->cleanup_domstring("\t");
 	$self->cleanup_domvar("\t");
         print "\n\tprintf(\"PASS\");\n";
-- 
cgit v1.2.3