38 lines
1.1 KiB
Go
38 lines
1.1 KiB
Go
// 功能描述：验证气站合同与配送订单展示查询的对象权限范围。
|
||
// 版本：v1.0.0
|
||
package gas
|
||
|
||
import (
|
||
"strings"
|
||
"testing"
|
||
|
||
"github.com/DATA-DOG/go-sqlmock"
|
||
"gorm.io/driver/postgres"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// TestScopedOrderDisplayQuery 验证订单用户名称直接按订单主键读取且查询受当前气站约束。
|
||
func TestScopedOrderDisplayQuery(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 := scopedOrderDisplayQuery(databaseService, 42).
|
||
Find(&[]gasOrderListDisplay{}).Statement.SQL.String()
|
||
for _, fragment := range []string{
|
||
"LEFT JOIN user_account AS order_user",
|
||
"user_account_display_name",
|
||
"gasorder_basic.gas_basic_id =",
|
||
} {
|
||
if !strings.Contains(statement, fragment) {
|
||
t.Fatalf("scoped order display SQL must include %q, got: %s", fragment, statement)
|
||
}
|
||
}
|
||
}
|