29 lines
740 B
Go
29 lines
740 B
Go
package helper
|
|
|
|
import (
|
|
"github.com/supabase/cli/pkg/migration"
|
|
"github.com/supabase/cli/pkg/pgtest"
|
|
)
|
|
|
|
func MockMigrationHistory(conn *pgtest.MockConn) *pgtest.MockConn {
|
|
conn.Query(migration.SET_LOCK_TIMEOUT).
|
|
Query(migration.CREATE_VERSION_SCHEMA).
|
|
Reply("CREATE SCHEMA").
|
|
Query(migration.CREATE_VERSION_TABLE).
|
|
Reply("CREATE TABLE").
|
|
Query(migration.ADD_STATEMENTS_COLUMN).
|
|
Reply("ALTER TABLE").
|
|
Query(migration.ADD_NAME_COLUMN).
|
|
Reply("ALTER TABLE")
|
|
return conn
|
|
}
|
|
|
|
func MockSeedHistory(conn *pgtest.MockConn) *pgtest.MockConn {
|
|
conn.Query(migration.SET_LOCK_TIMEOUT).
|
|
Query(migration.CREATE_VERSION_SCHEMA).
|
|
Reply("CREATE SCHEMA").
|
|
Query(migration.CREATE_SEED_TABLE).
|
|
Reply("CREATE TABLE")
|
|
return conn
|
|
}
|