mirror of
https://github.com/spf13/cobra
synced 2025-01-26 05:26:45 +00:00
Make detection for test-binary more universal (#2173)
When running tests in verbose mode (or other options), tests involving Cobra may fail if the test does not explicitly set Command.args to an empty slice; in this case, Cobra defaults to using `os.Args`, which will contain arguments passed to the test (such as `-v` (verbose)). Commitse5762054c9
and1ef0913976
implemented a workaround for this when running (unit) tests for Cobra itself, but this check is specifig to Cobra (checking for `cobra.test`), and don't work on Windows (which will have a `.exe` extension), This patch implements a more universal check, so that users of Cobra as a module also benefit from this workaround. go1.21 and up provides a `testing.Testing()` utility ([1]); as the Cobra module still supports Go1.16 and up, an alternative implementation was added for older versions, based on golang.org/x/mod/lazyregexp [2]. Before this patch: go test -c -o foo.test ./foo.test -test.run TestNoArgs --- FAIL: TestNoArgs (0.00s) args_test.go:37: Unexpected output: Error: unknown command "TestNoArgs" for "c" Usage: c [flags] Flags: -h, --help help for c args_test.go:40: Unexpected error: unknown command "TestNoArgs" for "c" FAIL After this patch: go test -c -o foo.test ./foo.test -test.run TestNoArgs PASS [1]: https://pkg.go.dev/testing#Testing [2]: https://cs.opensource.google/go/x/mod/+/refs/tags/v0.19.0:internal/lazyregexp/lazyre.go;l=66-78 Signed-off-by: Sebastiaan van Stijn <github@gone.nl> Signed-off-by: Marc Khouzam <marc.khouzam@gmail.com> Co-authored-by: Marc Khouzam <marc.khouzam@gmail.com>
This commit is contained in:
parent
9f9056765c
commit
d1e9d85fcf
4 changed files with 76 additions and 3 deletions
|
@ -23,7 +23,6 @@ import (
|
|||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sort"
|
||||
"strings"
|
||||
|
||||
|
@ -1078,8 +1077,11 @@ func (c *Command) ExecuteC() (cmd *Command, err error) {
|
|||
|
||||
args := c.args
|
||||
|
||||
// Workaround FAIL with "go test -v" or "cobra.test -test.v", see #155
|
||||
if c.args == nil && filepath.Base(os.Args[0]) != "cobra.test" {
|
||||
// If running unit tests, we don't want to take the os.Args, see #155 and #2173.
|
||||
// For example, the following would fail:
|
||||
// go test -c -o foo.test
|
||||
// ./foo.test -test.run TestNoArgs
|
||||
if c.args == nil && !isTesting() {
|
||||
args = os.Args[1:]
|
||||
}
|
||||
|
||||
|
|
33
command_go120.go
Normal file
33
command_go120.go
Normal file
|
@ -0,0 +1,33 @@
|
|||
// Copyright 2013-2024 The Cobra Authors
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
//go:build !go1.21
|
||||
// +build !go1.21
|
||||
|
||||
package cobra
|
||||
|
||||
import (
|
||||
"os"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// based on golang.org/x/mod/internal/lazyregexp: https://cs.opensource.google/go/x/mod/+/refs/tags/v0.19.0:internal/lazyregexp/lazyre.go;l=66
|
||||
// For a non-go-test program which still has a name ending with ".test[.exe]", it will need to either:
|
||||
// 1- Use go >= 1.21, or
|
||||
// 2- call "rootCmd.SetArgs(os.Args[1:])" before calling "rootCmd.Execute()"
|
||||
var inTest = len(os.Args) > 0 && strings.HasSuffix(strings.TrimSuffix(os.Args[0], ".exe"), ".test")
|
||||
|
||||
func isTesting() bool {
|
||||
return inTest
|
||||
}
|
25
command_go121.go
Normal file
25
command_go121.go
Normal file
|
@ -0,0 +1,25 @@
|
|||
// Copyright 2013-2024 The Cobra Authors
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
//go:build go1.21
|
||||
// +build go1.21
|
||||
|
||||
package cobra
|
||||
|
||||
import "testing"
|
||||
|
||||
func isTesting() bool {
|
||||
// Only available starting with go 1.21
|
||||
return testing.Testing()
|
||||
}
|
|
@ -2839,3 +2839,16 @@ func TestUnknownFlagShouldReturnSameErrorRegardlessOfArgPosition(t *testing.T) {
|
|||
})
|
||||
}
|
||||
}
|
||||
|
||||
// This tests verifies that when running unit tests, os.Args are not used.
|
||||
// This is because we don't want to process any arguments that are provided
|
||||
// by "go test"; instead, unit tests must set the arguments they need using
|
||||
// rootCmd.SetArgs().
|
||||
func TestNoOSArgsWhenTesting(t *testing.T) {
|
||||
root := &Command{Use: "root", Run: emptyRun}
|
||||
os.Args = append(os.Args, "--unknown")
|
||||
|
||||
if _, err := root.ExecuteC(); err != nil {
|
||||
t.Errorf("error: %v", err)
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue