forked from smarty/gunit
/
generate.go
41 lines (36 loc) · 1.23 KB
/
generate.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
package generate
import (
"bytes"
"fmt"
"go/format"
"log"
"text/template"
"github.com/smartystreets/gunit/gunit/parse"
)
// TestFile generates complete source code for a _test.go file from the provided fixtures.
func TestFile(packageName string, fixtures []*parse.Fixture, checksum string, code map[string]string) ([]byte, error) {
buffer := bytes.NewBufferString(fmt.Sprintf(header, packageName))
buffer.WriteString("\n///////////////////////////////////////////////////////////////////////////////\n\n")
for _, fixture := range fixtures {
function, err := TestCases(fixture)
if err != nil {
return nil, err
}
buffer.Write(function)
buffer.WriteString("\n\n///////////////////////////////////////////////////////////////////////////////\n\n")
}
buffer.WriteString(fmt.Sprintf(footer, code, checksum))
return format.Source(buffer.Bytes())
}
func TestCases(fixture *parse.Fixture) ([]byte, error) {
writer := &bytes.Buffer{}
err := testFixtureTemplate.Execute(writer, fixture)
if err != nil {
log.Println(err)
return nil, err
}
return format.Source(writer.Bytes())
}
var testFixtureTemplate = template.Must(template.
New("testFunction").Funcs(map[string]interface{}{"sentence": toSentence}).
Parse(rawTestFunction))