forked from vikunja/vikunja
jtojnar
ce5be947b4
Revert fixture fixes for postgres Use postgres connection string with spaces instead of url Fix label order Make postgres tests in ci less verbose Add sequence update script Skip resets in postgres Remove option to skip resets in postgres Make postgres tests in ci verboseq Update test fixtures database Fix file tests on postgres Add postgres options to sample config Make sure tests init test fixtures before running the actual tests Fix issues with IDs too big to fit in an int Fix duplicate auto incremented IDs Refactor / Fix team tests Refactor team member tests Fix team member create Fix label test Fix getting labels Fix test fixtures for postgresql Fix connection string params Disable ssl mode on postgres integration tests Disable ssl mode on postgres tests Use sprintf to create the connection string for postgresql fixup! Add postgres support Add postgres support Added generate as a make dependency for make build Clarify docs on building Co-authored-by: kolaente <k@knt.li> Co-authored-by: Jan Tojnar <jtojnar@gmail.com> Reviewed-on: vikunja/api#135
132 lines
2.4 KiB
Go
132 lines
2.4 KiB
Go
package testfixtures
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
)
|
|
|
|
type mySQL struct {
|
|
baseHelper
|
|
tables []string
|
|
tablesChecksum map[string]int64
|
|
}
|
|
|
|
func (h *mySQL) init(db *sql.DB) error {
|
|
var err error
|
|
h.tables, err = h.tableNames(db)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (*mySQL) paramType() int {
|
|
return paramTypeQuestion
|
|
}
|
|
|
|
func (*mySQL) quoteKeyword(str string) string {
|
|
return fmt.Sprintf("`%s`", str)
|
|
}
|
|
|
|
func (*mySQL) databaseName(q queryable) (string, error) {
|
|
var dbName string
|
|
err := q.QueryRow("SELECT DATABASE()").Scan(&dbName)
|
|
return dbName, err
|
|
}
|
|
|
|
func (h *mySQL) tableNames(q queryable) ([]string, error) {
|
|
query := `
|
|
SELECT table_name
|
|
FROM information_schema.tables
|
|
WHERE table_schema = ?
|
|
AND table_type = 'BASE TABLE';
|
|
`
|
|
dbName, err := h.databaseName(q)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
rows, err := q.Query(query, dbName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var tables []string
|
|
for rows.Next() {
|
|
var table string
|
|
if err = rows.Scan(&table); err != nil {
|
|
return nil, err
|
|
}
|
|
tables = append(tables, table)
|
|
}
|
|
if err = rows.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
return tables, nil
|
|
|
|
}
|
|
|
|
func (h *mySQL) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) (err error) {
|
|
tx, err := db.Begin()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer tx.Rollback()
|
|
|
|
if _, err = tx.Exec("SET FOREIGN_KEY_CHECKS = 0"); err != nil {
|
|
return err
|
|
}
|
|
|
|
err = loadFn(tx)
|
|
_, err2 := tx.Exec("SET FOREIGN_KEY_CHECKS = 1")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err2 != nil {
|
|
return err2
|
|
}
|
|
|
|
return tx.Commit()
|
|
}
|
|
|
|
func (h *mySQL) isTableModified(q queryable, tableName string) (bool, error) {
|
|
checksum, err := h.getChecksum(q, tableName)
|
|
if err != nil {
|
|
return true, err
|
|
}
|
|
|
|
oldChecksum := h.tablesChecksum[tableName]
|
|
|
|
return oldChecksum == 0 || checksum != oldChecksum, nil
|
|
}
|
|
|
|
func (h *mySQL) afterLoad(q queryable) error {
|
|
if h.tablesChecksum != nil {
|
|
return nil
|
|
}
|
|
|
|
h.tablesChecksum = make(map[string]int64, len(h.tables))
|
|
for _, t := range h.tables {
|
|
checksum, err := h.getChecksum(q, t)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
h.tablesChecksum[t] = checksum
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (h *mySQL) getChecksum(q queryable, tableName string) (int64, error) {
|
|
sql := fmt.Sprintf("CHECKSUM TABLE %s", h.quoteKeyword(tableName))
|
|
var (
|
|
table string
|
|
checksum int64
|
|
)
|
|
if err := q.QueryRow(sql).Scan(&table, &checksum); err != nil {
|
|
return 0, err
|
|
}
|
|
return checksum, nil
|
|
}
|