66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package session
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"supervisor/internal/domain"
|
|
"supervisor/internal/store/memory"
|
|
)
|
|
|
|
func TestDeleteSessionDeletesStoredExitedSessionWithoutRuntime(t *testing.T) {
|
|
ctx := context.Background()
|
|
store := memory.NewStore()
|
|
exitCode := -1
|
|
session := domain.Session{
|
|
ID: "sess_stale",
|
|
Name: "stale",
|
|
Command: "bash",
|
|
Status: domain.SessionStatusExited,
|
|
CreatedAt: time.Now().UTC(),
|
|
ExitCode: &exitCode,
|
|
}
|
|
if err := store.Upsert(ctx, session); err != nil {
|
|
t.Fatalf("upsert stale session: %v", err)
|
|
}
|
|
|
|
manager := NewManager(store, nil)
|
|
if err := manager.DeleteSession(ctx, session.ID); err != nil {
|
|
t.Fatalf("delete stale exited session: %v", err)
|
|
}
|
|
|
|
if _, ok, err := store.Get(ctx, session.ID); err != nil {
|
|
t.Fatalf("get deleted session: %v", err)
|
|
} else if ok {
|
|
t.Fatal("expected stale session to be deleted from store")
|
|
}
|
|
}
|
|
|
|
func TestDeleteSessionRejectsStoredRunningSessionWithoutRuntime(t *testing.T) {
|
|
ctx := context.Background()
|
|
store := memory.NewStore()
|
|
session := domain.Session{
|
|
ID: "sess_running",
|
|
Name: "running",
|
|
Command: "bash",
|
|
Status: domain.SessionStatusRunning,
|
|
CreatedAt: time.Now().UTC(),
|
|
}
|
|
if err := store.Upsert(ctx, session); err != nil {
|
|
t.Fatalf("upsert running session: %v", err)
|
|
}
|
|
|
|
manager := NewManager(store, nil)
|
|
if err := manager.DeleteSession(ctx, session.ID); !errors.Is(err, ErrSessionRunning) {
|
|
t.Fatalf("DeleteSession error = %v, want %v", err, ErrSessionRunning)
|
|
}
|
|
|
|
if _, ok, err := store.Get(ctx, session.ID); err != nil {
|
|
t.Fatalf("get running session: %v", err)
|
|
} else if !ok {
|
|
t.Fatal("expected running session to remain in store")
|
|
}
|
|
}
|