GOMODULES
/
testify
mirror of https://github.com/stretchr/testify.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

run go/format on generated code

pull/361/head
Ernesto Jiménez 2016-09-25 23:00:40 +01:00
parent 75f4cfe6b3
commit 4b9bfb8845
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
_codegen/main.go
View File

@ -10,6 +10,7 @@ import (
"go/ast"
"go/build"
"go/doc"
"go/format"
"go/importer"
"go/parser"
"go/token"
@ -77,13 +78,18 @@ func generateCode(importer imports.Importer, funcs []testFunc) error {
}
}
code, err := format.Source(buff.Bytes())
if err != nil {
return err
}
// Write file
output, err := outputFile()
if err != nil {
return err
}
defer output.Close()
_, err = io.Copy(output, buff)
_, err = io.Copy(output, bytes.NewReader(code))
return err
}