78 lines
3.3 KiB
Go
78 lines
3.3 KiB
Go
package autoassignment
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
)
|
|
|
|
func setupCapacityAssignmentDB(t *testing.T) *gorm.DB {
|
|
t.Helper()
|
|
db, err := gorm.Open(sqlite.Open("file::memory:"), &gorm.Config{Logger: logger.Default.LogMode(logger.Silent)})
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.AutoMigrate(
|
|
&model.Account{},
|
|
&model.User{},
|
|
&model.AccountUser{},
|
|
&model.Inbox{},
|
|
&model.InboxMember{},
|
|
&model.Contact{},
|
|
&model.Conversation{},
|
|
&model.AgentCapacityPolicy{},
|
|
&model.InboxCapacityLimit{},
|
|
))
|
|
return db
|
|
}
|
|
|
|
func TestAssignmentService_SkipsAgentsAtInboxCapacity(t *testing.T) {
|
|
_, rdb := setupTestRedis(t)
|
|
db := setupCapacityAssignmentDB(t)
|
|
svc := NewAssignmentService(db, rdb)
|
|
ctx := context.Background()
|
|
|
|
account := &model.Account{Name: "Capacity Account"}
|
|
require.NoError(t, db.Create(account).Error)
|
|
inbox := &model.Inbox{AccountID: account.ID, Name: "Capacity Inbox", ChannelType: string(model.InboxChannelTypeWebWidget), EnableAutoAssignment: true}
|
|
require.NoError(t, db.Create(inbox).Error)
|
|
contact := &model.Contact{AccountID: account.ID, Name: "Capacity Contact"}
|
|
require.NoError(t, db.Create(contact).Error)
|
|
|
|
agentAtCapacity := createCapacityUser(t, db, account.ID, "full@test.com")
|
|
agentWithCapacity := createCapacityUser(t, db, account.ID, "available@test.com")
|
|
require.NoError(t, db.Create(&model.InboxMember{InboxID: inbox.ID, UserID: agentAtCapacity.ID}).Error)
|
|
require.NoError(t, db.Create(&model.InboxMember{InboxID: inbox.ID, UserID: agentWithCapacity.ID}).Error)
|
|
|
|
policy := &model.AgentCapacityPolicy{AccountID: account.ID, Name: "Capacity", AssignmentLogic: "round_robin", ExclusionRules: []byte(`{}`)}
|
|
require.NoError(t, db.Create(policy).Error)
|
|
require.NoError(t, db.Model(&model.AccountUser{}).
|
|
Where("account_id = ? AND user_id = ?", account.ID, agentAtCapacity.ID).
|
|
Update("agent_capacity_policy_id", policy.ID).Error)
|
|
require.NoError(t, db.Create(&model.InboxCapacityLimit{AgentCapacityPolicyID: policy.ID, InboxID: inbox.ID, ConversationLimit: 1}).Error)
|
|
|
|
now := time.Now().Unix()
|
|
existing := &model.Conversation{AccountID: account.ID, InboxID: inbox.ID, ContactID: contact.ID, AssigneeID: &agentAtCapacity.ID, Status: string(model.ConversationStatusOpen), ChannelType: "web_widget", Channel: "web_widget", LastActivityAt: &now}
|
|
unassigned := &model.Conversation{AccountID: account.ID, InboxID: inbox.ID, ContactID: contact.ID, Status: string(model.ConversationStatusOpen), ChannelType: "web_widget", Channel: "web_widget", LastActivityAt: &now}
|
|
require.NoError(t, db.Create(existing).Error)
|
|
require.NoError(t, db.Create(unassigned).Error)
|
|
|
|
agentID, err := svc.AssignConversation(ctx, unassigned.ID, inbox.ID, account.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, agentWithCapacity.ID, agentID)
|
|
}
|
|
|
|
func createCapacityUser(t *testing.T, db *gorm.DB, accountID uint, email string) *model.User {
|
|
t.Helper()
|
|
user := &model.User{AccountID: accountID, Name: email, Email: email, Password: "hashed", Active: true, Available: true}
|
|
require.NoError(t, db.Create(user).Error)
|
|
require.NoError(t, db.Create(&model.AccountUser{AccountID: accountID, UserID: user.ID, Role: "agent", Availability: "online"}).Error)
|
|
return user
|
|
}
|