sqlboiler/cmds/sqlboiler_test.go

90 lines
1.9 KiB
Go
Raw Normal View History

package cmds
2016-03-03 05:14:21 +01:00
import (
"bytes"
"io/ioutil"
"os"
"os/exec"
2016-03-03 05:14:21 +01:00
"testing"
"github.com/pobri19/sqlboiler/dbdrivers"
)
2016-03-27 17:03:14 +02:00
var cmdData *CmdData
func init() {
cmdData = &CmdData{
2016-03-23 07:18:41 +01:00
Tables: []dbdrivers.Table{
{
Name: "patrick_table",
Columns: []dbdrivers.Column{
{Name: "patrick_column", Type: "string", IsNullable: false},
{Name: "aaron_column", Type: "null.String", IsNullable: true},
{Name: "id", Type: "null.Int", IsNullable: true},
{Name: "fun_id", Type: "int64", IsNullable: false},
{Name: "time", Type: "null.Time", IsNullable: true},
{Name: "fun_time", Type: "time.Time", IsNullable: false},
{Name: "cool_stuff_forever", Type: "[]byte", IsNullable: false},
},
},
2016-03-23 07:18:41 +01:00
{
Name: "spiderman",
Columns: []dbdrivers.Column{
{Name: "patrick", Type: "string", IsNullable: false},
},
},
},
PkgName: "patrick",
OutFolder: "",
2016-03-23 07:18:41 +01:00
Interface: nil,
}
}
2016-03-03 05:14:21 +01:00
func TestTemplates(t *testing.T) {
if testing.Short() {
t.SkipNow()
}
// Initialize the templates
var err error
2016-03-27 17:03:14 +02:00
cmdData.Templates, err = loadTemplates("templates")
if err != nil {
2016-03-04 02:57:22 +01:00
t.Fatalf("Unable to initialize templates: %s", err)
}
if len(cmdData.Templates) == 0 {
t.Errorf("Templates is empty.")
}
cmdData.TestTemplates, err = loadTemplates("templates_test")
if err != nil {
t.Fatalf("Unable to initialize templates: %s", err)
}
if len(cmdData.Templates) == 0 {
t.Errorf("Templates is empty.")
}
cmdData.OutFolder, err = ioutil.TempDir("", "templates")
if err != nil {
t.Fatalf("Unable to create tempdir: %s", err)
}
defer os.RemoveAll(cmdData.OutFolder)
if err = cmdData.run(true); err != nil {
t.Errorf("Unable to run SQLBoilerRun: %s", err)
}
buf := bytes.Buffer{}
buf2 := bytes.Buffer{}
cmd := exec.Command("go", "test")
cmd.Dir = cmdData.OutFolder
cmd.Stderr = &buf
cmd.Stdout = &buf2
if err = cmd.Run(); err != nil {
2016-03-04 02:57:22 +01:00
t.Errorf("go test cmd execution failed: %s\n\n%s", err, buf.String())
}
2016-03-03 05:14:21 +01:00
}