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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
| // Copyright 2017 syzkaller project authors. All rights reserved.
| // Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
|
| package prog
|
| import (
| "fmt"
| "math/rand"
| "testing"
| "time"
| )
|
| // Export guts for testing.
|
| func init() {
| debug = true
| }
|
| var (
| CalcChecksumsCall = calcChecksumsCall
| InitTest = initTest
| )
|
| func initTargetTest(t *testing.T, os, arch string) *Target {
| t.Parallel()
| target, err := GetTarget(os, arch)
| if err != nil {
| t.Fatal(err)
| }
| return target
| }
|
| func randSource(t *testing.T) rand.Source {
| seed := int64(time.Now().UnixNano())
| t.Logf("seed=%v", seed)
| return rand.NewSource(seed)
| }
|
| func iterCount() int {
| iters := 10000
| if testing.Short() {
| iters = 100
| }
| if raceEnabled {
| iters /= 10
| }
| return iters
| }
|
| func initRandomTargetTest(t *testing.T, os, arch string) (*Target, rand.Source, int) {
| target := initTargetTest(t, os, arch)
| return target, randSource(t), iterCount()
| }
|
| func initTest(t *testing.T) (*Target, rand.Source, int) {
| return initRandomTargetTest(t, "linux", "amd64")
| }
|
| func testEachTarget(t *testing.T, fn func(t *testing.T, target *Target)) {
| t.Parallel()
| for _, target := range AllTargets() {
| target := target
| t.Run(fmt.Sprintf("%v/%v", target.OS, target.Arch), func(t *testing.T) {
| t.Parallel()
| fn(t, target)
| })
| }
| }
|
| func testEachTargetRandom(t *testing.T, fn func(t *testing.T, target *Target, rs rand.Source, iters int)) {
| t.Parallel()
| targets := AllTargets()
| iters := iterCount()
| iters /= len(targets)
| if iters < 3 {
| iters = 3
| }
| rs0 := randSource(t)
| for _, target := range targets {
| target := target
| rs := rand.NewSource(rs0.Int63())
| t.Run(fmt.Sprintf("%v/%v", target.OS, target.Arch), func(t *testing.T) {
| t.Parallel()
| fn(t, target, rs, iters)
| })
| }
| }
|
|
