mirror of
https://github.com/tailscale/tailscale.git
synced 2024-12-05 07:55:36 +00:00
98 lines
2.3 KiB
Go
98 lines
2.3 KiB
Go
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||
|
|
||
|
package main
|
||
|
|
||
|
import (
|
||
|
"slices"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func TestSplitArgs(t *testing.T) {
|
||
|
tests := []struct {
|
||
|
name string
|
||
|
in []string
|
||
|
pre, pkgs, post []string
|
||
|
}{
|
||
|
{
|
||
|
name: "empty",
|
||
|
},
|
||
|
{
|
||
|
name: "all",
|
||
|
in: []string{"-v", "pkg1", "pkg2", "-run", "TestFoo", "-timeout=20s"},
|
||
|
pre: []string{"-v"},
|
||
|
pkgs: []string{"pkg1", "pkg2"},
|
||
|
post: []string{"-run", "TestFoo", "-timeout=20s"},
|
||
|
},
|
||
|
{
|
||
|
name: "only_pkgs",
|
||
|
in: []string{"./..."},
|
||
|
pkgs: []string{"./..."},
|
||
|
},
|
||
|
{
|
||
|
name: "pkgs_and_post",
|
||
|
in: []string{"pkg1", "-run", "TestFoo"},
|
||
|
pkgs: []string{"pkg1"},
|
||
|
post: []string{"-run", "TestFoo"},
|
||
|
},
|
||
|
{
|
||
|
name: "pkgs_and_post",
|
||
|
in: []string{"-v", "pkg2"},
|
||
|
pre: []string{"-v"},
|
||
|
pkgs: []string{"pkg2"},
|
||
|
},
|
||
|
{
|
||
|
name: "only_args",
|
||
|
in: []string{"-v", "-run=TestFoo"},
|
||
|
pre: []string{"-run", "TestFoo", "-v"}, // sorted
|
||
|
},
|
||
|
{
|
||
|
name: "space_in_pre_arg",
|
||
|
in: []string{"-run", "TestFoo", "./cmd/testwrapper"},
|
||
|
pre: []string{"-run", "TestFoo"},
|
||
|
pkgs: []string{"./cmd/testwrapper"},
|
||
|
},
|
||
|
{
|
||
|
name: "space_in_arg",
|
||
|
in: []string{"-exec", "sudo -E", "./cmd/testwrapper"},
|
||
|
pre: []string{"-exec", "sudo -E"},
|
||
|
pkgs: []string{"./cmd/testwrapper"},
|
||
|
},
|
||
|
{
|
||
|
name: "test-arg",
|
||
|
in: []string{"-exec", "sudo -E", "./cmd/testwrapper", "--", "--some-flag"},
|
||
|
pre: []string{"-exec", "sudo -E"},
|
||
|
pkgs: []string{"./cmd/testwrapper"},
|
||
|
post: []string{"--", "--some-flag"},
|
||
|
},
|
||
|
{
|
||
|
name: "dupe-args",
|
||
|
in: []string{"-v", "-v", "-race", "-race", "./cmd/testwrapper", "--", "--some-flag"},
|
||
|
pre: []string{"-race", "-v"},
|
||
|
pkgs: []string{"./cmd/testwrapper"},
|
||
|
post: []string{"--", "--some-flag"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
for _, tt := range tests {
|
||
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
pre, pkgs, post, err := splitArgs(tt.in)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if !slices.Equal(pre, tt.pre) {
|
||
|
t.Errorf("pre = %q; want %q", pre, tt.pre)
|
||
|
}
|
||
|
if !slices.Equal(pkgs, tt.pkgs) {
|
||
|
t.Errorf("pattern = %q; want %q", pkgs, tt.pkgs)
|
||
|
}
|
||
|
if !slices.Equal(post, tt.post) {
|
||
|
t.Errorf("post = %q; want %q", post, tt.post)
|
||
|
}
|
||
|
if t.Failed() {
|
||
|
t.Logf("SplitArgs(%q) = %q %q %q", tt.in, pre, pkgs, post)
|
||
|
}
|
||
|
})
|
||
|
}
|
||
|
}
|