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
88
89
|
// Copyright 2022 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package testing_test
import (
"flag"
"internal/testenv"
"os"
"os/exec"
"testing"
)
var testFlagArg = flag.String("test_flag_arg", "", "TestFlag: passing -v option")
const flagTestEnv = "GO_WANT_FLAG_HELPER_PROCESS"
func TestFlag(t *testing.T) {
if os.Getenv(flagTestEnv) == "1" {
testFlagHelper(t)
return
}
testenv.MustHaveExec(t)
for _, flag := range []string{"", "-test.v", "-test.v=test2json"} {
flag := flag
t.Run(flag, func(t *testing.T) {
t.Parallel()
exe, err := os.Executable()
if err != nil {
exe = os.Args[0]
}
cmd := exec.Command(exe, "-test.run=TestFlag", "-test_flag_arg="+flag)
if flag != "" {
cmd.Args = append(cmd.Args, flag)
}
cmd.Env = append(cmd.Environ(), flagTestEnv+"=1")
b, err := cmd.CombinedOutput()
if len(b) > 0 {
// When we set -test.v=test2json, we need to escape the ^V control
// character used for JSON framing so that the JSON parser doesn't
// misinterpret the subprocess output as output from the parent test.
t.Logf("%q", b)
}
if err != nil {
t.Error(err)
}
})
}
}
// testFlagHelper is called by the TestFlagHelper subprocess.
func testFlagHelper(t *testing.T) {
f := flag.Lookup("test.v")
if f == nil {
t.Fatal(`flag.Lookup("test.v") failed`)
}
bf, ok := f.Value.(interface{ IsBoolFlag() bool })
if !ok {
t.Errorf("test.v flag (type %T) does not have IsBoolFlag method", f)
} else if !bf.IsBoolFlag() {
t.Error("test.v IsBoolFlag() returned false")
}
gf, ok := f.Value.(flag.Getter)
if !ok {
t.Fatalf("test.v flag (type %T) does not have Get method", f)
}
v := gf.Get()
var want any
switch *testFlagArg {
case "":
want = false
case "-test.v":
want = true
case "-test.v=test2json":
want = "test2json"
default:
t.Fatalf("unexpected test_flag_arg %q", *testFlagArg)
}
if v != want {
t.Errorf("test.v is %v want %v", v, want)
}
}
|
