34 lines
1.2 KiB
Go
34 lines
1.2 KiB
Go
// 功能描述：验证配送点人员查询始终限制为当前配送点的配送角色。版本：v1.0.0。
|
||
package delivery
|
||
|
||
import (
|
||
"strings"
|
||
"testing"
|
||
|
||
"git.apinb.com/heqiapp/platforms/backend/api/internal/models"
|
||
"github.com/DATA-DOG/go-sqlmock"
|
||
"gorm.io/driver/postgres"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// TestDeliveryStaffQueryKeepsRoleScope 验证列表、详情和头像共用的范围包含 delivery 角色。
|
||
func TestDeliveryStaffQueryKeepsRoleScope(t *testing.T) {
|
||
connection, _, err := sqlmock.New()
|
||
if err != nil {
|
||
t.Fatalf("创建 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("打开 GORM 失败：%v", err)
|
||
}
|
||
point := models.DeliveryBasic{Entity: models.Entity{ID: 22}, GasBasicID: 11}
|
||
statement := deliveryStaffQuery(databaseService, point).
|
||
Where("identity = ?", "staff-identity").Find(&models.StaffAccount{}).Statement.SQL.String()
|
||
for _, required := range []string{"gas_basic_id", "delivery_basic_id", "role_code"} {
|
||
if !strings.Contains(statement, required) {
|
||
t.Fatalf("配送人员范围缺少 %s：%s", required, statement)
|
||
}
|
||
}
|
||
}
|