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) } } }