-
Notifications
You must be signed in to change notification settings - Fork 156
/
render_test.go
35 lines (28 loc) · 857 Bytes
/
render_test.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
package main
import (
"os"
"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 := os.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)
}
}