36 lines
710 B
Go
36 lines
710 B
Go
|
package raft
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDataBackupHelper(t *testing.T) {
|
||
|
|
keep := 5
|
||
|
|
|
||
|
|
cleanup := func() {
|
||
|
|
os.RemoveAll("data_helper_testing")
|
||
|
|
for i := 0; i < 2*keep; i++ {
|
||
|
|
os.RemoveAll(fmt.Sprintf("data_helper_testing.old.%d", i))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
cleanup()
|
||
|
|
defer cleanup()
|
||
|
|
|
||
|
|
os.MkdirAll("data_helper_testing", 0700)
|
||
|
|
helper := newDataBackupHelper("data_helper_testing", keep)
|
||
|
|
for i := 0; i < 2*keep; i++ {
|
||
|
|
err := helper.makeBackup()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
backups := helper.listBackups()
|
||
|
|
if (i < keep && len(backups) != i+1) ||
|
||
|
|
(i >= keep && len(backups) != keep) {
|
||
|
|
t.Fatal("incorrect number of backups saved")
|
||
|
|
}
|
||
|
|
os.MkdirAll("data_helper_testing", 0700)
|
||
|
|
}
|
||
|
|
}
|