mirror of https://github.com/portainer/portainer
55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package adminmonitor
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
portainer "github.com/portainer/portainer/api"
|
|
i "github.com/portainer/portainer/api/internal/testhelpers"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_stopWithoutStarting(t *testing.T) {
|
|
monitor := New(1*time.Minute, nil, nil)
|
|
monitor.Stop()
|
|
}
|
|
|
|
func Test_stopCouldBeCalledMultipleTimes(t *testing.T) {
|
|
monitor := New(1*time.Minute, nil, nil)
|
|
monitor.Stop()
|
|
monitor.Stop()
|
|
}
|
|
|
|
func Test_startOrStopCouldBeCalledMultipleTimesConcurrently(t *testing.T) {
|
|
monitor := New(1*time.Minute, nil, context.Background())
|
|
|
|
go monitor.Start()
|
|
monitor.Start()
|
|
|
|
go monitor.Stop()
|
|
monitor.Stop()
|
|
|
|
time.Sleep(2 * time.Second)
|
|
}
|
|
|
|
func Test_canStopStartedMonitor(t *testing.T) {
|
|
monitor := New(1*time.Minute, nil, context.Background())
|
|
monitor.Start()
|
|
assert.NotNil(t, monitor.cancellationFunc, "cancellation function is missing in started monitor")
|
|
|
|
monitor.Stop()
|
|
assert.Nil(t, monitor.cancellationFunc, "cancellation function should absent in stopped monitor")
|
|
}
|
|
|
|
func Test_start_shouldDisableInstanceAfterTimeout_ifNotInitialized(t *testing.T) {
|
|
timeout := 10 * time.Millisecond
|
|
|
|
datastore := i.NewDatastore(i.WithUsers([]portainer.User{}))
|
|
monitor := New(timeout, datastore, context.Background())
|
|
monitor.Start()
|
|
|
|
<-time.After(20 * timeout)
|
|
assert.True(t, monitor.WasInstanceDisabled(), "monitor should have been timeout and instance is disabled")
|
|
}
|