clean up tests

1 files changed, 11 insertions, 1 deletions

diff --git a/weed/storage/erasure_coding/ec_test.go b/weed/storage/erasure_coding/ec_test.go
index fb6cb1f93..d631471e9 100644
--- a/weed/storage/erasure_coding/ec_test.go
+++ b/weed/storage/erasure_coding/ec_test.go
@@ -37,6 +37,8 @@ func TestEncodingDecoding(t *testing.T) {
 		t.Logf("writeSortedEcxFiles: %v", err)
 	}
 
+	removeGeneratedFiles(baseFileName)
+
 }
 
 func generateEcFiles(baseFileName string, bufferSize int, largeBlockSize int64, smallBlockSize int64) error {
@@ -251,7 +253,7 @@ func readFromOtherEcFiles(ecFiles []*os.File, ecFileIndex int, ecFileOffset int6
 
 	bufs := make([][]byte, DataShardsCount+ParityShardsCount)
 	for i := 0; i < DataShardsCount; {
-		n := int(rand.Int31n(DataShardsCount+ParityShardsCount))
+		n := int(rand.Int31n(DataShardsCount + ParityShardsCount))
 		if n == ecFileIndex || bufs[n] != nil {
 			continue
 		}
@@ -281,6 +283,14 @@ func readFromFile(file *os.File, data []byte, ecFileOffset int64) (err error) {
 	return
 }
 
+func removeGeneratedFiles(baseFileName string) {
+	for i := 0; i < DataShardsCount+ParityShardsCount; i++ {
+		fname := fmt.Sprintf("%s.ec%02d", baseFileName, i+1)
+		os.Remove(fname)
+	}
+	os.Remove(baseFileName+".ecx")
+}
+
 func TestLocateData(t *testing.T) {
 	intervals := locateData(largeBlockSize, smallBlockSize, DataShardsCount*largeBlockSize+1, DataShardsCount*largeBlockSize, 1)
 	if len(intervals) != 1 {