From 66b27a3d572d39a871cd1d52828f5d3445925fea Mon Sep 17 00:00:00 2001
From: Robert Gemmell <robbie@apache.org>
Date: Tue, 20 Oct 2009 14:45:21 +0000
Subject: QPID-2040: update test to ensure all test files created are cleared
 up

git-svn-id: https://svn.apache.org/repos/asf/qpid/trunk/qpid@827586 13f79535-47bb-0310-9956-ffa450edef68
---
 .../PlainPasswordFilePrincipalDatabaseTest.java      | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

(limited to 'java/broker/src')

diff --git a/java/broker/src/test/java/org/apache/qpid/server/security/auth/database/PlainPasswordFilePrincipalDatabaseTest.java b/java/broker/src/test/java/org/apache/qpid/server/security/auth/database/PlainPasswordFilePrincipalDatabaseTest.java
index 20b8d0a7b4..a3dad19bb4 100644
--- a/java/broker/src/test/java/org/apache/qpid/server/security/auth/database/PlainPasswordFilePrincipalDatabaseTest.java
+++ b/java/broker/src/test/java/org/apache/qpid/server/security/auth/database/PlainPasswordFilePrincipalDatabaseTest.java
@@ -34,6 +34,7 @@ import java.io.FileReader;
 import java.io.FileWriter;
 import java.io.IOException;
 import java.security.Principal;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.regex.Pattern;
 
@@ -47,12 +48,29 @@ public class PlainPasswordFilePrincipalDatabaseTest extends TestCase
     
     private Principal _principal = new UsernamePrincipal(TEST_USERNAME);
     private PlainPasswordFilePrincipalDatabase _database;
+    private List<File> _testPwdFiles = new ArrayList<File>();
 
     public void setUp() throws Exception
     {
         _database = new PlainPasswordFilePrincipalDatabase();
+        _testPwdFiles.clear();
     }
 
+    public void tearDown() throws Exception
+    {
+        //clean up any additional files and their backups
+        for(File f : _testPwdFiles)
+        {
+            File oldPwdFile = new File(f.getAbsolutePath() + ".old");
+            if(oldPwdFile.exists())
+            {
+                oldPwdFile.delete();
+            }
+            
+            f.delete();
+        }
+    }
+    
     // ******* Test Methods ********** //
 
     public void testCreatePrincipal()
@@ -368,6 +386,8 @@ public class PlainPasswordFilePrincipalDatabaseTest extends TestCase
 
             writer.flush();
             writer.close();
+            
+            _testPwdFiles.add(testFile);
 
             return testFile;
 
-- 
cgit v1.2.1