accounting: refactor bwlimit code to allow for multiple slots

This commit is contained in:
Nick Craig-Wood
2020-07-04 17:20:54 +01:00
parent 31de631b22
commit 3b6df71838
7 changed files with 127 additions and 90 deletions

View File

@@ -1079,13 +1079,13 @@ func TestSyncWithMaxDuration(t *testing.T) {
maxDuration := 250 * time.Millisecond
ci.MaxDuration = maxDuration
bytesPerSecond := 300
accounting.SetBwLimit(fs.SizeSuffix(bytesPerSecond))
accounting.TokenBucket.SetBwLimit(fs.SizeSuffix(bytesPerSecond))
oldTransfers := ci.Transfers
ci.Transfers = 1
defer func() {
ci.MaxDuration = 0 // reset back to default
ci.Transfers = oldTransfers
accounting.SetBwLimit(fs.SizeSuffix(0))
accounting.TokenBucket.SetBwLimit(fs.SizeSuffix(0))
}()
// 5 files of 60 bytes at 60 bytes/s 5 seconds