2023-01-12 04:25:44 +08:00
|
|
|
package seafile
|
|
|
|
|
|
|
|
import (
|
|
|
|
"sync/atomic"
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestShouldAllowShutdownTwice(t *testing.T) {
|
|
|
|
renew := NewRenew(time.Hour, func() error {
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
renew.Shutdown()
|
|
|
|
renew.Shutdown()
|
|
|
|
}
|
|
|
|
|
2023-03-03 00:39:58 +08:00
|
|
|
func TestRenewalInTimeLimit(t *testing.T) {
|
2023-01-12 04:25:44 +08:00
|
|
|
var count int64
|
|
|
|
|
2023-03-03 00:39:58 +08:00
|
|
|
renew := NewRenew(100*time.Millisecond, func() error {
|
2023-01-12 04:25:44 +08:00
|
|
|
atomic.AddInt64(&count, 1)
|
|
|
|
return nil
|
|
|
|
})
|
2023-03-03 00:39:58 +08:00
|
|
|
time.Sleep(time.Second)
|
2023-01-12 04:25:44 +08:00
|
|
|
renew.Shutdown()
|
|
|
|
|
2023-03-03 00:39:58 +08:00
|
|
|
// there's no guarantee the CI agent can handle a simple goroutine
|
|
|
|
renewCount := atomic.LoadInt64(&count)
|
|
|
|
t.Logf("renew count = %d", renewCount)
|
|
|
|
assert.Greater(t, renewCount, int64(0))
|
|
|
|
assert.Less(t, renewCount, int64(11))
|
2023-01-12 04:25:44 +08:00
|
|
|
}
|