xinmi/internal/application/service/agent_fusion_test.go

package service

import (
	"github.com/Tencent/WeKnora/internal/types"
	"github.com/stretchr/testify/require"
	"testing"
)

func TestNewAgentDefaultsToPrivateWithoutPublishingToOtherUsers(t *testing.T) {
	s := &customAgentService{repo: &lifecycleAgentRepo{rows: map[string]*types.CustomAgent{}}}
	agent, err := s.CreateAgent(instructionContext(1, "alice", types.TenantRoleContributor), &types.CustomAgent{Name: "Private creation"})
	require.NoError(t, err)
	require.Equal(t, "private", agent.Visibility)
	require.True(t, agent.Enabled)
	_, err = s.GetAgentByID(instructionContext(1, "bob", types.TenantRoleContributor), agent.ID)
	require.Error(t, err)
}

func TestFusionValidationRequiresAuthorizedEnabledProductionAgents(t *testing.T) {
	ctx := instructionContext(1, "bob", types.TenantRoleContributor)
	primary := &types.CustomAgent{ID: "primary", TenantID: 1, Config: types.CustomAgentConfig{Role: types.AgentRoleProduction, FusionAgentIDs: []string{"aux"}}}
	aux := &types.CustomAgent{ID: "aux", TenantID: 1, CreatedBy: "alice", Name: "Secret", Visibility: "private", Enabled: true, Config: types.CustomAgentConfig{Role: types.AgentRoleProduction}}
	s := &customAgentService{repo: &lifecycleAgentRepo{rows: map[string]*types.CustomAgent{"aux": aux}}}
	err := s.validateFusionAgents(ctx, primary)
	require.Error(t, err)
	require.NotContains(t, err.Error(), "Secret")
	aux.Visibility = "global"
	require.NoError(t, s.validateFusionAgents(ctx, primary))
	aux.Enabled = false
	require.Error(t, s.validateFusionAgents(ctx, primary))
	aux.Enabled = true
	aux.Config.Role = types.AgentRoleQC
	require.Error(t, s.validateFusionAgents(ctx, primary))
	aux.Config.Role = types.AgentRoleProduction
	aux.TenantID = 2
	require.Error(t, s.validateFusionAgents(ctx, primary))
}