close things properly in the test

Author: kaoudis

test/scripts/ghidra/util/FunctionSerializerTest.java

@@ -11,6 +11,10 @@
 
 import org.junit.jupiter.api.*;
 
+import com.google.gson.JsonObject;
+import com.google.gson.JsonParser;
+import com.google.gson.stream.JsonWriter;
+
 import ghidra.test.AbstractGhidraHeadlessIntegrationTest;
 import ghidra.test.TestEnv;
 
@@ -107,15 +111,40 @@ public void setUp() throws Exception {
         assertFalse(fns.isEmpty());
     }
 
-    @BeforeEach 
-    public void startTest() throws Exception {
+    @AfterAll
+    public void cleanUp() throws Exception {
+        if (program != null) {
+            program.release(this);
+        }
+        if (env != null) {
+            env.dispose();
+        }
+
+        if (!stringWriter.getBuffer().isEmpty()) {
+            writer.close();
+            stringWriter.close();
+        }
+    }
+
+    @Test 
+    public void testSerialize() throws Exception {
         stringWriter = new StringWriter();
         assertNotNull(stringWriter);
+
         writer = new BufferedWriter(stringWriter);
         assertNotNull(writer);
 
         functionSerializer = new FunctionSerializer(writer, program);
         assertNotNull(functionSerializer.currentProgram);
         assertFalse(functionSerializer.currentProgram.isClosed());
+
+        JsonWriter jsonWriter = functionSerializer.serialize(fns);
+        assertNotNull(jsonWriter);
+
+        assertDoesNotThrow(() -> { stringWriter.toString(); });
+
+        JsonObject jsonOutput = JsonParser
+            .parseString(stringWriter.toString())
+            .getAsJsonObject();
     }
 }