forgejo-federation/vendor/github.com/go-testfixtures/testfixtures/v3/helper.go

72 lines
1.6 KiB
Go
Raw Normal View History

2016-12-29 06:33:40 +05:30
package testfixtures
import (
"database/sql"
"fmt"
)
const (
paramTypeDollar = iota + 1
paramTypeQuestion
paramTypeAtSign
2016-12-29 06:33:40 +05:30
)
type loadFunction func(tx *sql.Tx) error
type helper interface {
2016-12-29 06:33:40 +05:30
init(*sql.DB) error
disableReferentialIntegrity(*sql.DB, loadFunction) error
paramType() int
databaseName(queryable) (string, error)
tableNames(queryable) ([]string, error)
isTableModified(queryable, string) (bool, error)
afterLoad(queryable) error
2016-12-29 06:33:40 +05:30
quoteKeyword(string) string
whileInsertOnTable(*sql.Tx, string, func() error) error
}
type queryable interface {
Exec(string, ...interface{}) (sql.Result, error)
Query(string, ...interface{}) (*sql.Rows, error)
QueryRow(string, ...interface{}) *sql.Row
}
// batchSplitter is an interface with method which returns byte slice for
// splitting SQL batches. This need to split sql statements and run its
// separately.
//
// For Microsoft SQL Server batch splitter is "GO". For details see
// https://docs.microsoft.com/en-us/sql/t-sql/language-elements/sql-server-utilities-statements-go
type batchSplitter interface {
splitter() []byte
}
var (
_ helper = &mySQL{}
_ helper = &postgreSQL{}
_ helper = &sqlite{}
_ helper = &sqlserver{}
)
2016-12-29 06:33:40 +05:30
type baseHelper struct{}
func (baseHelper) init(_ *sql.DB) error {
2016-12-29 06:33:40 +05:30
return nil
}
func (baseHelper) quoteKeyword(str string) string {
2016-12-29 06:33:40 +05:30
return fmt.Sprintf(`"%s"`, str)
}
func (baseHelper) whileInsertOnTable(_ *sql.Tx, _ string, fn func() error) error {
2016-12-29 06:33:40 +05:30
return fn()
}
func (baseHelper) isTableModified(_ queryable, _ string) (bool, error) {
return true, nil
}
func (baseHelper) afterLoad(_ queryable) error {
return nil
}