WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

2026-09-22 16:19:06 +08:00
commit f7fc992f57
3929 changed files with 1006607 additions and 0 deletions
@@ -0,0 +1,37 @@
+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))
+}