如何在Ballerina类单元测试中Mock私有final DAO对象

针对你提供的ConsentManagementImpl类，要Mock私有final的consentManagementDAO对象并自定义方法行为，有两种可行方案：

方案一：调整类设计，支持依赖注入（推荐） #

直接修改类的构造逻辑，新增一个重载的init方法允许传入DAO实例，这样测试时可以直接注入Mock对象，无需修改原有生产逻辑：

修改后的业务类代码 #

public isolated client class ConsentManagementImpl {
    private final dao:ConsentManagementDAO consentManagementDAO;

    // 生产环境用的初始化方法
    public isolated function init(string host, string username, string password, string database, int port) returns error? {
        self.consentManagementDAO = check new(host, username, password, database, port);
    }

    // 测试专用的初始化方法，直接传入DAO实例
    isolated function init(dao:ConsentManagementDAO dao) {
        self.consentManagementDAO = dao;
    }

    public isolated function handle consentAmendment(string consentId, string payload, string orgId) returns error? {
        check self.consentManagementDAO->updateConsent(consentId, orgId, payload);
    }
}

单元测试代码 #

import ballerina/test;
import dao;

@test:Config
function testConsentAmendment() returns error? {
    // 创建Mock的DAO实例
    dao:ConsentManagementDAO mockDAO = test:Mock(dao:ConsentManagementDAO);

    // 自定义updateConsent方法的Mock行为：调用时返回成功（nil）
    test:when(mockDAO).doRemoteCall("updateConsent", (string, string, string)).thenReturn(());

    // 用测试专用的初始化方法注入Mock DAO
    ConsentManagementImpl service = new(mockDAO);

    // 调用待测试方法
    error? result = service->consentAmendment("test-consent-id", "test-payload", "test-org-id");

    // 断言方法执行成功
    test:assertEquals(result, ());

    // 验证updateConsent方法被正确调用了一次
    test:verify(mockDAO).doRemoteCall("updateConsent", ("test-consent-id", "test-org-id", "test-payload")).times(1);
}

方案二：不修改原类，通过test:setField修改私有final字段 #

如果无法修改业务类代码，可以使用Ballerina测试模块的test:setField API直接替换私有final字段的值：

单元测试代码 #

import ballerina/test;
import dao;

@test:Config
function testConsentAmendmentWithSetField() returns error? {
    // 先初始化业务类（参数用占位符即可，后续会替换DAO）
    ConsentManagementImpl service = check new("dummy-host", "dummy-user", "dummy-pass", "dummy-db", 0);

    // 创建Mock的DAO实例
    dao:ConsentManagementDAO mockDAO = test:Mock(dao:ConsentManagementDAO);

    // 自定义Mock方法行为
    test:when(mockDAO).doRemoteCall("updateConsent", (string, string, string)).thenReturn(());

    // 替换私有final的consentManagementDAO字段
    check test:setField(service, "consentManagementDAO", mockDAO);

    // 执行测试
    error? result = service->consentAmendment("test-consent-id", "test-payload", "test-org-id");

    test:assertEquals(result, ());
    test:verify(mockDAO).doRemoteCall("updateConsent", ("test-consent-id", "test-org-id", "test-payload")).times(1);
}

注意事项 #

• 方案一遵循SOLID原则，既不影响生产逻辑，又提升了代码的可测试性和扩展性，优先推荐。
• 方案二依赖反射类操作，若业务类的字段名或结构发生变化，测试会失效，仅适合无法修改原代码的场景。
• 确保使用的Ballerina版本支持test:Mock和test:setField特性（Ballerina 2201.x及以上版本均支持）。

内容的提问来源于stack exchange，提问作者LUHEERATHAN THEVAKUMAR