sqlboiler/templates_test/main_test/mssql_main.tpl

type mssqlTester struct {
	dbConn     *sql.DB
	dbName     string
	host       string
	user       string
	pass       string
	sslmode    string
	port       int
	testDBName string
}

func init() {
	dbMain = &mssqlTester{}
}

func (m *mssqlTester) setup() error {
	var err error
	m.dbName = viper.GetString("mssql.dbname")
	m.host = viper.GetString("mssql.host")
	m.user = viper.GetString("mssql.user")
	m.pass = viper.GetString("mssql.pass")
	m.port = viper.GetInt("mssql.port")
	m.sslmode = viper.GetString("mssql.sslmode")
	// Create a randomized db name.
	m.testDBName = randomize.StableDBName(m.dbName)

	if err = m.dropTestDB(); err != nil {
		return err
	}
	if err = m.createTestDB(); err != nil {
		return err
	}

	createCmd := exec.Command("sqlcmd", "-S", m.host, "-U", m.user, "-P", m.pass, "-d", m.testDBName)

	f, err := os.Open("tables_schema.sql")
	if err != nil {
		return errors.Wrap(err, "failed to open tables_schema.sql file")
	}

	defer f.Close()

	createCmd.Stdin = newFKeyDestroyer(rgxMSSQLkey, f)

	if err = createCmd.Start(); err != nil {
		return errors.Wrap(err, "failed to start sqlcmd command")
	}

	if err = createCmd.Wait(); err != nil {
		fmt.Println(err)
		return errors.Wrap(err, "failed to wait for sqlcmd command")
	}

	return nil
}

func (m *mssqlTester) sslMode(mode string) string {
	switch mode {
	case "true":
		return "true"
	case "false":
		return "false"
	default:
		return "disable"
	}
}

func (m *mssqlTester) createTestDB() error {
	sql := fmt.Sprintf(`
	CREATE DATABASE %s;
	GO`, m.testDBName)
	return m.runCmd(sql, "sqlcmd", "-S", m.host, "-U", m.user, "-P", m.pass)
}

func (m *mssqlTester) dropTestDB() error {
	// Since MS SQL 2016 it can be done with
	// DROP DATABASE [ IF EXISTS ] { database_name | database_snapshot_name } [ ,...n ] [;]
	sql := fmt.Sprintf(`
	IF EXISTS(SELECT name FROM sys.databases 
		WHERE name = '%s')
		DROP DATABASE %s
	GO`, m.testDBName, m.testDBName)
	return m.runCmd(sql, "sqlcmd", "-S", m.host, "-U", m.user, "-P", m.pass)
}

func (m *mssqlTester) teardown() error {
	if m.dbConn != nil {
		m.dbConn.Close()
	}

	if err := m.dropTestDB(); err != nil {
		return err
	}

	return nil
}

func (m *mssqlTester) runCmd(stdin, command string, args ...string) error {
	cmd := exec.Command(command, args...)
	cmd.Stdin = strings.NewReader(stdin)

	stdout := &bytes.Buffer{}
	stderr := &bytes.Buffer{}
	cmd.Stdout = stdout
	cmd.Stderr = stderr
	if err := cmd.Run(); err != nil {
		fmt.Println("failed running:", command, args)
		fmt.Println(stdout.String())
		fmt.Println(stderr.String())
		return err
	}

	return nil
}

func (m *mssqlTester) conn() (*sql.DB, error) {
	if m.dbConn != nil {
		return m.dbConn, nil
	}

	var err error
	m.dbConn, err = sql.Open("mssql", drivers.MSSQLBuildQueryString(m.user, m.pass, m.testDBName, m.host, m.port, m.sslmode))
	if err != nil {
		return nil, err
	}

	return m.dbConn, nil
}