From 4ea2d4cbbc94f5db09de937ff304d32e302df9c0 Mon Sep 17 00:00:00 2001
From: vandomej <jacob.vandome15@gmail.com>
Date: Thu, 24 Jun 2021 06:28:26 -0700
Subject: [PATCH] File cleanup for unit tests

---
 gemla/src/tests/bracket.rs     | 4 ++++
 gemla/src/tests/file_linked.rs | 6 +++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/gemla/src/tests/bracket.rs b/gemla/src/tests/bracket.rs
index cc9b273..2027f98 100644
--- a/gemla/src/tests/bracket.rs
+++ b/gemla/src/tests/bracket.rs
@@ -60,6 +60,8 @@ fn test_new() {
         format!("{{\"tree\":{},\"step\":0,\"iteration_scaling\":{{\"enumType\":\"Linear\",\"enumContent\":1}}}}", 
         btree!(TestState::new(0.0)))
     );
+
+    std::fs::remove_file("./temp").expect("Unable to remove file");
 }
 
 #[test]
@@ -90,4 +92,6 @@ fn test_run() {
             Some(btree!(TestState::new(6.0)))
         ))
     );
+
+    std::fs::remove_file("./temp").expect("Unable to remove file");
 }
\ No newline at end of file
diff --git a/gemla/src/tests/file_linked.rs b/gemla/src/tests/file_linked.rs
index b33e075..ab4d755 100644
--- a/gemla/src/tests/file_linked.rs
+++ b/gemla/src/tests/file_linked.rs
@@ -1,9 +1,11 @@
 use super::super::file_linked::FileLinked;
 
+use std::fs;
+
 #[test]
 fn test_mutate() -> Result<(), String> {
     let tree = btree!(1, Some(btree!(2)), Some(btree!(3, Some(btree!(4)),)));
-    let mut linked_tree = FileLinked::new(tree, String::from("blah.txt"))?;
+    let mut linked_tree = FileLinked::new(tree, String::from("test.txt"))?;
 
     assert_eq!(
         format!("{}", linked_tree.readonly()),
@@ -28,5 +30,7 @@ fn test_mutate() -> Result<(), String> {
         "{\"val\":10,\"left\":{\"val\":13,\"left\":null,\"right\":null},\"right\":{\"val\":3,\"left\":{\"val\":4,\"left\":null,\"right\":null},\"right\":null}}"
     );
 
+    fs::remove_file("test.txt").expect("Unable to remove file");
+
     Ok(())
 }