github.com/xhghs/rclone@v1.51.1-0.20200430155106-e186a28cced8/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  }