mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 11:34:26 +00:00
5132ebe07c
* feat: add tenant column to eventstore * feat: read tenant from context on push and filter * Update 07_events_table.sql * pass tenant to queryFactory * fix some query tests * init in tests * add missing sql files Co-authored-by: Livio Amstutz <livio.a@gmail.com>
60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package sql
|
|
|
|
import (
|
|
"database/sql"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/caos/logging"
|
|
"github.com/cockroachdb/cockroach-go/v2/testserver"
|
|
|
|
"github.com/caos/zitadel/cmd/admin/initialise"
|
|
)
|
|
|
|
var (
|
|
testCRDBClient *sql.DB
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
ts, err := testserver.NewTestServer()
|
|
if err != nil {
|
|
logging.WithFields("error", err).Fatal("unable to start db")
|
|
}
|
|
|
|
testCRDBClient, err = sql.Open("postgres", ts.PGURL().String())
|
|
if err != nil {
|
|
logging.WithFields("error", err).Fatal("unable to connect to db")
|
|
}
|
|
if err = testCRDBClient.Ping(); err != nil {
|
|
logging.WithFields("error", err).Fatal("unable to ping db")
|
|
}
|
|
|
|
defer func() {
|
|
testCRDBClient.Close()
|
|
ts.Stop()
|
|
}()
|
|
|
|
if err = initDB(testCRDBClient); err != nil {
|
|
logging.WithFields("error", err).Fatal("migrations failed")
|
|
}
|
|
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
func initDB(db *sql.DB) error {
|
|
username := "zitadel"
|
|
database := "zitadel"
|
|
err := initialise.Initialise(db, initialise.VerifyUser(username, ""),
|
|
initialise.VerifyDatabase(database),
|
|
initialise.VerifyGrant(database, username))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return initialise.VerifyZitadel(db)
|
|
}
|
|
|
|
func fillUniqueData(unique_type, field string) error {
|
|
_, err := testCRDBClient.Exec("INSERT INTO eventstore.unique_constraints (unique_type, unique_field) VALUES ($1, $2)", unique_type, field)
|
|
return err
|
|
}
|