toolset_test.go

package vulnerability

import (
	"testing"

	"github.com/stackrox/stackrox-mcp/internal/client"
	"github.com/stackrox/stackrox-mcp/internal/config"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestNewToolset(t *testing.T) {
	cfg := &config.Config{
		Tools: config.ToolsConfig{
			Vulnerability: config.ToolsetVulnerabilityConfig{
				Enabled: true,
			},
		},
	}

	toolset := NewToolset(cfg, &client.Client{})

	require.NotNil(t, toolset)
	assert.Equal(t, "vulnerability", toolset.GetName())
}

func TestToolset_IsEnabled_True(t *testing.T) {
	cfg := &config.Config{
		Tools: config.ToolsConfig{
			Vulnerability: config.ToolsetVulnerabilityConfig{
				Enabled: true,
			},
		},
	}

	toolset := NewToolset(cfg, &client.Client{})
	assert.True(t, toolset.IsEnabled())

	tools := toolset.GetTools()
	require.NotEmpty(t, tools, "Should return tools when enabled")
	require.Len(t, tools, 3, "Should have all vulnerability tools")
	assert.Equal(t, "get_deployments_for_cve", tools[0].GetName())
	assert.Equal(t, "get_nodes_for_cve", tools[1].GetName())
	assert.Equal(t, "get_clusters_with_orchestrator_cve", tools[2].GetName())
}

func TestToolset_IsEnabled_False(t *testing.T) {
	cfg := &config.Config{
		Tools: config.ToolsConfig{
			Vulnerability: config.ToolsetVulnerabilityConfig{
				Enabled: false,
			},
		},
	}

	toolset := NewToolset(cfg, &client.Client{})
	assert.False(t, toolset.IsEnabled())

	tools := toolset.GetTools()
	assert.Empty(t, tools, "Should return empty list when toolset is disabled")
}