vrr/Pulse
2
0
Fork 0
mirror of https://github.com/rcourtman/Pulse.git synced 2026-05-01 21:10:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

refactor(ai): consolidate tool implementations into domain-specific files

Browse source 
- Merge tools_infrastructure.go, tools_intelligence.go, tools_patrol.go,
  tools_profiles.go into their respective domain tools
- Expand tools_control.go with command execution logic
- Expand tools_discovery.go with resource discovery handlers
- Expand tools_storage.go with storage-related operations
- Expand tools_metrics.go with metrics functionality
- Update tests to match new structure

This consolidation reduces file count and groups related functionality together.
This commit is contained in:
rcourtman 2026-01-28 21:21:28 +00:00
parent 23ff4d1337
commit a75393d1c5
15 changed files with 4236 additions and 106 deletions
Download patch file Download diff file Expand all files Collapse all files

68
internal/ai/tools/infrastructure_low_coverage_test.go
View file

@ -150,44 +150,6 @@ func TestExecuteListPBSJobs(t *testing.T) {
assert.Equal(t, "job1", resp.Jobs[0].ID)
}
func TestExecuteGetClusterStatus(t *testing.T) {
ctx := context.Background()
exec := NewPulseToolExecutor(ExecutorConfig{StateProvider: &mockStateProvider{state: models.StateSnapshot{}}})
result, err := exec.executeGetClusterStatus(ctx, map[string]interface{}{})
require.NoError(t, err)
assert.Equal(t, "No Proxmox nodes found.", result.Content[0].Text)
state := models.StateSnapshot{
Nodes: []models.Node{
{
Name: "node1",
Status: "online",
IsClusterMember: true,
ClusterName: "cluster1",
Instance: "pve1",
},
{
Name: "node2",
Status: "offline",
IsClusterMember: true,
ClusterName: "cluster1",
Instance: "pve1",
},
},
}
exec = NewPulseToolExecutor(ExecutorConfig{StateProvider: &mockStateProvider{state: state}})
result, err = exec.executeGetClusterStatus(ctx, map[string]interface{}{})
require.NoError(t, err)
var resp ClusterStatusResponse
require.NoError(t, json.Unmarshal([]byte(result.Content[0].Text), &resp))
require.Len(t, resp.Clusters, 1)
assert.Equal(t, "cluster1", resp.Clusters[0].ClusterName)
assert.False(t, resp.Clusters[0].QuorumOK)
}
func TestExecuteGetConnectionHealth(t *testing.T) {
ctx := context.Background()
exec := NewPulseToolExecutor(ExecutorConfig{StateProvider: &mockStateProvider{state: models.StateSnapshot{}}})
@ -349,36 +311,6 @@ func TestExecuteGetResourceDisks(t *testing.T) {
assert.Equal(t, "vm1", resp.Resources[0].ID)
}
func TestExecuteListRecentTasks(t *testing.T) {
ctx := context.Background()
now := time.Now()
state := models.StateSnapshot{
PVEBackups: models.PVEBackups{
BackupTasks: []models.BackupTask{
{
ID: "task1",
Node: "node1",
Instance: "pve1",
Type: "backup",
Status: "ok",
StartTime: now,
},
},
},
}
exec := NewPulseToolExecutor(ExecutorConfig{StateProvider: &mockStateProvider{state: state}})
result, err := exec.executeListRecentTasks(ctx, map[string]interface{}{
"type": "backup",
})
require.NoError(t, err)
var resp RecentTasksResponse
require.NoError(t, json.Unmarshal([]byte(result.Content[0].Text), &resp))
require.Len(t, resp.Tasks, 1)
assert.Equal(t, "task1", resp.Tasks[0].ID)
}
func TestExecuteListBackupTasks(t *testing.T) {
ctx := context.Background()
exec := NewPulseToolExecutor(ExecutorConfig{})