38 lines
1.7 KiB
Go
38 lines
1.7 KiB
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))
|
||
|
|
}
|