perso/syncthing-arm
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

test: Fix reconnect test

Browse Source
This commit is contained in:
Jakob Borg 2017-11-13 00:36:37 +01:00
parent d8a0a477ca
commit fd5f9f8968
1 changed files with 45 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
test/reconnect_test.go
View File

@ -15,22 +15,22 @@ import (
"time" "time"
) )
func TestRestartReceiverDuringTransfer(t *testing.T) { func TestReconnectReceiverDuringTransfer(t *testing.T) {
testRestartDuringTransfer(t, false, true, 0, 0) testReconnectDuringTransfer(t, false, true, 0, 0)
} }
func TestRestartSenderDuringTransfer(t *testing.T) { func TestReconnectSenderDuringTransfer(t *testing.T) {
testRestartDuringTransfer(t, true, false, 0, 0) testReconnectDuringTransfer(t, true, false, 0, 0)
} }
func TestRestartSenderAndReceiverDuringTransfer(t *testing.T) { func TestReconnectSenderAndReceiverDuringTransfer(t *testing.T) {
// Give the receiver some time to rot with needed files but // Give the receiver some time to rot with needed files but
// without any peer. This triggers // without any peer. This triggers
// https://github.com/syncthing/syncthing/issues/463 // https://github.com/syncthing/syncthing/issues/463
testRestartDuringTransfer(t, true, true, 10*time.Second, 0) testReconnectDuringTransfer(t, true, true, 10*time.Second, 0)
} }
func testRestartDuringTransfer(t *testing.T, restartSender, restartReceiver bool, senderDelay, receiverDelay time.Duration) { func testReconnectDuringTransfer(t *testing.T, ReconnectSender, ReconnectReceiver bool, senderDelay, receiverDelay time.Duration) {
log.Println("Cleaning...") log.Println("Cleaning...")
err := removeAll("s1", "s2", "h1/index*", "h2/index*") err := removeAll("s1", "s2", "h1/index*", "h2/index*")
if err != nil { if err != nil {
@ -38,7 +38,7 @@ func testRestartDuringTransfer(t *testing.T, restartSender, restartReceiver bool
} }
log.Println("Generating files...") log.Println("Generating files...")
err = generateFiles("s1", 250, 20, "../LICENSE") err = generateFiles("s1", 2500, 20, "../LICENSE")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -58,6 +58,20 @@ func testRestartDuringTransfer(t *testing.T, restartSender, restartReceiver bool
checkedStop(t, receiver) checkedStop(t, receiver)
}() }()
// Set rate limits
cfg, err := receiver.GetConfig()
if err != nil {
t.Fatal(err)
}
cfg.Options.MaxRecvKbps = 100
cfg.Options.MaxSendKbps = 100
if err := receiver.PostConfig(cfg); err != nil {
t.Fatal(err)
}
sender.ResumeAll()
receiver.ResumeAll()
var prevBytes int var prevBytes int
for { for {
recv, err := receiver.Model("default") recv, err := receiver.Model("default")
@ -72,32 +86,34 @@ func testRestartDuringTransfer(t *testing.T, restartSender, restartReceiver bool
// Receiver has made progress // Receiver has made progress
prevBytes = recv.InSyncBytes prevBytes = recv.InSyncBytes
if restartReceiver { if ReconnectReceiver {
log.Printf("Stopping receiver...") log.Printf("Pausing receiver...")
checkedStop(t, receiver) receiver.PauseAll()
} }
if restartSender { if ReconnectSender {
log.Printf("Stopping sender...") log.Printf("Pausing sender...")
checkedStop(t, sender) sender.PauseAll()
} }
var wg sync.WaitGroup var wg sync.WaitGroup
if restartReceiver { if ReconnectReceiver {
wg.Add(1) wg.Add(1)
go func() { go func() {
time.Sleep(receiverDelay) time.Sleep(receiverDelay)
receiver = startInstance(t, 2) log.Printf("Resuming receiver...")
receiver.ResumeAll()
wg.Done() wg.Done()
}() }()
} }
if restartSender { if ReconnectSender {
wg.Add(1) wg.Add(1)
go func() { go func() {
time.Sleep(senderDelay) time.Sleep(senderDelay)
sender = startInstance(t, 1) log.Printf("Resuming sender...")
sender.ResumeAll()
wg.Done() wg.Done()
}() }()
} }
@ -108,6 +124,17 @@ func testRestartDuringTransfer(t *testing.T, restartSender, restartReceiver bool
time.Sleep(time.Second) time.Sleep(time.Second)
} }
// Reset rate limits
cfg, err = receiver.GetConfig()
if err != nil {
t.Fatal(err)
}
cfg.Options.MaxRecvKbps = 0
cfg.Options.MaxSendKbps = 0
if err := receiver.PostConfig(cfg); err != nil {
t.Fatal(err)
}
checkedStop(t, sender) checkedStop(t, sender)
checkedStop(t, receiver) checkedStop(t, receiver)