跳到主要内容

测试

如果在单元测试中使用 ent.Client,可以使用生成的 enttest 包来创建客户端并自动运行模式迁移,具体方法如下:

package main

import (
"testing"

"<project>/ent/enttest"

_ "github.com/mattn/go-sqlite3"
)

func TestXXX(t *testing.T) {
client := enttest.Open(t, "sqlite3", "file:ent?mode=memory&_fk=1")
defer client.Close()
// ...
}

若需要向 Open 传递功能选项,请使用 enttest.Option

func TestXXX(t *testing.T) {
opts := []enttest.Option{
enttest.WithOptions(ent.Log(t.Log)),
enttest.WithMigrateOptions(migrate.WithGlobalUniqueID(true)),
}
client := enttest.Open(t, "sqlite3", "file:ent?mode=memory&_fk=1", opts...)
defer client.Close()
// ...
}