61 lines
2.0 KiB
Go
61 lines
2.0 KiB
Go
package gas
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// TestSupportedTicketValues 验证气站端工单枚举白名单。
|
|
func TestSupportedTicketValues(t *testing.T) {
|
|
for _, category := range []string{"delivery", "installation", "repair", "inspection", "reinspection", "customer_service"} {
|
|
if !isSupportedTicketCategory(category) {
|
|
t.Fatalf("标准工单分类未通过校验：%s", category)
|
|
}
|
|
}
|
|
for _, priority := range []string{"low", "normal", "high", "urgent"} {
|
|
if !isSupportedTicketPriority(priority) {
|
|
t.Fatalf("标准工单优先级未通过校验：%s", priority)
|
|
}
|
|
}
|
|
for _, value := range []string{"", "unknown", " normal "} {
|
|
if isSupportedTicketCategory(value) || isSupportedTicketPriority(value) {
|
|
t.Fatalf("非标准工单枚举不应通过校验：%q", value)
|
|
}
|
|
}
|
|
}
|
|
|
|
// TestTicketDetailQueryIncludesScopedRelationNames 验证详情名称查询保留气站范围并解析全部关系名称。
|
|
func TestTicketDetailQueryIncludesScopedRelationNames(t *testing.T) {
|
|
connection, _, err := sqlmock.New()
|
|
if err != nil {
|
|
t.Fatalf("create SQL mock: %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = connection.Close() })
|
|
databaseService, err := gorm.Open(postgres.New(postgres.Config{Conn: connection}), &gorm.Config{DryRun: true})
|
|
if err != nil {
|
|
t.Fatalf("open GORM database: %v", err)
|
|
}
|
|
|
|
statement := ticketDetailQueryWithDB(databaseService, 42).
|
|
Find(&ticketDetailDisplay{}).Statement.SQL.String()
|
|
for _, fragment := range []string{
|
|
"user_service_relation.gas_basic_id =",
|
|
"LEFT JOIN user_account AS ticket_user",
|
|
"LEFT JOIN delivery_basic AS ticket_delivery",
|
|
"LEFT JOIN staff_account AS ticket_staff",
|
|
"LEFT JOIN gas_basic AS ticket_gas",
|
|
"user_account_display_name",
|
|
"delivery_basic_display_name",
|
|
"staff_account_display_name",
|
|
"gas_basic_display_name",
|
|
} {
|
|
if !strings.Contains(statement, fragment) {
|
|
t.Fatalf("ticket detail display SQL must include %q, got: %s", fragment, statement)
|
|
}
|
|
}
|
|
}
|