Apollo iOS 中如何为带参数的 GraphQL 查询设置模拟数据

在 iOS 开发中使用 Apollo GraphQL 客户端时，单元测试中设置模拟数据是一个常见需求。本文将深入探讨如何为带有多种参数和选择器的复杂查询正确设置模拟数据。

问题背景

当开发者尝试为包含以下特性的 GraphQL 查询设置模拟数据时可能会遇到困难：

• 多个布尔标志参数
• 国家/地区等枚举参数
• 条件性字段（使用 @include 指令）
• 多个数据片段（如 dataA 和 dataB）

常见误区

开发者通常会尝试以下方法但可能无法获得预期结果：

1. 使用生成的 Mock 类型直接创建模拟数据
2. 从 JSON 响应生成数据对象
3. 尝试通过 fulfilledFragments 属性访问数据

其中最大的误区是认为 fulfilledFragments 属性包含实际数据。实际上，这个属性是 Apollo 内部用于字段收集和解析的机制，存储的是 ObjectIdentifier 数组，而非字段数据。

正确解决方案

使用选择集初始化器

Apollo iOS 提供了更可靠的选择集初始化器（Selection Set Initializers）来创建测试数据：

let data = SomeDataQuery.Data(
    dataA: .init(
        nodes: [
            .init(
                country: .australia,
                id: "Data 1"
            )
        ]
    ),
    dataB: nil
)

这种方法比测试模拟（TestMocks）更直观可靠，也是 Apollo 团队目前推荐的方式。

处理查询参数

对于带有变量的查询，需要确保模拟数据与查询参数匹配：

let variables = SomeDataQuery.Variables(
    flagA: true,
    flagB: false,
    countries: [.australia]
)

let data = SomeDataQuery.Data(
    dataA: .init(/* 数据A内容 */),
    dataB: nil // 因为 flagB 为 false
)

条件性字段处理

对于使用 @include 指令的字段，在创建模拟数据时应该：

1. 如果条件为真，提供相应字段的数据
2. 如果条件为假，将该字段设为 nil

最佳实践

1. 优先使用选择集初始化器而非测试模拟
2. 确保模拟数据结构与查询结构完全匹配
3. 对于条件性字段，根据测试场景设置适当的值或 nil
4. 考虑创建工厂方法来生成常用测试数据
5. 为复杂查询编写专门的测试数据构建器

通过遵循这些方法，开发者可以更可靠地为复杂的 GraphQL 查询创建测试数据，确保单元测试的准确性和可维护性。