t
3d47227fbd
Remove embedded frontend and workflow editor assets, add standalone API deployment configuration, and retain the current backend service updates.
94 lines
3.0 KiB
Go
94 lines
3.0 KiB
Go
package services
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"agent-desk/internal/models"
|
|
"agent-desk/internal/pkg/dto"
|
|
"agent-desk/internal/pkg/dto/request"
|
|
"agent-desk/internal/pkg/enums"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/mlogclub/simple/sqls"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
func TestUpdateAIAgentKeepsPublishedRevisionActive(t *testing.T) {
|
|
dbName := strings.NewReplacer("/", "_", " ", "_").Replace(t.Name())
|
|
db, err := gorm.Open(sqlite.Open("file:"+dbName+"?mode=memory&cache=shared"), &gorm.Config{
|
|
NamingStrategy: schema.NamingStrategy{
|
|
TablePrefix: "t_",
|
|
SingularTable: true,
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("open sqlite: %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
sqlDB, dbErr := db.DB()
|
|
if dbErr == nil {
|
|
_ = sqlDB.Close()
|
|
}
|
|
})
|
|
if err := db.AutoMigrate(&models.AIConfig{}, &models.AIAgent{}, &models.AIAgentWorkflowBinding{}); err != nil {
|
|
t.Fatalf("auto migrate: %v", err)
|
|
}
|
|
sqls.SetDB(db)
|
|
|
|
config := &models.AIConfig{
|
|
Name: "test", Status: enums.StatusOk, Provider: enums.AIProviderOpenAI,
|
|
ModelType: enums.AIModelTypeLLM, ModelName: "test-model",
|
|
}
|
|
if err := db.Create(config).Error; err != nil {
|
|
t.Fatalf("create ai config: %v", err)
|
|
}
|
|
createdModel, err := AIAgentService.buildAIAgentModel(0, request.CreateAIAgentRequest{
|
|
Name: "fixed ai reception agent",
|
|
AIConfigID: config.ID,
|
|
ServiceMode: enums.IMConversationServiceModeHumanOnly,
|
|
HandoffMode: enums.AIAgentHandoffModeWaitPool,
|
|
FallbackMode: enums.AIAgentFallbackModeNoAnswer,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("build ai agent model: %v", err)
|
|
}
|
|
if createdModel.ServiceMode != enums.IMConversationServiceModeAIFirst {
|
|
t.Fatalf("service mode = %d, want AI first", createdModel.ServiceMode)
|
|
}
|
|
agent := &models.AIAgent{
|
|
Name: "published agent", Status: enums.StatusOk, AIConfigID: config.ID,
|
|
ServiceMode: enums.IMConversationServiceModeAIFirst, HandoffMode: enums.AIAgentHandoffModeWaitPool,
|
|
FallbackMode: enums.AIAgentFallbackModeNoAnswer, RolloutPercent: 100, PublishedRevisionID: 18,
|
|
}
|
|
if err := db.Create(agent).Error; err != nil {
|
|
t.Fatalf("create ai agent: %v", err)
|
|
}
|
|
|
|
err = AIAgentService.UpdateAIAgent(request.UpdateAIAgentRequest{
|
|
ID: agent.ID,
|
|
CreateAIAgentRequest: request.CreateAIAgentRequest{
|
|
Name: "updated draft", AIConfigID: config.ID,
|
|
ServiceMode: enums.IMConversationServiceModeAIFirst,
|
|
HandoffMode: enums.AIAgentHandoffModeWaitPool,
|
|
FallbackMode: enums.AIAgentFallbackModeNoAnswer,
|
|
RolloutPercent: 100,
|
|
},
|
|
}, &dto.AuthPrincipal{UserID: 1, Username: "admin"})
|
|
if err != nil {
|
|
t.Fatalf("UpdateAIAgent: %v", err)
|
|
}
|
|
|
|
var updated models.AIAgent
|
|
if err := db.First(&updated, agent.ID).Error; err != nil {
|
|
t.Fatalf("get updated ai agent: %v", err)
|
|
}
|
|
if updated.PublishedRevisionID != agent.PublishedRevisionID {
|
|
t.Fatalf("published revision id = %d, want %d", updated.PublishedRevisionID, agent.PublishedRevisionID)
|
|
}
|
|
if updated.Name != "updated draft" {
|
|
t.Fatalf("draft name = %q, want updated draft", updated.Name)
|
|
}
|
|
}
|