summaryrefslogtreecommitdiffstats
path: root/src/go/collectors/go.d.plugin/modules/vsphere/task_test.go
blob: ed55a28a368cde4be2426221498118026dc2cba7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
// SPDX-License-Identifier: GPL-3.0-or-later

package vsphere

import (
	"sync/atomic"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func Test_task(t *testing.T) {
	var i int64
	job := func() {
		atomic.AddInt64(&i, 1)
	}

	task := newTask(job, time.Millisecond*200)
	defer task.stop()
	time.Sleep(time.Second)
	assert.True(t, atomic.LoadInt64(&i) > 0)
}

func Test_task_isStopped(t *testing.T) {
	task := newTask(func() {}, time.Second)
	assert.False(t, task.isStopped())

	task.stop()
	time.Sleep(time.Millisecond * 500)
	assert.True(t, task.isStopped())
}

func Test_task_isRunning(t *testing.T) {
	task := newTask(func() {}, time.Second)
	assert.True(t, task.isRunning())

	task.stop()
	time.Sleep(time.Millisecond * 500)
	assert.False(t, task.isRunning())
}