103 lines
2.9 KiB
Go
103 lines
2.9 KiB
Go
package arg
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestWriteUsage(t *testing.T) {
|
|
expectedUsage := "usage: example [--name NAME] [--value VALUE] [--verbose] [--dataset DATASET] [--optimize OPTIMIZE] [--ids IDS] [--values VALUES] [--workers WORKERS] INPUT [OUTPUT [OUTPUT ...]]\n"
|
|
|
|
expectedHelp := `usage: example [--name NAME] [--value VALUE] [--verbose] [--dataset DATASET] [--optimize OPTIMIZE] [--ids IDS] [--values VALUES] [--workers WORKERS] INPUT [OUTPUT [OUTPUT ...]]
|
|
|
|
positional arguments:
|
|
input
|
|
output list of outputs
|
|
|
|
options:
|
|
--name NAME name to use [default: Foo Bar]
|
|
--value VALUE secret value [default: 42]
|
|
--verbose, -v verbosity level
|
|
--dataset DATASET dataset to use
|
|
--optimize OPTIMIZE, -O OPTIMIZE
|
|
optimization level
|
|
--ids IDS Ids
|
|
--values VALUES Values [default: [3.14 42 256]]
|
|
--workers WORKERS, -w WORKERS
|
|
number of workers to start
|
|
--help, -h display this help and exit
|
|
`
|
|
var args struct {
|
|
Input string `arg:"positional"`
|
|
Output []string `arg:"positional,help:list of outputs"`
|
|
Name string `arg:"help:name to use"`
|
|
Value int `arg:"help:secret value"`
|
|
Verbose bool `arg:"-v,help:verbosity level"`
|
|
Dataset string `arg:"help:dataset to use"`
|
|
Optimize int `arg:"-O,help:optimization level"`
|
|
Ids []int64 `arg:"help:Ids"`
|
|
Values []float64 `arg:"help:Values"`
|
|
Workers int `arg:"-w,env:WORKERS,help:number of workers to start"`
|
|
}
|
|
args.Name = "Foo Bar"
|
|
args.Value = 42
|
|
args.Values = []float64{3.14, 42, 256}
|
|
p, err := NewParser(Config{}, &args)
|
|
require.NoError(t, err)
|
|
|
|
os.Args[0] = "example"
|
|
|
|
var usage bytes.Buffer
|
|
p.WriteUsage(&usage)
|
|
assert.Equal(t, expectedUsage, usage.String())
|
|
|
|
var help bytes.Buffer
|
|
p.WriteHelp(&help)
|
|
assert.Equal(t, expectedHelp, help.String())
|
|
}
|
|
|
|
func TestUsageLongPositionalWithHelp(t *testing.T) {
|
|
expectedHelp := `usage: example VERYLONGPOSITIONALWITHHELP
|
|
|
|
positional arguments:
|
|
verylongpositionalwithhelp
|
|
this positional argument is very long
|
|
|
|
options:
|
|
--help, -h display this help and exit
|
|
`
|
|
var args struct {
|
|
VeryLongPositionalWithHelp string `arg:"positional,help:this positional argument is very long"`
|
|
}
|
|
|
|
p, err := NewParser(Config{}, &args)
|
|
require.NoError(t, err)
|
|
|
|
os.Args[0] = "example"
|
|
var help bytes.Buffer
|
|
p.WriteHelp(&help)
|
|
assert.Equal(t, expectedHelp, help.String())
|
|
}
|
|
|
|
func TestUsageWithProgramName(t *testing.T) {
|
|
expectedHelp := `usage: myprogram
|
|
|
|
options:
|
|
--help, -h display this help and exit
|
|
`
|
|
config := Config{
|
|
Program: "myprogram",
|
|
}
|
|
p, err := NewParser(config, &struct{}{})
|
|
require.NoError(t, err)
|
|
|
|
os.Args[0] = "example"
|
|
var help bytes.Buffer
|
|
p.WriteHelp(&help)
|
|
assert.Equal(t, expectedHelp, help.String())
|
|
}
|