#414 fix testgen tester: wrong dir was removed

informalsystems · Sep 23, 2020 · 85b8814 · 85b8814
1 parent b479d59
commit 85b8814
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/testgen/src/tester.rs b/testgen/src/tester.rs
@@ -202,9 +202,7 @@ impl Tester {
 
     pub fn output_env(&self) -> Option<TestEnv> {
         let output_dir = self.root_dir.clone() + "/_" + &self.name;
-        fs::create_dir_all(&output_dir)
-            .ok()
-            .and(TestEnv::new(&output_dir))
+        TestEnv::new(&output_dir)
     }
 
     fn capture_test<F>(test: F) -> TestResult
@@ -270,8 +268,8 @@ impl Tester {
                 let env = TestEnv::new(dir.path().to_str().unwrap()).unwrap();
                 let output_dir = output_env.full_path(path);
                 let output_env = TestEnv::new(output_dir.to_str().unwrap()).unwrap();
-                fs::remove_dir_all(&output_dir).unwrap();
                 test(test_case, &env, &test_env, &output_env);
+                fs::remove_dir_all(&env.current_dir()).unwrap();
             }),
             Err(_) => ParseError,
         };