github.com/ncw/rclone@v1.48.1-0.20190724201158-a35aa1360e3e/lib/pacer/tokens_test.go (about)

     1  package pacer
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  )
     8  
     9  func TestTokenDispenser(t *testing.T) {
    10  	td := NewTokenDispenser(5)
    11  	assert.Equal(t, 5, len(td.tokens))
    12  	td.Get()
    13  	assert.Equal(t, 4, len(td.tokens))
    14  	td.Put()
    15  	assert.Equal(t, 5, len(td.tokens))
    16  }