跳到主要内容

测试

如果你想在单元测试中测试 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()
// ...
}