71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package service
|
|
|
|
import (
|
|
"3588AdminBackend/internal/config"
|
|
"3588AdminBackend/internal/models"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestTaskService_CreateTask(t *testing.T) {
|
|
cfg := &config.Config{
|
|
Concurrency: 5,
|
|
}
|
|
// Mock registry
|
|
reg := NewRegistryService(cfg, nil)
|
|
reg.UpdateDevice(&models.Device{
|
|
DeviceID: "dev1",
|
|
IP: "127.0.0.1",
|
|
AgentPort: 9100,
|
|
Online: true,
|
|
})
|
|
|
|
agent := NewAgentClient(cfg)
|
|
svc := NewTaskService(cfg, agent, reg)
|
|
|
|
task, err := svc.CreateTask("config_apply", []string{"dev1"}, map[string]string{"foo": "bar"})
|
|
if err != nil {
|
|
t.Fatalf("failed to create task: %v", err)
|
|
}
|
|
|
|
if task.ID == "" {
|
|
t.Error("expected task ID to be set")
|
|
}
|
|
|
|
// Wait for task to finish or fail (since agent is nil, it will fail)
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
task.Mu.RLock()
|
|
defer task.Mu.RUnlock()
|
|
|
|
if task.Devices["dev1"].Status == models.TaskPending {
|
|
t.Error("expected task status to change from pending")
|
|
}
|
|
}
|
|
|
|
func TestTaskService_Subscribe(t *testing.T) {
|
|
cfg := &config.Config{
|
|
Concurrency: 5,
|
|
}
|
|
svc := NewTaskService(cfg, nil, NewRegistryService(cfg, nil))
|
|
|
|
taskID := "test-task"
|
|
svc.tasks[taskID] = models.NewTask(taskID, "test", []string{"dev1"}, nil)
|
|
|
|
ch, cleanup := svc.Subscribe(taskID)
|
|
defer cleanup()
|
|
|
|
go func() {
|
|
svc.updateDeviceStatus(taskID, "dev1", models.TaskRunning, 0.5, "")
|
|
}()
|
|
|
|
select {
|
|
case update := <-ch:
|
|
if update.DeviceID != "dev1" || update.Status != models.TaskRunning {
|
|
t.Errorf("unexpected update: %+v", update)
|
|
}
|
|
case <-time.After(1 * time.Second):
|
|
t.Error("timed out waiting for event")
|
|
}
|
|
}
|