Minor fixes

lib/core/testing.py

@@ -11,6 +11,7 @@ import logging
 import os
 import random
 import re
+import shutil
 import socket
 import sqlite3
 import subprocess
@@ -212,6 +213,7 @@ def vulnTest():
         if "<tmpfile>" in cmd:
             handle, tmp = tempfile.mkstemp()
             os.close(handle)
+            cleanups.append(tmp)
             cmd = cmd.replace("<tmpfile>", tmp)
 
         os.environ["SQLMAP_UNSAFE_EVAL"] = '1'
@@ -237,6 +239,11 @@ def vulnTest():
         except:
             pass
 
+    try:
+        shutil.rmtree(tmpdir)
+    except:
+        pass
+
     return retVal
 
 def apiTest():