fixed session manager
This commit is contained in:
65
internal/session/manager_test.go
Normal file
65
internal/session/manager_test.go
Normal file
@ -0,0 +1,65 @@
|
||||
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")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user