Rogeeandrogee
2b182f9956
* H-300: wire Captain Skills into Web runtime * H-300: enforce effective model and conservative skill budget * H-300: fix CI gosec step * ci: extend golangci-lint timeout * fix lint findings across backend * fix(push): resolve delivery protocol blockers * test(repository): close SQLite test databases * test(repository): reuse SQLite schema per package * H-307: restore backend Go cache in CI * H-307: prefetch modules before cold lint * H-307: resolve govulncheck security gate * H-307: build lint with patched Go toolchain * H-307: clear remaining security scan findings --------- Co-authored-by: Rogee <rogee@ipao.vip>
253 lines
9.2 KiB
Go
253 lines
9.2 KiB
Go
package automation
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestBotRuleService_Create(t *testing.T) {
|
|
dbProvider := setupAutomationTestDBProvider(t)
|
|
accountID, _ := seedTestAccount(dbProvider.DB(), t)
|
|
svc := NewBotRuleService(dbProvider)
|
|
|
|
rule := &BotRule{
|
|
AccountID: accountID,
|
|
AgentBotID: 1,
|
|
EventName: BotRuleEventConversationCreated,
|
|
Name: "Auto-respond on create",
|
|
Conditions: Conditions{
|
|
{Attribute: "status", FilterOperator: "equal", Values: []string{"open"}, QueryOperator: "and"},
|
|
},
|
|
Actions: Actions{
|
|
{ActionName: "send_message", ActionParams: map[string]interface{}{"content": "Hello!"}},
|
|
},
|
|
Status: BotRuleStatusActive,
|
|
}
|
|
|
|
err := svc.Create(context.Background(), rule)
|
|
if err != nil {
|
|
t.Fatalf("expected no error creating bot rule, got: %v", err)
|
|
}
|
|
if rule.ID == 0 {
|
|
t.Fatal("expected bot rule ID to be set after creation")
|
|
}
|
|
}
|
|
|
|
func TestBotRuleService_GetByID(t *testing.T) {
|
|
dbProvider := setupAutomationTestDBProvider(t)
|
|
accountID, _ := seedTestAccount(dbProvider.DB(), t)
|
|
svc := NewBotRuleService(dbProvider)
|
|
|
|
rule := &BotRule{
|
|
AccountID: accountID,
|
|
AgentBotID: 1,
|
|
EventName: BotRuleEventMessageCreated,
|
|
Name: "Auto-label spam",
|
|
Conditions: Conditions{
|
|
{Attribute: "content", FilterOperator: "contains", Values: []string{"spam"}, QueryOperator: "and"},
|
|
},
|
|
Actions: Actions{
|
|
{ActionName: "add_label", ActionParams: map[string]interface{}{"labels": []string{"spam"}}},
|
|
},
|
|
Status: BotRuleStatusActive,
|
|
}
|
|
|
|
err := svc.Create(context.Background(), rule)
|
|
if err != nil {
|
|
t.Fatalf("create failed: %v", err)
|
|
}
|
|
|
|
found, err := svc.GetByID(context.Background(), rule.ID)
|
|
if err != nil {
|
|
t.Fatalf("GetByID failed: %v", err)
|
|
}
|
|
if found.EventName != BotRuleEventMessageCreated {
|
|
t.Fatalf("expected event_name message_created, got %s", found.EventName)
|
|
}
|
|
}
|
|
|
|
func TestBotRuleService_GetByID_NotFound(t *testing.T) {
|
|
dbProvider := setupAutomationTestDBProvider(t)
|
|
svc := NewBotRuleService(dbProvider)
|
|
|
|
_, err := svc.GetByID(context.Background(), 9999)
|
|
if err == nil {
|
|
t.Fatal("expected error for non-existent bot rule, got nil")
|
|
}
|
|
}
|
|
|
|
func TestBotRuleService_ListByAccount(t *testing.T) {
|
|
dbProvider := setupAutomationTestDBProvider(t)
|
|
accountID, _ := seedTestAccount(dbProvider.DB(), t)
|
|
svc := NewBotRuleService(dbProvider)
|
|
|
|
r1 := &BotRule{AccountID: accountID, AgentBotID: 1, EventName: BotRuleEventConversationCreated, Name: "rule1", Conditions: Conditions{}, Actions: Actions{}, Status: BotRuleStatusActive}
|
|
r2 := &BotRule{AccountID: accountID, AgentBotID: 2, EventName: BotRuleEventMessageCreated, Name: "rule2", Conditions: Conditions{}, Actions: Actions{}, Status: BotRuleStatusActive}
|
|
require.NoError(t, svc.Create(context.Background(), r1))
|
|
require.NoError(t, svc.Create(context.Background(), r2))
|
|
|
|
rules, err := svc.ListByAccount(context.Background(), accountID)
|
|
if err != nil {
|
|
t.Fatalf("ListByAccount failed: %v", err)
|
|
}
|
|
if len(rules) != 2 {
|
|
t.Fatalf("expected 2 rules, got %d", len(rules))
|
|
}
|
|
}
|
|
|
|
func TestBotRuleService_ListByAgentBot(t *testing.T) {
|
|
dbProvider := setupAutomationTestDBProvider(t)
|
|
accountID, _ := seedTestAccount(dbProvider.DB(), t)
|
|
svc := NewBotRuleService(dbProvider)
|
|
|
|
r1 := &BotRule{AccountID: accountID, AgentBotID: 1, EventName: BotRuleEventConversationCreated, Name: "rule1", Conditions: Conditions{}, Actions: Actions{}, Status: BotRuleStatusActive}
|
|
r2 := &BotRule{AccountID: accountID, AgentBotID: 1, EventName: BotRuleEventMessageCreated, Name: "rule2", Conditions: Conditions{}, Actions: Actions{}, Status: BotRuleStatusActive}
|
|
r3 := &BotRule{AccountID: accountID, AgentBotID: 2, EventName: BotRuleEventConversationUpdated, Name: "rule3", Conditions: Conditions{}, Actions: Actions{}, Status: BotRuleStatusActive}
|
|
require.NoError(t, svc.Create(context.Background(), r1))
|
|
require.NoError(t, svc.Create(context.Background(), r2))
|
|
require.NoError(t, svc.Create(context.Background(), r3))
|
|
|
|
rules, err := svc.ListByAgentBot(context.Background(), accountID, 1)
|
|
if err != nil {
|
|
t.Fatalf("ListByAgentBot failed: %v", err)
|
|
}
|
|
if len(rules) != 2 {
|
|
t.Fatalf("expected 2 rules for agent_bot_id=1, got %d", len(rules))
|
|
}
|
|
}
|
|
|
|
func TestBotRuleService_ListActiveByAccountAndEvent(t *testing.T) {
|
|
dbProvider := setupAutomationTestDBProvider(t)
|
|
accountID, _ := seedTestAccount(dbProvider.DB(), t)
|
|
svc := NewBotRuleService(dbProvider)
|
|
|
|
r1 := &BotRule{AccountID: accountID, AgentBotID: 1, EventName: BotRuleEventConversationCreated, Name: "rule1", Conditions: Conditions{}, Actions: Actions{}, Status: BotRuleStatusActive}
|
|
r2 := &BotRule{AccountID: accountID, AgentBotID: 2, EventName: BotRuleEventConversationCreated, Name: "rule2", Conditions: Conditions{}, Actions: Actions{}, Status: BotRuleStatusActive}
|
|
r3 := &BotRule{AccountID: accountID, AgentBotID: 1, EventName: BotRuleEventMessageCreated, Name: "rule3", Conditions: Conditions{}, Actions: Actions{}, Status: BotRuleStatusInactive}
|
|
require.NoError(t, svc.Create(context.Background(), r1))
|
|
require.NoError(t, svc.Create(context.Background(), r2))
|
|
require.NoError(t, svc.Create(context.Background(), r3))
|
|
|
|
rules, err := svc.ListActiveByAccountAndEvent(context.Background(), accountID, "conversation_created")
|
|
if err != nil {
|
|
t.Fatalf("ListActiveByAccountAndEvent failed: %v", err)
|
|
}
|
|
if len(rules) != 2 {
|
|
t.Fatalf("expected 2 active rules for conversation_created, got %d", len(rules))
|
|
}
|
|
}
|
|
|
|
func TestBotRuleService_Update(t *testing.T) {
|
|
dbProvider := setupAutomationTestDBProvider(t)
|
|
accountID, _ := seedTestAccount(dbProvider.DB(), t)
|
|
svc := NewBotRuleService(dbProvider)
|
|
|
|
rule := &BotRule{
|
|
AccountID: accountID,
|
|
AgentBotID: 1,
|
|
EventName: BotRuleEventConversationCreated,
|
|
Name: "rule1",
|
|
Conditions: Conditions{{Attribute: "status", FilterOperator: "equal", Values: []string{"open"}, QueryOperator: "and"}},
|
|
Actions: Actions{{ActionName: "send_message", ActionParams: map[string]interface{}{"content": "Hi"}}},
|
|
Status: BotRuleStatusActive,
|
|
}
|
|
require.NoError(t, svc.Create(context.Background(), rule))
|
|
|
|
// Update the rule
|
|
rule.EventName = BotRuleEventMessageCreated
|
|
rule.Status = BotRuleStatusInactive
|
|
err := svc.Update(context.Background(), rule)
|
|
if err != nil {
|
|
t.Fatalf("Update failed: %v", err)
|
|
}
|
|
|
|
found, _ := svc.GetByID(context.Background(), rule.ID)
|
|
if found.EventName != BotRuleEventMessageCreated {
|
|
t.Fatalf("expected event_name message_created, got %s", found.EventName)
|
|
}
|
|
if found.Status != BotRuleStatusInactive {
|
|
t.Fatal("expected status=inactive after update")
|
|
}
|
|
}
|
|
|
|
func TestBotRuleService_Delete(t *testing.T) {
|
|
dbProvider := setupAutomationTestDBProvider(t)
|
|
accountID, _ := seedTestAccount(dbProvider.DB(), t)
|
|
svc := NewBotRuleService(dbProvider)
|
|
|
|
rule := &BotRule{AccountID: accountID, AgentBotID: 1, EventName: BotRuleEventConversationCreated, Name: "rule1", Conditions: Conditions{}, Actions: Actions{}, Status: BotRuleStatusActive}
|
|
require.NoError(t, svc.Create(context.Background(), rule))
|
|
|
|
err := svc.Delete(context.Background(), rule.ID)
|
|
if err != nil {
|
|
t.Fatalf("Delete failed: %v", err)
|
|
}
|
|
|
|
_, err = svc.GetByID(context.Background(), rule.ID)
|
|
if err == nil {
|
|
t.Fatal("expected error after deletion, got nil")
|
|
}
|
|
}
|
|
|
|
func TestBotRuleService_Clone(t *testing.T) {
|
|
dbProvider := setupAutomationTestDBProvider(t)
|
|
accountID, _ := seedTestAccount(dbProvider.DB(), t)
|
|
svc := NewBotRuleService(dbProvider)
|
|
|
|
rule := &BotRule{
|
|
AccountID: accountID,
|
|
AgentBotID: 1,
|
|
EventName: BotRuleEventConversationCreated,
|
|
Name: "Auto-respond",
|
|
Conditions: Conditions{{Attribute: "status", FilterOperator: "equal", Values: []string{"open"}, QueryOperator: "and"}},
|
|
Actions: Actions{{ActionName: "send_message", ActionParams: map[string]interface{}{"content": "Hi"}}},
|
|
Status: BotRuleStatusActive,
|
|
}
|
|
require.NoError(t, svc.Create(context.Background(), rule))
|
|
|
|
cloned, err := svc.Clone(context.Background(), rule.ID)
|
|
if err != nil {
|
|
t.Fatalf("Clone failed: %v", err)
|
|
}
|
|
if cloned.Name != "Auto-respond (copy)" {
|
|
t.Fatalf("expected cloned name 'Auto-respond (copy)', got %s", cloned.Name)
|
|
}
|
|
if cloned.ID == rule.ID {
|
|
t.Fatal("cloned rule should have a different ID")
|
|
}
|
|
}
|
|
|
|
func TestBotRuleService_ToggleStatus(t *testing.T) {
|
|
dbProvider := setupAutomationTestDBProvider(t)
|
|
accountID, _ := seedTestAccount(dbProvider.DB(), t)
|
|
svc := NewBotRuleService(dbProvider)
|
|
|
|
rule := &BotRule{AccountID: accountID, AgentBotID: 1, EventName: BotRuleEventConversationCreated, Name: "rule1", Conditions: Conditions{}, Actions: Actions{}, Status: BotRuleStatusActive}
|
|
require.NoError(t, svc.Create(context.Background(), rule))
|
|
|
|
err := svc.ToggleStatus(context.Background(), rule.ID, BotRuleStatusInactive)
|
|
if err != nil {
|
|
t.Fatalf("ToggleStatus failed: %v", err)
|
|
}
|
|
|
|
found, err := svc.GetByID(context.Background(), rule.ID)
|
|
if err != nil {
|
|
t.Fatalf("GetByID after toggle failed: %v", err)
|
|
}
|
|
if found.Status != BotRuleStatusInactive {
|
|
t.Fatal("expected status=inactive after toggle")
|
|
}
|
|
|
|
// Toggle back to active
|
|
require.NoError(t, svc.ToggleStatus(context.Background(), rule.ID, BotRuleStatusActive))
|
|
found2, err := svc.GetByID(context.Background(), rule.ID)
|
|
if err != nil {
|
|
t.Fatalf("GetByID after second toggle failed: %v", err)
|
|
}
|
|
if found2.Status != BotRuleStatusActive {
|
|
t.Fatal("expected status=active after second toggle")
|
|
}
|
|
}
|