From 8dd29d34bf0186945d53ba1ca0cde2324952a6e9 Mon Sep 17 00:00:00 2001 From: brettlangdon Date: Mon, 18 Jan 2016 13:42:04 -0500 Subject: Add support for environment variables --- usage_test.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'usage_test.go') diff --git a/usage_test.go b/usage_test.go index 2375e81..255018d 100644 --- a/usage_test.go +++ b/usage_test.go @@ -10,9 +10,9 @@ import ( ) func TestWriteUsage(t *testing.T) { - expectedUsage := "usage: example [--name NAME] [--value VALUE] [--verbose] [--dataset DATASET] [--optimize OPTIMIZE] [--ids IDS] INPUT [OUTPUT [OUTPUT ...]]\n" + expectedUsage := "usage: example [--name NAME] [--value VALUE] [--verbose] [--dataset DATASET] [--optimize OPTIMIZE] [--ids IDS] [--workers WORKERS] INPUT [OUTPUT [OUTPUT ...]]\n" - expectedHelp := `usage: example [--name NAME] [--value VALUE] [--verbose] [--dataset DATASET] [--optimize OPTIMIZE] [--ids IDS] INPUT [OUTPUT [OUTPUT ...]] + expectedHelp := `usage: example [--name NAME] [--value VALUE] [--verbose] [--dataset DATASET] [--optimize OPTIMIZE] [--ids IDS] [--workers WORKERS] INPUT [OUTPUT [OUTPUT ...]] positional arguments: input @@ -26,6 +26,8 @@ options: --optimize OPTIMIZE, -O OPTIMIZE optimization level --ids IDS Ids + --workers WORKERS, -w WORKERS + number of workers to start --help, -h display this help and exit ` var args struct { @@ -37,6 +39,7 @@ options: Dataset string `arg:"help:dataset to use"` Optimize int `arg:"-O,help:optimization level"` Ids []int64 `arg:"help:Ids"` + Workers int `arg:"-w,env:WORKERS,help:number of workers to start"` } args.Name = "Foo Bar" args.Value = 42 -- cgit v1.2.3