Mathis/llamactl
1
0
Fork 0
mirror of https://github.com/lordmathis/llamactl.git synced 2025-11-06 00:54:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add TimeoutCheckInterval to instance configuration in tests

Browse Source
This commit is contained in:
Matúš Námešný
2025-08-17 21:42:52 +02:00
parent c45fa13206
commit 41eaebc927
1 changed files with 49 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

88
pkg/manager/manager_test.go
View File

@@ -15,13 +15,14 @@ import (
func TestNewInstanceManager(t *testing.T) { func TestNewInstanceManager(t *testing.T) {
cfg := config.InstancesConfig{ cfg := config.InstancesConfig{
PortRange: [2]int{8000, 9000}, PortRange: [2]int{8000, 9000},
LogsDir: "/tmp/test", LogsDir: "/tmp/test",
MaxInstances: 5, MaxInstances: 5,
LlamaExecutable: "llama-server", LlamaExecutable: "llama-server",
DefaultAutoRestart: true, DefaultAutoRestart: true,
DefaultMaxRestarts: 3, DefaultMaxRestarts: 3,
DefaultRestartDelay: 5, DefaultRestartDelay: 5,
TimeoutCheckInterval: 5,
} }
manager := manager.NewInstanceManager(cfg) manager := manager.NewInstanceManager(cfg)
@@ -99,8 +100,9 @@ func TestCreateInstance_DuplicateName(t *testing.T) {
func TestCreateInstance_MaxInstancesLimit(t *testing.T) { func TestCreateInstance_MaxInstancesLimit(t *testing.T) {
// Create manager with low max instances limit // Create manager with low max instances limit
cfg := config.InstancesConfig{ cfg := config.InstancesConfig{
PortRange: [2]int{8000, 9000}, PortRange: [2]int{8000, 9000},
MaxInstances: 2, // Very low limit for testing MaxInstances: 2, // Very low limit for testing
TimeoutCheckInterval: 5,
} }
manager := manager.NewInstanceManager(cfg) manager := manager.NewInstanceManager(cfg)
@@ -235,8 +237,9 @@ func TestCreateInstance_MultiplePortAssignment(t *testing.T) {
func TestCreateInstance_PortExhaustion(t *testing.T) { func TestCreateInstance_PortExhaustion(t *testing.T) {
// Create manager with very small port range // Create manager with very small port range
cfg := config.InstancesConfig{ cfg := config.InstancesConfig{
PortRange: [2]int{8000, 8001}, // Only 2 ports available PortRange: [2]int{8000, 8001}, // Only 2 ports available
MaxInstances: 10, // Higher than available ports MaxInstances: 10, // Higher than available ports
TimeoutCheckInterval: 5,
} }
manager := manager.NewInstanceManager(cfg) manager := manager.NewInstanceManager(cfg)
@@ -497,9 +500,10 @@ func TestPersistence_InstancePersistedOnCreation(t *testing.T) {
tempDir := t.TempDir() tempDir := t.TempDir()
cfg := config.InstancesConfig{ cfg := config.InstancesConfig{
PortRange: [2]int{8000, 9000}, PortRange: [2]int{8000, 9000},
InstancesDir: tempDir, InstancesDir: tempDir,
MaxInstances: 10, MaxInstances: 10,
TimeoutCheckInterval: 5,
} }
manager := manager.NewInstanceManager(cfg) manager := manager.NewInstanceManager(cfg)
@@ -542,9 +546,10 @@ func TestPersistence_InstancePersistedOnUpdate(t *testing.T) {
tempDir := t.TempDir() tempDir := t.TempDir()
cfg := config.InstancesConfig{ cfg := config.InstancesConfig{
PortRange: [2]int{8000, 9000}, PortRange: [2]int{8000, 9000},
InstancesDir: tempDir, InstancesDir: tempDir,
MaxInstances: 10, MaxInstances: 10,
TimeoutCheckInterval: 5,
} }
manager := manager.NewInstanceManager(cfg) manager := manager.NewInstanceManager(cfg)
@@ -599,9 +604,10 @@ func TestPersistence_InstanceFileDeletedOnDeletion(t *testing.T) {
tempDir := t.TempDir() tempDir := t.TempDir()
cfg := config.InstancesConfig{ cfg := config.InstancesConfig{
PortRange: [2]int{8000, 9000}, PortRange: [2]int{8000, 9000},
InstancesDir: tempDir, InstancesDir: tempDir,
MaxInstances: 10, MaxInstances: 10,
TimeoutCheckInterval: 5,
} }
manager := manager.NewInstanceManager(cfg) manager := manager.NewInstanceManager(cfg)
@@ -670,9 +676,10 @@ func TestPersistence_InstancesLoadedFromDisk(t *testing.T) {
// Create manager - should load instances from disk // Create manager - should load instances from disk
cfg := config.InstancesConfig{ cfg := config.InstancesConfig{
PortRange: [2]int{8000, 9000}, PortRange: [2]int{8000, 9000},
InstancesDir: tempDir, InstancesDir: tempDir,
MaxInstances: 10, MaxInstances: 10,
TimeoutCheckInterval: 5,
} }
manager := manager.NewInstanceManager(cfg) manager := manager.NewInstanceManager(cfg)
@@ -737,9 +744,10 @@ func TestPersistence_PortMapPopulatedFromLoadedInstances(t *testing.T) {
// Create manager - should load instance and register port // Create manager - should load instance and register port
cfg := config.InstancesConfig{ cfg := config.InstancesConfig{
PortRange: [2]int{8000, 9000}, PortRange: [2]int{8000, 9000},
InstancesDir: tempDir, InstancesDir: tempDir,
MaxInstances: 10, MaxInstances: 10,
TimeoutCheckInterval: 5,
} }
manager := manager.NewInstanceManager(cfg) manager := manager.NewInstanceManager(cfg)
@@ -764,12 +772,13 @@ func TestPersistence_CompleteInstanceDataRoundTrip(t *testing.T) {
tempDir := t.TempDir() tempDir := t.TempDir()
cfg := config.InstancesConfig{ cfg := config.InstancesConfig{
PortRange: [2]int{8000, 9000}, PortRange: [2]int{8000, 9000},
InstancesDir: tempDir, InstancesDir: tempDir,
MaxInstances: 10, MaxInstances: 10,
DefaultAutoRestart: true, DefaultAutoRestart: true,
DefaultMaxRestarts: 3, DefaultMaxRestarts: 3,
DefaultRestartDelay: 5, DefaultRestartDelay: 5,
TimeoutCheckInterval: 5,
} }
// Create first manager and instance with comprehensive options // Create first manager and instance with comprehensive options
@@ -880,13 +889,14 @@ func TestPersistence_CompleteInstanceDataRoundTrip(t *testing.T) {
// Helper function to create a test manager with standard config // Helper function to create a test manager with standard config
func createTestManager() manager.InstanceManager { func createTestManager() manager.InstanceManager {
cfg := config.InstancesConfig{ cfg := config.InstancesConfig{
PortRange: [2]int{8000, 9000}, PortRange: [2]int{8000, 9000},
LogsDir: "/tmp/test", LogsDir: "/tmp/test",
MaxInstances: 10, MaxInstances: 10,
LlamaExecutable: "llama-server", LlamaExecutable: "llama-server",
DefaultAutoRestart: true, DefaultAutoRestart: true,
DefaultMaxRestarts: 3, DefaultMaxRestarts: 3,
DefaultRestartDelay: 5, DefaultRestartDelay: 5,
TimeoutCheckInterval: 5,
} }
return manager.NewInstanceManager(cfg) return manager.NewInstanceManager(cfg)
} }