140 lines
2.7 KiB
Go
140 lines
2.7 KiB
Go
package boilingcore
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
var state *State
|
|
var rgxHasSpaces = regexp.MustCompile(`^\s+`)
|
|
|
|
func TestNew(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
}
|
|
|
|
var err error
|
|
out, err := ioutil.TempDir("", "boil_templates")
|
|
if err != nil {
|
|
t.Fatalf("unable to create tempdir: %s", err)
|
|
}
|
|
|
|
// Defer cleanup of the tmp folder
|
|
defer func() {
|
|
if t.Failed() {
|
|
t.Log("template test output:", state.Config.OutFolder)
|
|
return
|
|
}
|
|
os.RemoveAll(state.Config.OutFolder)
|
|
}()
|
|
|
|
config := &Config{
|
|
DriverName: "mock",
|
|
PkgName: "models",
|
|
OutFolder: out,
|
|
BlacklistTables: []string{"hangars"},
|
|
}
|
|
|
|
state, err = New(config)
|
|
if err != nil {
|
|
t.Fatalf("Unable to create State using config: %s", err)
|
|
}
|
|
|
|
if err = state.Run(false); err != nil {
|
|
t.Errorf("Unable to execute State.Run: %s", err)
|
|
}
|
|
|
|
buf := &bytes.Buffer{}
|
|
|
|
cmd := exec.Command("go", "test", "-c")
|
|
cmd.Dir = state.Config.OutFolder
|
|
cmd.Stderr = buf
|
|
|
|
if err = cmd.Run(); err != nil {
|
|
t.Errorf("go test cmd execution failed: %s", err)
|
|
outputCompileErrors(buf, state.Config.OutFolder)
|
|
fmt.Println()
|
|
}
|
|
}
|
|
|
|
func outputCompileErrors(buf *bytes.Buffer, outFolder string) {
|
|
type errObj struct {
|
|
errMsg string
|
|
fileName string
|
|
lineNumber int
|
|
}
|
|
|
|
var errObjects []errObj
|
|
lineBuf := &bytes.Buffer{}
|
|
|
|
bufLines := bytes.Split(buf.Bytes(), []byte{'\n'})
|
|
for i := 0; i < len(bufLines); i++ {
|
|
lineBuf.Reset()
|
|
if !bytes.HasPrefix(bufLines[i], []byte("./")) {
|
|
continue
|
|
}
|
|
|
|
fmt.Fprintf(lineBuf, "%s\n", bufLines[i])
|
|
|
|
splits := bytes.Split(bufLines[i], []byte{':'})
|
|
lineNum, err := strconv.Atoi(string(splits[1]))
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Cant convert line number to int: %s", bufLines[i]))
|
|
}
|
|
|
|
eObj := errObj{
|
|
fileName: string(splits[0]),
|
|
lineNumber: lineNum,
|
|
}
|
|
|
|
for y := i; y < len(bufLines); y++ {
|
|
if !rgxHasSpaces.Match(bufLines[y]) {
|
|
break
|
|
}
|
|
fmt.Fprintf(lineBuf, "%s\n", bufLines[y])
|
|
i++
|
|
}
|
|
|
|
eObj.errMsg = lineBuf.String()
|
|
|
|
errObjects = append(errObjects, eObj)
|
|
}
|
|
|
|
for _, eObj := range errObjects {
|
|
fmt.Printf("-----------------\n")
|
|
fmt.Println(eObj.errMsg)
|
|
|
|
filePath := filepath.Join(outFolder, eObj.fileName)
|
|
fh, err := os.Open(filePath)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Cant open the file: %#v", eObj))
|
|
}
|
|
|
|
scanner := bufio.NewScanner(fh)
|
|
throwaway := eObj.lineNumber - 5
|
|
for throwaway > 0 && scanner.Scan() {
|
|
throwaway--
|
|
}
|
|
|
|
for i := 0; i < 6; i++ {
|
|
if scanner.Scan() {
|
|
b := scanner.Bytes()
|
|
if len(b) != 0 {
|
|
fmt.Printf("%s\n", b)
|
|
} else {
|
|
i--
|
|
}
|
|
}
|
|
}
|
|
|
|
fh.Close()
|
|
}
|
|
}
|