mirror of
https://github.com/coder/coder.git
synced 2026-06-06 06:28:20 +00:00
Steven Masley
8b125d6c5d
* feat: Implement view for workspace builds to include rbac info * Removes the need to fetch the workspace to run an rbac check. * chore: Use workspace build as RBAC object * chore: Use golang templates instead of sqlc files
52 lines
1.0 KiB
Go
52 lines
1.0 KiB
Go
package dbtestutil
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/coderd/database"
|
|
"github.com/coder/coder/coderd/database/dbfake"
|
|
"github.com/coder/coder/coderd/database/postgres"
|
|
)
|
|
|
|
func UsingRealDatabase() bool {
|
|
return os.Getenv("DB") != ""
|
|
}
|
|
|
|
func NewDB(t *testing.T) (database.Store, database.Pubsub) {
|
|
t.Helper()
|
|
|
|
db := dbfake.New()
|
|
pubsub := database.NewPubsubInMemory()
|
|
if UsingRealDatabase() {
|
|
connectionURL := os.Getenv("CODER_PG_CONNECTION_URL")
|
|
if connectionURL == "" {
|
|
var (
|
|
err error
|
|
closePg func()
|
|
)
|
|
connectionURL, closePg, err = postgres.Open()
|
|
require.NoError(t, err)
|
|
t.Cleanup(closePg)
|
|
}
|
|
sqlDB, err := sql.Open("postgres", connectionURL)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
_ = sqlDB.Close()
|
|
})
|
|
db = database.New(sqlDB)
|
|
|
|
pubsub, err = database.NewPubsub(context.Background(), sqlDB, connectionURL)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
_ = pubsub.Close()
|
|
})
|
|
}
|
|
|
|
return db, pubsub
|
|
}
|