35 lines
1.0 KiB
Go
35 lines
1.0 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestStartCloudAgentPiDoesNotRestartAnActiveRunner(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
s := &Service{piRunners: map[string]context.CancelFunc{"run-a": cancel}}
|
|
s.startCloudAgentPi("user-a", "run-a")
|
|
|
|
if err := ctx.Err(); err != nil {
|
|
t.Fatalf("duplicate start cancelled the active Pi runner: %v", err)
|
|
}
|
|
}
|
|
|
|
// 审批恢复时旧会话可能还在收尾：不能取消它，也不能丢弃恢复请求。
|
|
func TestResumeCloudAgentPiQueuesRestartWhileRunnerIsExiting(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
s := &Service{piRunners: map[string]context.CancelFunc{"run-a": cancel}}
|
|
s.resumeCloudAgentPi("user-a", "run-a")
|
|
|
|
if err := ctx.Err(); err != nil {
|
|
t.Fatalf("resume cancelled the exiting Pi runner: %v", err)
|
|
}
|
|
if _, queued := s.piRunnerRestarts["run-a"]; !queued {
|
|
t.Fatal("resume request was dropped while the previous runner was still exiting")
|
|
}
|
|
}
|