1 files changed, 14 insertions, 0 deletions

diff --git a/src/unitsuite.cpp b/src/unitsuite.cpp
index a66160e..51cca10 100644
--- a/src/unitsuite.cpp
+++ b/src/unitsuite.cpp
@@ -6,6 +6,9 @@
  */
 
 #include "bu/unitsuite.h"
+#include "bu/file.h"
+
+#include <unistd.h>
 
 Bu::UnitSuite::UnitSuite() :
         iOptions( 0 )
@@ -102,9 +105,20 @@ int Bu::UnitSuite::run( int /*argc*/, char * /*argv */ [] )
         if( iUPass == 0 && iUFail == 0 )
                 printf("\tNothing unexpected.\n\n");
 
+        for( StrList::iterator i = lFileCleanup.begin(); i; i++ )
+        {
+                unlink( (*i).getStr() );
+        }
+
         return 0;
 }
 
+void Bu::UnitSuite::tempFile( Bu::FString &sFileName )
+{
+        Bu::File::tempFile( sFileName );
+        lFileCleanup.append( sFileName );
+}
+
 void Bu::UnitSuite::add( Test fTest, const Bu::FString &sName, Expect e )
 {
         TestInfo ti;