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
|
package main
import (
"io/ioutil"
"regexp"
"testing"
"bytes"
)
func TestRender(t *testing.T) {
buf := &bytes.Buffer{}
if err := render(buf); err != nil {
t.Fatal(err.Error())
}
generatedString := buf.String()
existingFilePath := "../jen/generated.go"
existingBytes, err := ioutil.ReadFile(existingFilePath)
if err != nil {
t.Fatal(err.Error())
}
existingString := string(existingBytes)
// The "goimports" tool will often re-order the imports, so this is a
// kludge to remove it before comparing. This is not ideal!
importsRegex := regexp.MustCompile(`(?ms:\nimport \(\n.*\n\)\n)`)
generatedString = importsRegex.ReplaceAllString(generatedString, "-")
existingString = importsRegex.ReplaceAllString(existingString, "-")
if generatedString != existingString {
t.Fatalf("Generated code is not what is present:\n%s", generatedString)
}
}
|
