aides-repo-api/tests/integration/utils.go

package integration

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"

	"code.aides.space/aides-infra/aides-repo-api/internal/config"
	"code.aides.space/aides-infra/aides-repo-api/internal/models"
)

func prepareDb(t *testing.T) *gorm.DB {
	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
	assert.NoError(t, err)

	err = db.AutoMigrate(&models.Task{}, &models.GitRepoAltRepoTask{}, &models.RPMFile{})
	assert.NoError(t, err)

	return db
}

func prepareConfig() *config.Config {
	return &config.Config{
		UploadDir: "./test_uploads",
	}
}