mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-04 15:06:37 +08:00
ryan
416603b616
- Replace db.DB(ctx) with database.DB(ctx) from plugins/infra/database
- Replace db.Redis/db.PrefixedKey/db.GetJSON/db.SetJSON with cachepkg.* from plugins/infra/cache
- Replace pkg/persistence/idgen with pkg/idgen (already exists)
- Replace pkg/persistence/batchwriter with pkg/batchwriter (already exists)
- Replace pkg/persistence/migrator with pkg/migrator (already exists)
- Replace pkg/persistence/logstore with plugins/domain/risk_control/logstore
- Delete defunct pkg/{persistence,cap,message_gateway,push,shared,task}
- Fix vet issues: db alias in domain_test.go, driver_asynq_worker.TaskHandler reference
- Update Makefile architecture guard
- Update docs and skill references
- Update go.mod: gorilla/sessions promotion to direct dependency
43 lines
769 B
Go
43 lines
769 B
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package driver_asynq_cron
|
|
|
|
import "testing"
|
|
|
|
func TestWaitForStop(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
closeDone bool
|
|
closeSignal bool
|
|
wantSignal bool
|
|
}{
|
|
{
|
|
name: "explicit stop",
|
|
closeDone: true,
|
|
},
|
|
{
|
|
name: "process signal",
|
|
closeSignal: true,
|
|
wantSignal: true,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
done := make(chan struct{})
|
|
signals := make(chan struct{})
|
|
if tt.closeDone {
|
|
close(done)
|
|
}
|
|
if tt.closeSignal {
|
|
close(signals)
|
|
}
|
|
|
|
if got := waitForStop(done, signals); got != tt.wantSignal {
|
|
t.Errorf("waitForStop() = %t, want %t", got, tt.wantSignal)
|
|
}
|
|
})
|
|
}
|
|
}
|