mirror of
https://github.com/netbirdio/netbird.git
synced 2026-09-29 10:09:07 +02:00
Pascal Fischer
782c943410
* extract shared db conn + data repository * extract repository interface * protect against nested transactions * fix mysql and db conn creation * fix context management * remove query warpper * remove withContext and withLock wrapper * remove context from function call * remove in memory mode * fix nested transaction handling * use db directly * remove leftover test * remove pool close on error during conn creation
24 lines
718 B
Go
24 lines
718 B
Go
package dbtest
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/netbirdio/netbird/management/internals/shared/db"
|
|
)
|
|
|
|
// NewConn opens a fresh SQLite database in a temporary directory, migrates the
|
|
// given models and closes the connection when the test ends. It ignores
|
|
// NB_STORE_ENGINE_SQLITE_FILE, so a developer's configured database is never
|
|
// touched, and is safe to call from parallel tests.
|
|
func NewConn(t testing.TB, models ...any) *db.Conn {
|
|
t.Helper()
|
|
conn, err := db.OpenSqliteFile(context.Background(), t.TempDir(), db.SqliteFileName)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { _ = conn.Close() })
|
|
require.NoError(t, conn.AutoMigrate(models...))
|
|
return conn
|
|
}
|