我正在try 为我的Postgres数据库编写集成测试.为此,我在Go中使用了测试容器.我已经在我的实际数据库中使用了TERN迁移,并将迁移放在Migrations文件夹中.问题是我的一些迁移使用.tern.conf文件中的属性,因为这些值可能会因环境而异.我为local创建了一个与local.tern.conf类似的配置文件,并try 设置全局变量.但由于某些原因,当我运行迁移时,没有 Select 这些值.下面是一个例子--
func SetupTestDatabase() *TestDatabase {
// Set the TERN_CONFIG environment variable to the path of your local.tern.conf file
if err := os.Setenv("TERN_CONFIG", "../local.tern.conf"); err != nil {
// Handle error if setting the environment variable fails
zap.S().Fatal("failed to set TERN_CONFIG", err)
}
// setup db container
ctx, cancel := context.WithTimeout(context.Background(), time.Second*60)
defer cancel()
container, dbInstance, dbAddr, err := createContainer(ctx)
if err != nil {
zap.S().Fatal("failed to setup test", err)
}
conn, _ := dbInstance.Acquire(ctx)
m, err := migrate.NewMigrator(ctx, conn.Conn(), "schema_version_table")
if err != nil {
zap.S().Fatal("failed to setup migrator", err)
}
if err := m.LoadMigrations("../migrations"); err != nil {
zap.S().Fatal("failed to load migrations", err)
return nil
}
if len(m.Migrations) == 0 {
zap.S().Fatal("no migrations found")
return nil
}
if err := m.Migrate(ctx); err != nil {
zap.S().Fatal("failed to migrate", err)
return nil
}
return &TestDatabase{
container: container,
DbInstance: dbInstance,
DbAddress: dbAddr,
}
}
这是我迁徙的一部分-
....
CREATE TABLE {{.version_table}} (
version integer NOT NULL
);
alter table {{.version_table}}
owner to {{.master_user}};
.....
这里是本地.tern.conf-
[database]
host = <host>
port = 5432
database = <db name>
version_table = schema_version_table
[data]
master_user = <user>