mirror of
https://github.com/SecurityBrewery/catalyst.git
synced 2025-12-06 07:12:46 +01:00
28 lines
595 B
Go
28 lines
595 B
Go
package data
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/SecurityBrewery/catalyst/app/database"
|
|
"github.com/SecurityBrewery/catalyst/app/database/sqlc"
|
|
"github.com/SecurityBrewery/catalyst/app/migration"
|
|
"github.com/SecurityBrewery/catalyst/app/upload"
|
|
)
|
|
|
|
func NewTestDB(t *testing.T, dir string) *sqlc.Queries {
|
|
t.Helper()
|
|
|
|
queries := database.TestDB(t, dir)
|
|
uploader, err := upload.New(dir)
|
|
require.NoError(t, err)
|
|
|
|
err = migration.Apply(t.Context(), queries, dir, uploader)
|
|
require.NoError(t, err)
|
|
|
|
DefaultTestData(t, dir, queries)
|
|
|
|
return queries
|
|
}
|