Files
gochat/backend/internal/channel/dispatcher_worker_test.go
T
rogee aeddedf2a3 Reorganize repo: backend/, deploy/, docs/ layout + AGENTS.md
Restructure the monorepo into clear top-level directories:
- backend/: Go module root (cmd, internal, pkg, configs, migrations,
  docs/swagger, scripts, tests, go.mod, Makefile, .air.toml)
- deploy/: Docker (Dockerfile, docker-compose*), quickstart, fluentd
- docs/: project documentation + reports/ (moved from repo root)
- AGENTS.md: new AI coding-agent guide at repo root

Update all references to the new layout:
- Dockerfile: COPY backend/go.mod, COPY backend/ (context = repo root)
- docker-compose files: context ../.., dockerfile deploy/docker/Dockerfile,
  env_file ../../.env, volume mounts ../../backend:/app
- deploy/quickstart/compose.yaml: dockerfile deploy/docker/Dockerfile
- CI: working-directory: backend for go commands, file deploy/docker/Dockerfile,
  coverage path backend/coverage.out, health_check backend/scripts/
- backend/Makefile: docker target uses -f ../deploy/docker/Dockerfile ../
- README: architecture tree, quickstart, config paths updated

Move root stray scripts (rename_models.*, run_m11_tests.sh, verify_build.sh,
gorm_bool_main.go) to backend/scripts/legacy/. All moves via git mv to
preserve history. Build, vet, SQLite tests, and docker compose config verified.
2026-07-07 14:44:12 +08:00

80 lines
2.2 KiB
Go

package channel
import (
"context"
"sync/atomic"
"testing"
"time"
"github.com/gochat/gochat/internal/model"
"github.com/gochat/gochat/internal/worker"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
"gorm.io/gorm/logger"
)
type workerDispatchListener struct {
name string
count atomic.Int32
}
func (l *workerDispatchListener) Name() string { return l.name }
func (l *workerDispatchListener) OnEvent(ctx context.Context, event *ChannelEvent) error {
l.count.Add(1)
return nil
}
func newChannelWorkerDB(t *testing.T) *gorm.DB {
t.Helper()
db, err := gorm.Open(sqlite.Open("file:channel-dispatcher-worker?mode=memory&cache=shared"), &gorm.Config{Logger: logger.Default.LogMode(logger.Silent)})
if err != nil {
t.Fatalf("open sqlite: %v", err)
}
sqlDB, err := db.DB()
if err != nil {
t.Fatalf("sqlite db handle: %v", err)
}
sqlDB.SetMaxOpenConns(1)
if err := db.AutoMigrate(&model.BackgroundJob{}); err != nil {
t.Fatalf("migrate background jobs: %v", err)
}
t.Cleanup(func() {
db.Exec("DELETE FROM background_jobs")
sqlDB.Close()
})
return db
}
func TestDispatcherDispatchAsyncEnqueuesDurableJob(t *testing.T) {
db := newChannelWorkerDB(t)
wp := worker.NewWorkerPoolWithOptions(db, worker.WithNow(func() time.Time { return time.Date(2026, 6, 5, 11, 0, 0, 0, time.UTC) }))
dispatcher := NewDispatcher(wp)
listener := &workerDispatchListener{name: "capture"}
dispatcher.Register(listener)
event := NewChannelEvent(EventConversationCreated, ChannelWebWidget, 1, 2)
if err := dispatcher.DispatchAsync(context.Background(), event); err != nil {
t.Fatalf("dispatch async: %v", err)
}
if listener.count.Load() != 0 {
t.Fatalf("listener ran synchronously before worker processed job")
}
var count int64
if err := db.Model(&model.BackgroundJob{}).Where("job_type = ? AND status = ?", TaskTypeEventDispatch, model.BackgroundJobStatusQueued).Count(&count).Error; err != nil {
t.Fatalf("count jobs: %v", err)
}
if count != 1 {
t.Fatalf("expected one queued dispatch job, got %d", count)
}
processed, err := wp.ProcessOne(context.Background())
if err != nil || !processed {
t.Fatalf("process dispatch job: processed=%v err=%v", processed, err)
}
if listener.count.Load() != 1 {
t.Fatalf("listener was not called by durable dispatch job")
}
}