include type methods in generator test

everesio · Oct 23, 2018 · 73b546a · 73b546a
1 parent aae5d47
commit 73b546a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/dictionarygen/generator_test.go b/dictionarygen/generator_test.go
@@ -152,6 +152,17 @@ func generateGoDoc(source []byte) ([]byte, error) {
 				printCfg.Indent = 1
 				fmt.Fprintf(&b, "\n")
 			}
+
+			if len(t.Methods) > 0 {
+				printCfg.Indent = 2
+				for _, c := range t.Methods {
+					if err := printCfg.Fprint(&b, fs, c.Decl); err != nil {
+						return nil, err
+					}
+				}
+				printCfg.Indent = 1
+				fmt.Fprintf(&b, "\n")
+			}
 		}
 	}
 

diff --git a/dictionarygen/testdata/value-collision.expected b/dictionarygen/testdata/value-collision.expected
@@ -8,6 +8,7 @@ Types:
 			DSLTransmissionSystem_Value_ADSL2	DSLTransmissionSystem	= 2
 			DSLTransmissionSystem_Value_ADSL2Plus	DSLTransmissionSystem	= 3
 		)
+		func (a DSLTransmissionSystem) String() string
 Functions:
 		func DSLTransmissionSystem_Add(p *radius.Packet, value DSLTransmissionSystem) (err error)
 		func DSLTransmissionSystem_Del(p *radius.Packet)