snapshot of https://github.com/ory/kratos.git (master, 2026-09-15; blobs > 131072B omitted)

oryx/dbal/canonicalize.go

@@ -0,0 +1,26 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package dbal
+
+const (
+	DriverMySQL       = "mysql"
+	DriverCockroachDB = "cockroach"
+	DriverPostgreSQL  = "postgres"
+	// DriverYugabyteDB is the dialect name YugabyteDB connections report. It is
+	// a commercial (Ory Enterprise License) dialect that speaks the PostgreSQL
+	// wire protocol and SQL dialect, so most SQL paths treat it like postgres.
+	DriverYugabyteDB = "yugabyte"
+)
+
+// IsPostgresCompatible reports whether dialect speaks the PostgreSQL SQL
+// dialect. It does not imply support for CockroachDB-specific syntax or
+// features.
+func IsPostgresCompatible(dialect string) bool {
+	switch dialect {
+	case DriverPostgreSQL, DriverCockroachDB, DriverYugabyteDB:
+		return true
+	default:
+		return false
+	}
+}

oryx/dbal/driver.go

@@ -0,0 +1,14 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package dbal
+
+import (
+	"strings"
+)
+
+// IsSQLite returns true if the connection is a SQLite string.
+func IsSQLite(dsn string) bool {
+	scheme := strings.Split(dsn, "://")[0]
+	return scheme == "sqlite" || scheme == "sqlite3"
+}

oryx/dbal/dsn.go

@@ -0,0 +1,47 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package dbal
+
+import (
+	"fmt"
+	"os"
+	"regexp"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+var sqliteMemoryRegexp = regexp.MustCompile(`^sqlite://file:.+\?.*&?mode=memory($|&.*)|sqlite://(file:)?:memory:\?.*$|^(:memory:|memory)$`)
+
+// IsMemorySQLite returns true if a given DSN string is pointing to a SQLite database.
+//
+// SQLite can be written in different styles depending on the use case
+// - just in memory
+// - shared connection
+// - shared but unique in the same process
+// see: https://sqlite.org/inmemorydb.html
+func IsMemorySQLite(dsn string) bool { return sqliteMemoryRegexp.MatchString(dsn) }
+
+// NewSQLiteTestDatabase creates a new, unique SQLite database
+// which is shared amongst all callers and identified by an individual file name.
+// The database file is created in the system's temporary directory, and not actively
+// removed to allow debugging in case of test failures.
+func NewSQLiteTestDatabase(t testing.TB) string {
+	fn, err := os.MkdirTemp("", "sqlite-test-db-*")
+	require.NoError(t, err)
+	return NewSQLiteDatabase(fn)
+}
+
+// NewSQLiteDatabase creates a new on-disk, unique SQLite database
+// which is shared amongst all callers and identified by an individual file name.
+// This is sometimes necessary over a in-memory database, for example when multiple tests/goroutines run in parallel
+// and access the same table.
+// This would result in a locking error from SQLite when running in-memory.
+// Additionally, shared cache mode is deprecated and discouraged, and the problem is better solved with the WAL,
+// according to official docs.
+//
+// In tests, use [NewSQLiteTestDatabase] instead.
+func NewSQLiteDatabase(name string) string {
+	return fmt.Sprintf("sqlite://%s/db.sqlite?_fk=true&_busy_timeout=100000&_pragma=journal_mode(WAL)&_time_format=sqlite", name)
+}

oryx/dbal/testhelpers.go

@@ -0,0 +1,62 @@
+// Copyright © 2026 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package dbal
+
+import (
+	"bytes"
+	"encoding/hex"
+	"fmt"
+	"io/fs"
+	"os"
+	"path/filepath"
+	"regexp"
+	"testing"
+
+	"github.com/pkg/errors"
+	"github.com/stretchr/testify/require"
+
+	"github.com/ory/pop/v6"
+	"github.com/ory/x/fsx"
+	"github.com/ory/x/sqlcon/dockertest"
+)
+
+var hashDumpRegex = regexp.MustCompile(`-- migrations hash: ([^\n]+)\n`)
+
+func RestoreFromSchemaDump(t testing.TB, c *pop.Connection, migrations fs.FS, writeTo string) func(testing.TB) {
+	newHash, err := fsx.DirHash(migrations)
+	require.NoError(t, err)
+
+	dumpFilename := filepath.Join(writeTo, c.Dialect.Name()+"_dump.sql")
+
+	updateDump := func(t testing.TB) {
+		dump := dockertest.DumpSchema(t, c)
+		f, err := os.OpenFile(dumpFilename, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
+		require.NoError(t, err)
+		defer f.Close()
+
+		_, _ = fmt.Fprintf(f, "-- migrations hash: %x\n\n%s", newHash, dump)
+		t.Fatal("database schema restored from migrations and dump updated, please re-run the test")
+	}
+
+	dump, err := os.ReadFile(dumpFilename)
+	if errors.Is(err, fs.ErrNotExist) {
+		return updateDump
+	}
+	require.NoError(t, err)
+
+	matches := hashDumpRegex.FindSubmatch(dump)
+	if len(matches) != 2 {
+		return updateDump
+	}
+
+	currentHash, err := hex.DecodeString(string(matches[1]))
+	require.NoError(t, err)
+
+	if !bytes.Equal(newHash, currentHash) {
+		return updateDump
+	}
+
+	require.NoError(t, c.RawQuery(string(dump)).Exec())
+	return nil
+}