root
44406b72db
Moteva-style AI design workbench replica. Home prompt creates a project; the project page provides an infinite canvas with node drag, zoom/pan, a design chat panel, history replay, image asset upload, and project save/regenerate. - Backend: go-zero, DDD layering, sqlc/pgx, optional PostgreSQL; memory or Redis cache; asynq job queue; MinIO/S3/R2/OSS object storage; sky-valley/pi agent runtime adapter. - Frontend: Next.js App Router + Vite artifact build, TypeScript, i18n, shadcn/ui components, auth (OTP/Turnstile/Google/WeChat). - Deploy: Docker Compose and k3s manifests. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
72 lines
2.4 KiB
Go
72 lines
2.4 KiB
Go
package job
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"img_infinite_canvas/internal/domain/design"
|
|
|
|
"github.com/hibiken/asynq"
|
|
)
|
|
|
|
type fakeJobProcessor struct {
|
|
job design.Job
|
|
failedJob design.Job
|
|
failure error
|
|
}
|
|
|
|
func (p *fakeJobProcessor) ProcessJob(ctx context.Context, job design.Job) error {
|
|
p.job = job
|
|
return nil
|
|
}
|
|
|
|
func (p *fakeJobProcessor) HandleJobFailure(ctx context.Context, job design.Job, err error) error {
|
|
p.failedJob = job
|
|
p.failure = err
|
|
return nil
|
|
}
|
|
|
|
func TestAsynqWorkerDispatchesJobPayload(t *testing.T) {
|
|
processor := &fakeJobProcessor{}
|
|
worker, err := NewAsynqWorker(AsynqOptions{}, processor)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
task := asynq.NewTask(string(design.JobTextExtraction), []byte(`{"Kind":"canvas:text-extraction","ProjectID":"project-1","ThreadID":"thread-1","Feature":"edit-text"}`))
|
|
if err := worker.handle(processor)(context.Background(), task); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if processor.job.Kind != design.JobTextExtraction || processor.job.ProjectID != "project-1" || processor.job.Feature != "edit-text" {
|
|
t.Fatalf("unexpected dispatched job: %#v", processor.job)
|
|
}
|
|
}
|
|
|
|
func TestAsynqFailureHandlerReportsVisualGenerationFailures(t *testing.T) {
|
|
processor := &fakeJobProcessor{}
|
|
task := asynq.NewTask(string(design.JobCreateProjectGeneration), []byte(`{"Kind":"canvas:create-project-generation","ProjectID":"project-1","ThreadID":"thread-1"}`))
|
|
failure := context.DeadlineExceeded
|
|
|
|
handleAsynqTaskFailure(context.Background(), task, failure, processor)
|
|
|
|
if processor.failure == nil {
|
|
t.Fatal("expected visual generation failure to be reported")
|
|
}
|
|
if processor.failedJob.Kind != design.JobCreateProjectGeneration || processor.failedJob.ProjectID != "project-1" || processor.failedJob.ThreadID != "thread-1" {
|
|
t.Fatalf("unexpected failed job: %#v", processor.failedJob)
|
|
}
|
|
}
|
|
|
|
func TestMaxRetryForVisualGenerationJobsDisablesQueueRetry(t *testing.T) {
|
|
if got := maxRetryForJob(design.Job{Kind: design.JobCreateProjectGeneration}, 3); got != 0 {
|
|
t.Fatalf("expected visual generation max retry 0, got %d", got)
|
|
}
|
|
if got := maxRetryForJob(design.Job{Kind: design.JobMockupModelGeneration}, 3); got != 0 {
|
|
t.Fatalf("expected mockup model generation max retry 0, got %d", got)
|
|
}
|
|
if got := maxRetryForJob(design.Job{Kind: design.JobTextExtraction}, 3); got != 3 {
|
|
t.Fatalf("expected non visual generation max retry passthrough, got %d", got)
|
|
}
|
|
}
|