51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package cli
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestRunHelpLongFlag(t *testing.T) {
|
|
var stdout bytes.Buffer
|
|
var stderr bytes.Buffer
|
|
|
|
err := Run(context.Background(), []string{"--help"}, &stdout, &stderr)
|
|
if err != nil {
|
|
t.Fatalf("Run() error = %v", err)
|
|
}
|
|
|
|
if !strings.Contains(stdout.String(), "Usage:") {
|
|
t.Fatalf("help output missing usage:\n%s", stdout.String())
|
|
}
|
|
}
|
|
|
|
func TestRunHelpShortFlag(t *testing.T) {
|
|
var stdout bytes.Buffer
|
|
var stderr bytes.Buffer
|
|
|
|
err := Run(context.Background(), []string{"-h"}, &stdout, &stderr)
|
|
if err != nil {
|
|
t.Fatalf("Run() error = %v", err)
|
|
}
|
|
|
|
if !strings.Contains(stdout.String(), "weatherreporter --help") {
|
|
t.Fatalf("help output missing root help command:\n%s", stdout.String())
|
|
}
|
|
}
|
|
|
|
func TestRunUnknownArgument(t *testing.T) {
|
|
var stdout bytes.Buffer
|
|
var stderr bytes.Buffer
|
|
|
|
err := Run(context.Background(), []string{"generate"}, &stdout, &stderr)
|
|
if err == nil {
|
|
t.Fatal("Run() error = nil, want unknown argument error")
|
|
}
|
|
|
|
if !strings.Contains(err.Error(), `unknown argument "generate"`) {
|
|
t.Fatalf("Run() error = %q, want unknown argument message", err.Error())
|
|
}
|
|
}
|