Atlas项目中的PostgreSQL公共模式测试问题解析

问题背景

在Atlas项目v0.30.1-canary版本中，当开发者在PostgreSQL数据库中使用public模式并安装扩展时，测试套件会出现一个特定问题：第三个测试用例会失败并报告"schema 'public' does not exist"错误。这个问题在稳定版v0.30.0中并不存在。

问题重现

要重现这个问题，需要以下配置：

1. 定义一个包含public模式的HCL文件，并在该模式下安装扩展：

schema "public" {
}

extension "tablefunc" {
  schema  = schema.public
  version = "1.0"
}

2. 创建包含三个简单测试用例的测试文件：

test "schema" "a" {
  exec {
    sql = "SELECT 1"
  }
}

test "schema" "b" {
  exec {
    sql = "SELECT 1"
  }
}

test "schema" "c" {
  exec {
    sql = "SELECT 1"
  }
}

3. 执行测试命令后，前两个测试通过，但第三个测试会失败，报错信息表明无法找到public模式。

技术分析

这个问题涉及到Atlas测试框架与PostgreSQL交互的几个关键方面：

1. 测试环境初始化：Atlas在运行每个测试前会初始化一个干净的数据库环境。对于PostgreSQL，这包括创建必要的模式。

2. 模式管理：PostgreSQL默认包含一个public模式，但Atlas可能需要显式创建它以确保测试环境的一致性。

3. 扩展依赖：当扩展被安装在特定模式下时，Atlas需要确保该模式在测试环境中可用。

在v0.30.1-canary版本中，测试框架可能在处理多个测试用例时未能正确维护模式状态，导致后续测试中public模式不可用。

解决方案

Atlas团队已经通过内部提交修复了这个问题。修复的核心在于确保：

1. 测试环境初始化时正确处理默认模式
2. 在多测试用例场景下保持模式状态一致性
3. 正确处理扩展与模式的依赖关系

最佳实践建议

对于使用Atlas进行数据库测试的开发人员，建议：

1. 在定义模式时显式声明public模式，即使它是PostgreSQL的默认模式
2. 对于包含扩展的测试，确保扩展依赖的模式在测试文件中明确定义
3. 在多测试用例场景中，考虑测试之间的依赖关系和状态隔离

这个问题提醒我们，在使用数据库迁移和测试工具时，即使是默认行为也可能需要显式声明，以确保测试环境的一致性和可重复性。