huntedbytheirsandSisyphus
38ddc7cfb2
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <[email protected]>
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestNormalizeArgs(t *testing.T) {
|
|
valueFlags := map[string]bool{
|
|
"--output": true,
|
|
"--repo": true,
|
|
"--arch": true,
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
in []string
|
|
want []string
|
|
}{
|
|
{
|
|
name: "flags after positional",
|
|
in: []string{"input.pkg.tar.zst", "--output", "/tmp/out"},
|
|
want: []string{"--output", "/tmp/out", "input.pkg.tar.zst"},
|
|
},
|
|
{
|
|
name: "flags before positional",
|
|
in: []string{"--output", "/tmp/out", "input.pkg.tar.zst"},
|
|
want: []string{"--output", "/tmp/out", "input.pkg.tar.zst"},
|
|
},
|
|
{
|
|
name: "flag=value form",
|
|
in: []string{"input.pkg.tar.zst", "--output=/tmp/out"},
|
|
want: []string{"--output=/tmp/out", "input.pkg.tar.zst"},
|
|
},
|
|
{
|
|
name: "unknown flag stays a flag",
|
|
in: []string{"input.pkg.tar.zst", "--bogus"},
|
|
want: []string{"--bogus", "input.pkg.tar.zst"},
|
|
},
|
|
{
|
|
name: "value-taking flag at end without value",
|
|
in: []string{"input.pkg.tar.zst", "--output"},
|
|
want: []string{"--output", "input.pkg.tar.zst"},
|
|
},
|
|
{
|
|
name: "mixed flags and positionals keep order",
|
|
in: []string{"--repo", "https://x", "a.zst", "--output", "/tmp/out", "b.zst"},
|
|
want: []string{"--repo", "https://x", "--output", "/tmp/out", "a.zst", "b.zst"},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := normalizeArgs(tt.in, valueFlags)
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
t.Fatalf("normalizeArgs(%v) = %v, want %v", tt.in, got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|