Function TestMain

integration/integration_test.go:36–80

(m *testing.M)

)

func TestMain(m *testing.M) {
	var driver, dataSource string
	flag.StringVar(&driver, "driver", "", "integration database driver")
	flag.StringVar(&dataSource, "dataSource", "", "integration database data source")
	flag.Parse()

  if driver == "" {
    driver = "sqlite3"
  }

	if driver == "sqlite3" && dataSource == "" {
		dataSource = ":memory:"
	}

	db, err := sql.Open(driver, dataSource)
	if err != nil {
		fmt.Printf("error opening database: %v\n", err)
		os.Exit(-1)
	}

	_, err = db.Exec(testSchema)
	if err != nil {
		fmt.Printf("error creating test schema: %v\n", err)
		os.Exit(-2)
	}

	defer func() {
		_, err = db.Exec("DROP TABLE squirrel_integration")
		fmt.Printf("error removing test schema: %v\n", err)
	}()

	_, err = db.Exec(testData)
	if err != nil {
		fmt.Printf("error inserting test data: %v\n", err)
		os.Exit(-3)
	}

	sb = sqrl.StatementBuilder.RunWith(db)

	if driver == "postgres" {
		sb = sb.PlaceholderFormat(sqrl.Dollar)
	}

	os.Exit(m.Run())
}

func assertVals(t *testing.T, s sqrl.SelectBuilder, expected ...string) {
	rows, err := s.Query()