Issue #25: cleaning up test case

Netflix · Sep 7, 2013 · fe23479 · fe23479
1 parent f3a33d5
commit fe23479
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/genie-server/src/test/java/com/netflix/genie/server/jobmanager/impl/TestJobJanitor.java b/genie-server/src/test/java/com/netflix/genie/server/jobmanager/impl/TestJobJanitor.java
@@ -60,5 +60,8 @@ public void testJobJanitor() throws Exception {
         int numRows = janitor.markZombies();
         System.out.println("Number of rows marked as zombies: " + numRows);
         Assert.assertEquals(numRows >= 2, true);
+
+        // shut down cleanly
+        PersistenceManager.shutdown();
     }
 }
diff --git a/genie-server/src/test/java/com/netflix/genie/server/metrics/TestGenieNodeStatistics.java b/genie-server/src/test/java/com/netflix/genie/server/metrics/TestGenieNodeStatistics.java
@@ -27,6 +27,7 @@
 
 import com.netflix.config.ConfigurationManager;
 import com.netflix.genie.common.exceptions.CloudServiceException;
+import com.netflix.genie.server.persistence.PersistenceManager;
 
 /**
  * Test case for GenieNodeStatistics.
@@ -164,5 +165,6 @@ public void testRunningJobs() throws InterruptedException, CloudServiceException
     public static void shutdown() {
         // shut down cleanly
         stats.shutdown();
+        PersistenceManager.shutdown();
     }
 }