package repository_test import ( "context" "os" "path" "queryorchestration/internal/database/repository" "queryorchestration/internal/test" "testing" "github.com/stretchr/testify/assert" ) func TestDocument(t *testing.T) { ctx := context.Background() cfg := test.CreateBaseConfig() cfg.BasePath = path.Join(os.Getenv("PWD"), "../../..") _, cleanup := test.CreateDB(t, ctx, &test.CreateDatabaseConfig{ Cfg: cfg, RunMigrations: true, }) defer cleanup() queries := cfg.DBQueries clientId, err := queries.CreateClient(ctx, "example_client") assert.NoError(t, err) jobId, err := queries.CreateJob(ctx, clientId) assert.NoError(t, err) hash := "example_hash" id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{ Jobid: jobId, Hash: hash, }) assert.NoError(t, err) assert.NotEmpty(t, id) doc, err := queries.GetDocument(ctx, id) assert.NoError(t, err) assert.EqualExportedValues(t, &repository.Document{ ID: id, Jobid: jobId, Hash: hash, }, doc) }