sqlboiler/sqlboiler_test.go

141 lines
2.6 KiB
Go
Raw Normal View History

package main
2016-03-03 05:14:21 +01:00
import (
"bufio"
"bytes"
"fmt"
"io/ioutil"
"os"
"os/exec"
"path/filepath"
"regexp"
"strconv"
2016-03-03 05:14:21 +01:00
"testing"
)
var state *State
var rgxHasSpaces = regexp.MustCompile(`^\s+`)
2016-03-27 17:03:14 +02:00
func TestNew(t *testing.T) {
2016-03-03 05:14:21 +01:00
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{
2016-09-09 07:41:57 +02:00
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()
}
2016-03-03 05:14:21 +01:00
}