kefu_cloud/server/internal/service/session_list_test.go

package service

import (
	"fmt"
	"testing"
	"time"

	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"kefu-cloud/server/internal/model"
)

func setupSessionListTestDB(t *testing.T) *gorm.DB {
	t.Helper()
	db, err := gorm.Open(
		sqlite.Open(fmt.Sprintf("file:kefu_session_list_%d?mode=memory&cache=shared", time.Now().UnixNano())),
		&gorm.Config{},
	)
	if err != nil {
		t.Fatalf("打开测试数据库失败: %v", err)
	}
	model.DB = db
	if err := model.Migrate(db); err != nil {
		t.Fatalf("迁移测试数据库失败: %v", err)
	}
	return db
}

func TestListSessionsWorkbenchPaginatesSortsAndExcludesEnded(t *testing.T) {
	db := setupSessionListTestDB(t)
	tenant := model.Tenant{Name: "分页测试租户", Status: "normal"}
	if err := db.Create(&tenant).Error; err != nil {
		t.Fatalf("创建租户失败: %v", err)
	}
	customer := model.Customer{TenantID: tenant.ID, Name: "星河科技"}
	if err := db.Create(&customer).Error; err != nil {
		t.Fatalf("创建客户失败: %v", err)
	}

	now := time.Now().Truncate(time.Second)
	sessions := []model.Session{
		{
			TenantID: tenant.ID, CustomerID: customer.ID, Status: "active", Priority: "urgent",
			VisitorIP: "10.0.0.1", CreatedAt: now.Add(-4 * time.Hour),
		},
		{
			TenantID: tenant.ID, CustomerID: customer.ID, Status: "waiting", Priority: "normal",
			VisitorIP: "10.0.0.2", CreatedAt: now.Add(-3 * time.Hour),
		},
		{
			TenantID: tenant.ID, CustomerID: customer.ID, Status: "active", Priority: "normal",
			VisitorIP: "10.0.0.3", CreatedAt: now.Add(-2 * time.Hour),
		},
		{
			TenantID: tenant.ID, CustomerID: customer.ID, Status: "ended", Priority: "urgent",
			VisitorIP: "10.0.0.4", CreatedAt: now.Add(-time.Hour),
		},
	}
	if err := db.Create(&sessions).Error; err != nil {
		t.Fatalf("创建会话失败: %v", err)
	}
	messages := []model.Message{
		{SessionID: sessions[0].ID, SenderType: "visitor", Content: "紧急报价单", Type: "text", Seq: 1, SentAt: now.Add(-10 * time.Minute)},
		{SessionID: sessions[1].ID, SenderType: "visitor", Content: "等待接入", Type: "text", Seq: 1, SentAt: now.Add(-5 * time.Minute)},
		{SessionID: sessions[2].ID, SenderType: "visitor", Content: "普通咨询", Type: "text", Seq: 1, SentAt: now.Add(-time.Minute)},
		{SessionID: sessions[3].ID, SenderType: "visitor", Content: "已结束会话", Type: "text", Seq: 1, SentAt: now},
	}
	if err := db.Create(&messages).Error; err != nil {
		t.Fatalf("创建消息失败: %v", err)
	}

	allData := Actor{SessionAll: true, HistoryAll: true}
	firstPage, total, err := ListSessions(allData, tenant.ID, SessionFilter{
		Scope: "workbench", Page: 1, PageSize: 2,
	})
	if err != nil {
		t.Fatalf("查询第一页失败: %v", err)
	}
	if total != 3 {
		t.Fatalf("工作台总数=%d，期望 3（不包含已结束会话）", total)
	}
	if len(firstPage) != 2 {
		t.Fatalf("第一页数量=%d，期望 2", len(firstPage))
	}
	if firstPage[0].ID != sessions[0].ID || firstPage[1].ID != sessions[1].ID {
		t.Fatalf("排序异常: got [%d, %d], want [%d, %d]",
			firstPage[0].ID, firstPage[1].ID, sessions[0].ID, sessions[1].ID)
	}
	if firstPage[0].CustomerName != "星河科技" || firstPage[0].LastMessage != "紧急报价单" {
		t.Fatalf("会话摘要补全异常: %+v", firstPage[0])
	}
	if firstPage[0].UnreadCount != 1 || firstPage[0].MessageCount != 1 {
		t.Fatalf("未读/消息数异常: unread=%d messages=%d", firstPage[0].UnreadCount, firstPage[0].MessageCount)
	}

	secondPage, secondTotal, err := ListSessions(allData, tenant.ID, SessionFilter{
		Scope: "workbench", Page: 2, PageSize: 2,
	})
	if err != nil {
		t.Fatalf("查询第二页失败: %v", err)
	}
	if secondTotal != 3 || len(secondPage) != 1 || secondPage[0].ID != sessions[2].ID {
		t.Fatalf("第二页异常: total=%d items=%+v", secondTotal, secondPage)
	}

	historyItems, historyTotal, err := ListSessions(allData, tenant.ID, SessionFilter{Page: 1, PageSize: 20})
	if err != nil {
		t.Fatalf("查询普通会话列表失败: %v", err)
	}
	if historyTotal != 4 || len(historyItems) != 4 {
		t.Fatalf("非工作台查询不应排除已结束会话: total=%d items=%+v", historyTotal, historyItems)
	}

	agentItems, agentTotal, err := ListSessions(Actor{UserID: 999}, tenant.ID, SessionFilter{
		Scope: "workbench", Page: 1, PageSize: 20,
	})
	if err != nil {
		t.Fatalf("查询普通坐席工作台失败: %v", err)
	}
	if agentTotal != 1 || len(agentItems) != 1 || agentItems[0].ID != sessions[1].ID {
		t.Fatalf("普通坐席只能看到等待中或自己的进行中会话: total=%d items=%+v", agentTotal, agentItems)
	}
}

func TestListSessionsWorkbenchFiltersSearchAndCustomer(t *testing.T) {
	db := setupSessionListTestDB(t)
	tenant := model.Tenant{Name: "筛选测试租户", Status: "normal"}
	if err := db.Create(&tenant).Error; err != nil {
		t.Fatalf("创建租户失败: %v", err)
	}
	customers := []model.Customer{
		{TenantID: tenant.ID, Name: "远山贸易"},
		{TenantID: tenant.ID, Name: "青禾设计"},
	}
	if err := db.Create(&customers).Error; err != nil {
		t.Fatalf("创建客户失败: %v", err)
	}
	sessions := []model.Session{
		{TenantID: tenant.ID, CustomerID: customers[0].ID, Status: "waiting", Priority: "urgent"},
		{TenantID: tenant.ID, CustomerID: customers[1].ID, Status: "active", Priority: "normal"},
	}
	if err := db.Create(&sessions).Error; err != nil {
		t.Fatalf("创建会话失败: %v", err)
	}
	messages := []model.Message{
		{SessionID: sessions[0].ID, SenderType: "visitor", Content: "需要产品报价", Type: "text", Seq: 1, SentAt: time.Now()},
		{SessionID: sessions[0].ID, SenderType: "agent", Content: "稍等", Type: "text", Seq: 2, SentAt: time.Now()},
		{SessionID: sessions[1].ID, SenderType: "visitor", Content: "售后服务", Type: "text", Seq: 1, SentAt: time.Now()},
	}
	if err := db.Create(&messages).Error; err != nil {
		t.Fatalf("创建消息失败: %v", err)
	}

	allData := Actor{SessionAll: true, HistoryAll: true}
	cases := []struct {
		name   string
		filter SessionFilter
		want   uint
	}{
		{name: "按客户名搜索", filter: SessionFilter{Scope: "workbench", Search: "远山"}, want: sessions[0].ID},
		{name: "按最后消息搜索", filter: SessionFilter{Scope: "workbench", Search: "稍等"}, want: sessions[0].ID},
		{name: "按状态筛选", filter: SessionFilter{Scope: "workbench", Status: "active"}, want: sessions[1].ID},
		{name: "按优先级筛选", filter: SessionFilter{Scope: "workbench", Priority: "urgent"}, want: sessions[0].ID},
		{name: "按客户筛选", filter: SessionFilter{Scope: "workbench", CustomerID: fmt.Sprint(customers[1].ID)}, want: sessions[1].ID},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			tc.filter.Page = 1
			tc.filter.PageSize = 20
			items, total, err := ListSessions(allData, tenant.ID, tc.filter)
			if err != nil {
				t.Fatalf("查询失败: %v", err)
			}
			if total != 1 || len(items) != 1 || items[0].ID != tc.want {
				t.Fatalf("筛选结果异常: total=%d items=%+v want=%d", total, items, tc.want)
			}
		})
	}
}