SunYF
a2afd2f6dc
- MES(B): 工单/BOM/备料/工序字典/PLC下发/拧紧/扫码报工/半成品/AGV/追溯逻辑，WMS与海康RCS客户端，看板Redis缓存API(概览/设备/进度/报警/趋势)+SSE - WMS(C): JWT滑动续签、Excel导入、盘点、内部API、种子数据、独立Postgres配置 - WMS客户端(E): Go网关8891反向代理+内嵌Vue3十页 - 工位终端(D): SQLite本地缓存+模拟拧紧源+内嵌Vue3页面 - Dashboard(A): 看板数据改接MES内部缓存API，SSE实时刷新+vite代理 - 清理各项目球形磨遗留代码，新增部署手册.md
86 lines
1.8 KiB
Go
86 lines
1.8 KiB
Go
// Code generated by ent, DO NOT EDIT.
|
|
|
|
package enttest
|
|
|
|
import (
|
|
"context"
|
|
|
|
"bj_power_wms/ent"
|
|
// required by schema hooks.
|
|
_ "bj_power_wms/ent/runtime"
|
|
|
|
"bj_power_wms/ent/migrate"
|
|
|
|
"entgo.io/ent/dialect/sql/schema"
|
|
)
|
|
|
|
type (
|
|
// TestingT is the interface that is shared between
|
|
// testing.T and testing.B and used by enttest.
|
|
TestingT interface {
|
|
FailNow()
|
|
Error(...any)
|
|
}
|
|
|
|
// Option configures client creation.
|
|
Option func(*options)
|
|
|
|
options struct {
|
|
opts []ent.Option
|
|
migrateOpts []schema.MigrateOption
|
|
}
|
|
)
|
|
|
|
// WithOptions forwards options to client creation.
|
|
func WithOptions(opts ...ent.Option) Option {
|
|
return func(o *options) {
|
|
o.opts = append(o.opts, opts...)
|
|
}
|
|
}
|
|
|
|
// WithMigrateOptions forwards options to auto migration.
|
|
func WithMigrateOptions(opts ...schema.MigrateOption) Option {
|
|
return func(o *options) {
|
|
o.migrateOpts = append(o.migrateOpts, opts...)
|
|
}
|
|
}
|
|
|
|
func newOptions(opts []Option) *options {
|
|
o := &options{}
|
|
for _, opt := range opts {
|
|
opt(o)
|
|
}
|
|
return o
|
|
}
|
|
|
|
// Open calls ent.Open and auto-run migration.
|
|
func Open(t TestingT, driverName, dataSourceName string, opts ...Option) *ent.Client {
|
|
o := newOptions(opts)
|
|
c, err := ent.Open(driverName, dataSourceName, o.opts...)
|
|
if err != nil {
|
|
t.Error(err)
|
|
t.FailNow()
|
|
}
|
|
migrateSchema(t, c, o)
|
|
return c
|
|
}
|
|
|
|
// NewClient calls ent.NewClient and auto-run migration.
|
|
func NewClient(t TestingT, opts ...Option) *ent.Client {
|
|
o := newOptions(opts)
|
|
c := ent.NewClient(o.opts...)
|
|
migrateSchema(t, c, o)
|
|
return c
|
|
}
|
|
func migrateSchema(t TestingT, c *ent.Client, o *options) {
|
|
tables, err := schema.CopyTables(migrate.Tables)
|
|
if err != nil {
|
|
t.Error(err)
|
|
t.FailNow()
|
|
}
|
|
if err := migrate.Create(context.Background(), c.Schema, tables, o.migrateOpts...); err != nil {
|
|
t.Error(err)
|
|
t.FailNow()
|
|
}
|
|
}
|