should fix teardown

hackshackers · Dec 23, 2016 · 41b7415 · 41b7415
1 parent 31f41c2
commit 41b7415
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/hugo-content-test/.gitkeep b/hugo-content-test/.gitkeep
@@ -0,0 +1 @@
+# Keep this dir around for unit tests
diff --git a/tests/test-write-file.php b/tests/test-write-file.php
@@ -5,10 +5,16 @@ function setUp() {
 	}
 
 	function tearDown() {
-		unlink( './hugo-content-test/2016/12/the-post.md' );
-		$dirs = array( './2016/12/23', './2016/12', './hugo-content-test/2016/12' );
+		$test_file = './hugo-content-test/2016/12/the-post.md';
+		if ( file_exists( $test_file ) ) {
+			unlink( $test_file );
+		}
+
+		$dirs = array( './2016/12/23', './2016/12', './2016', './hugo-content-test/2016/12', './hugo-content-test/2016' );
 		foreach ( $dirs as $dir ) {
-			rmdir( $dir );
+			if ( is_dir( $dir ) ) {
+				rmdir( $dir );
+			}
 		}
 	}