33 lines
893 B
Go
33 lines
893 B
Go
package main
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestRootHasExplicitRoles(t *testing.T) {
|
|
root := newRootCommand()
|
|
if _, _, err := root.Find([]string{"agent"}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, _, err := root.Find([]string{"dispatcher"}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestRootAgentRejectsOldOneShotFlag(t *testing.T) {
|
|
root := newRootCommand()
|
|
root.SetArgs([]string{"agent", "--call-once"})
|
|
if err := root.Execute(); err == nil || !strings.Contains(err.Error(), "unknown flag") {
|
|
t.Fatalf("obsolete one-shot entry was admitted: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestRootDispatcherRejectsOldConfigFlag(t *testing.T) {
|
|
root := newRootCommand()
|
|
root.SetArgs([]string{"dispatcher", "--config", "/nonexistent"})
|
|
if err := root.Execute(); err == nil || !strings.Contains(err.Error(), "unknown flag") {
|
|
t.Fatalf("obsolete Dispatcher configuration entry was admitted: %v", err)
|
|
}
|
|
}
|