devstatscode/threads_test.go at master · cncf/devstatscode · GitHub

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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
package devstatscode

import (
	"fmt"
	"os"
	"testing"

	lib "github.com/cncf/devstatscode"
)

func TestGetThreadsNum(t *testing.T) {
	// Environment context parse
	var ctx lib.Ctx
	ctx.Init()
	ctx.TestMode = true

	// Get actual number of threads available
	nThreads := lib.GetThreadsNum(&ctx)

	// Set context's ST/NCPUs manually (don't need to repeat tests from context_test.go)
	var testCases = []struct {
		ST       bool
		NCPUs    int
		expected int
	}{
		{ST: false, NCPUs: 0, expected: nThreads},
		{ST: false, NCPUs: 1, expected: 1},
		{ST: false, NCPUs: -1, expected: nThreads},
		{ST: false, NCPUs: 2, expected: 2},
		{ST: true, NCPUs: 0, expected: 1},
		{ST: true, NCPUs: 1, expected: 1},
		{ST: true, NCPUs: -1, expected: 1},
		{ST: true, NCPUs: 2, expected: 2},
		{ST: true, NCPUs: nThreads + 1, expected: nThreads},
	}
	// Execute test cases
	for index, test := range testCases {
		ctx.ST = test.ST
		ctx.NCPUs = test.NCPUs
		expected := test.expected
		if test.ST {
			os.Setenv("GHA2DB_ST", "1")
		} else {
			os.Unsetenv("GHA2DB_ST")
		}
		if test.NCPUs > 0 {
			os.Setenv("GHA2DB_NCPUS", fmt.Sprintf("%d", test.NCPUs))
		} else {
			os.Unsetenv("GHA2DB_NCPUS")
		}
		got := lib.GetThreadsNum(&ctx)
		if got != expected {
			t.Errorf(
				"test number %d, expected to return %d threads, got %d (default is %d on this machine)",
				index+1, expected, got, nThreads,
			)
		}
	}
}