clean up tests

This commit is contained in:
Chris Lu
2019-05-19 21:16:01 -07:00
parent 6386a3174b
commit 693d6c0f1b

View File

@@ -37,6 +37,8 @@ func TestEncodingDecoding(t *testing.T) {
t.Logf("writeSortedEcxFiles: %v", err) t.Logf("writeSortedEcxFiles: %v", err)
} }
removeGeneratedFiles(baseFileName)
} }
func generateEcFiles(baseFileName string, bufferSize int, largeBlockSize int64, smallBlockSize int64) error { func generateEcFiles(baseFileName string, bufferSize int, largeBlockSize int64, smallBlockSize int64) error {
@@ -281,6 +283,14 @@ func readFromFile(file *os.File, data []byte, ecFileOffset int64) (err error) {
return 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) { func TestLocateData(t *testing.T) {
intervals := locateData(largeBlockSize, smallBlockSize, DataShardsCount*largeBlockSize+1, DataShardsCount*largeBlockSize, 1) intervals := locateData(largeBlockSize, smallBlockSize, DataShardsCount*largeBlockSize+1, DataShardsCount*largeBlockSize, 1)
if len(intervals) != 1 { if len(intervals) != 1 {