GOMODULES/testify
1
0
Fork 0
mirror of https://github.com/stretchr/testify.git synced 2025-04-27 13:13:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

run go/format on generated code

Browse Source
This commit is contained in:
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/ast"
"go/build" "go/build"
"go/doc" "go/doc"
"go/format"
"go/importer" "go/importer"
"go/parser" "go/parser"
"go/token" "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 // Write file
output, err := outputFile() output, err := outputFile()
if err != nil { if err != nil {
return err return err
} }
defer output.Close() defer output.Close()
_, err = io.Copy(output, buff) _, err = io.Copy(output, bytes.NewReader(code))
return err return err
} }