Allow debug to be toggled by a flag.

This commit is contained in:
Aaron L 2016-07-14 22:44:52 -07:00
parent 726fae9af1
commit 95ef4ce74b
2 changed files with 4 additions and 1 deletions
imports.go
templates_test/main_test

View file

@ -209,6 +209,7 @@ var defaultTestMainImports = map[string]imports{
`"testing"`, `"testing"`,
`"os"`, `"os"`,
`"os/exec"`, `"os/exec"`,
`"flag"`,
`"fmt"`, `"fmt"`,
`"io/ioutil"`, `"io/ioutil"`,
`"bytes"`, `"bytes"`,

View file

@ -11,11 +11,13 @@ type Config struct {
Postgres PostgresCfg `toml:"postgres"` Postgres PostgresCfg `toml:"postgres"`
} }
var flagDebugMode = flag.Bool("test.sqldebug", false, "Turns on debug mode for SQL statements")
func TestMain(m *testing.M) { func TestMain(m *testing.M) {
rand.Seed(time.Now().UnixNano()) rand.Seed(time.Now().UnixNano())
// Set DebugMode so we can see generated sql statements // Set DebugMode so we can see generated sql statements
boil.DebugMode = false boil.DebugMode = *flagDebugMode
var err error var err error
if err = setup(); err != nil { if err = setup(); err != nil {