From b91b2a94780f4e6b4d3b0c12fd9b5f4b05b1aa45 Mon Sep 17 00:00:00 2001 From: Anastasis Andronidis Date: Tue, 12 May 2015 15:45:00 +0200 Subject: Added a test for normalizing usage name [close #23] --- flag_test.go | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) (limited to 'flag_test.go') diff --git a/flag_test.go b/flag_test.go index efd6666..765dfe0 100644 --- a/flag_test.go +++ b/flag_test.go @@ -349,6 +349,31 @@ func TestCustomNormalizedNames(t *testing.T) { } } +// Every flag we add, the name (displayed also in usage) should normalized +func TestNormalizationFuncShouldChangeFlagName(t *testing.T) { + // Test normalization after addition + f := NewFlagSet("normalized", ContinueOnError) + + f.Bool("valid_flag", false, "bool value") + if f.Lookup("valid_flag").Name != "valid_flag" { + t.Error("The new flag should have the name 'valid_flag' instead of ", f.Lookup("valid_flag").Name) + } + + f.SetNormalizeFunc(wordSepNormalizeFunc) + if f.Lookup("valid_flag").Name != "valid.flag" { + t.Error("The new flag should have the name 'valid.flag' instead of ", f.Lookup("valid_flag").Name) + } + + // Test normalization before addition + f = NewFlagSet("normalized", ContinueOnError) + f.SetNormalizeFunc(wordSepNormalizeFunc) + + f.Bool("valid_flag", false, "bool value") + if f.Lookup("valid_flag").Name != "valid.flag" { + t.Error("The new flag should have the name 'valid.flag' instead of ", f.Lookup("valid_flag").Name) + } +} + // Declare a user-defined flag type. type flagVar []string -- cgit v1.2.3