Erasing in parallel mode now deletes all the data files. #262

1 files changed, 15 insertions, 3 deletions

diff --git a/coverage/data.py b/coverage/data.py
index 4a5ae27a..0fa2c878 100644
--- a/coverage/data.py
+++ b/coverage/data.py
@@ -6,6 +6,7 @@
 import glob
 import json
 import os
+import os.path
 import random
 import socket
 
@@ -553,9 +554,20 @@ class CoverageDataFiles(object):
         # Construct the filename that will be used for data storage.
         self.filename = os.path.abspath(basename or ".coverage")
 
-    def erase(self):
-        """Erase the data from the file storage."""
+    def erase(self, parallel=False):
+        """Erase the data from the file storage.
+
+        If `parallel` is true, then also deletes data files created from the
+        basename by parallel-mode.
+
+        """
         file_be_gone(self.filename)
+        if parallel:
+            data_dir, local = os.path.split(self.filename)
+            localdot = local + '.*'
+            pattern = os.path.join(os.path.abspath(data_dir), localdot)
+            for filename in glob.glob(pattern):
+                file_be_gone(filename)
 
     def read(self, data):
         """Read the coverage data."""
@@ -630,7 +642,7 @@ class CoverageDataFiles(object):
             new_data = CoverageData()
             new_data.read_file(f)
             data.update(new_data, aliases=aliases)
-            os.remove(f)
+            file_be_gone(f)
 
 
 def debug_main(args):