mlogclub
101577f163
- Updated import paths in multiple service files to reflect the new agent-desk module. - Added a new configuration file for agent-desk in the Docker setup.
105 lines
2.6 KiB
Go
105 lines
2.6 KiB
Go
package services
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"agent-desk/internal/models"
|
|
"agent-desk/internal/pkg/config"
|
|
"agent-desk/internal/pkg/enums"
|
|
"agent-desk/internal/repositories"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/mlogclub/simple/sqls"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
func TestWxWorkNotifyBuildTextContent(t *testing.T) {
|
|
svc := newWxWorkNotifyService()
|
|
got := svc.buildTextContent("工单提醒", "这是一条测试消息")
|
|
if got != "工单提醒\n\n这是一条测试消息" {
|
|
t.Fatalf("unexpected content: %q", got)
|
|
}
|
|
}
|
|
|
|
func TestWxWorkNotifyDefaultRecipients(t *testing.T) {
|
|
db := setupWxWorkNotifyTestDB(t)
|
|
config.SetCurrent(&config.Config{
|
|
WxWork: config.WxWorkConfig{
|
|
CorpID: "corp-1",
|
|
Notify: config.WxWorkNotifyConfig{
|
|
Enabled: true,
|
|
ToUsers: []int64{11, 11, 12},
|
|
},
|
|
},
|
|
})
|
|
now := time.Now()
|
|
for _, identity := range []*models.UserIdentity{
|
|
{
|
|
UserID: 11,
|
|
Provider: enums.ThirdProviderWxWork,
|
|
ProviderUserID: "wx_user_a",
|
|
ProviderCorpID: "corp-1",
|
|
Status: enums.StatusOk,
|
|
LastAuthAt: &now,
|
|
},
|
|
{
|
|
UserID: 12,
|
|
Provider: enums.ThirdProviderWxWork,
|
|
ProviderUserID: "wx_user_b",
|
|
ProviderCorpID: "corp-1",
|
|
Status: enums.StatusOk,
|
|
LastAuthAt: &now,
|
|
},
|
|
} {
|
|
if err := repositories.UserIdentityRepository.Create(db, identity); err != nil {
|
|
t.Fatalf("create user identity error = %v", err)
|
|
}
|
|
}
|
|
|
|
svc := newWxWorkNotifyService()
|
|
toUsers := svc.defaultToUsers()
|
|
if len(toUsers) != 2 || toUsers[0] != "wx_user_a" || toUsers[1] != "wx_user_b" {
|
|
t.Fatalf("unexpected users: %#v", toUsers)
|
|
}
|
|
}
|
|
|
|
func TestWxWorkNotifyNormalizeDuplicateCheckInterval(t *testing.T) {
|
|
svc := newWxWorkNotifyService()
|
|
if got := svc.normalizeDuplicateCheckInterval(0); got != 1800 {
|
|
t.Fatalf("expected default interval 1800, got %d", got)
|
|
}
|
|
if got := svc.normalizeDuplicateCheckInterval(20000); got != 14400 {
|
|
t.Fatalf("expected capped interval 14400, got %d", got)
|
|
}
|
|
if got := svc.normalizeDuplicateCheckInterval(600); got != 600 {
|
|
t.Fatalf("expected interval 600, got %d", got)
|
|
}
|
|
}
|
|
|
|
func setupWxWorkNotifyTestDB(t *testing.T) *gorm.DB {
|
|
t.Helper()
|
|
|
|
db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{
|
|
NamingStrategy: schema.NamingStrategy{
|
|
TablePrefix: "t_",
|
|
SingularTable: true,
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("open sqlite error = %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
sqlDB, err := db.DB()
|
|
if err == nil {
|
|
_ = sqlDB.Close()
|
|
}
|
|
})
|
|
if err := db.AutoMigrate(&models.UserIdentity{}); err != nil {
|
|
t.Fatalf("auto migrate error = %v", err)
|
|
}
|
|
sqls.SetDB(db)
|
|
return db
|
|
}
|