17370845950

使用 Mockito 模拟对象时 ObjectMapper 报错的解决方案

当尝试使用 Jackson ObjectMapper 序列化 Mockito 模拟对象时,可能会遇到 InvalidDefinitionException 异常。这是因为模拟对象并非普通的数据结构,ObjectMapper 无法直接序列化其内部的 Mockito 特有成员。

问题分析

Mockito 创建的模拟对象与普通的数据结构有着本质的区别。ObjectMapper 在序列化对象时,会检查对象的成员变量,并尝试序列化每个成员。对于基本类型(如 int、Long)或已配置的类型,ObjectMapper 可以直接处理。对于自定义类型,ObjectMapper 会寻找相应的序列化器或依赖于注解来指导序列化过程。

然而,Mockito 模拟对象内部包含 Mockito 特有的成员,例如 ByteBuddyCrossClassLoaderSerializationSupport。ObjectMapper 默认情况下无法识别和处理这些成员,因此会抛出 InvalidDefinitionException 异常。

以下代码演示了导致问题的场景:

import com.fasterxml.jackson.databind.ObjectMapper;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;

import static org.mockito.Mockito.mock;

public class MockitoObjectMapperTest {

    @Test
    public void testSerializeMockObject() throws Exception {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            objectMapper.writeValueAsString(mock(Object.class));
        } catch (com.fasterxml.jackson.databind.exc.InvalidDefinitionException e) {
            System.err.println("Caught expected exception: " + e.getMessage());
        }
    }
}

上述代码片段中,我们使用 Mockito.mock(Object.class) 创建了一个 Object 类型的模拟对象,并尝试使用 ObjectMapper.writeValueAsString() 将其序列化为 JSON 字符串。这会导致 InvalidDefinitionException 异常,异常信息提示 ObjectMapper 无法找到 org.mockito.internal.creation.bytebuddy.ByteBuddyCrossClassLoaderSerializationSupport 类的序列化器。

解决方案与建议

  1. 不要尝试序列化模拟对象: 最直接的解决方案是避免尝试序列化模拟对象。Mockito 模拟对象的主要目的是在单元测试中模拟依赖项的行为,而不是作为数据传输对象(DTO)进行序列化。

  2. 只序列化所需的数据: 如果您需要序列化包含模拟对象的数据结构,请仅序列化您需要的部分,例如从模拟对象中提取相关属性并构建一个新的对象进行序列化。

    例如,假设您有一个包含模拟对象 MyService 的类 MyController:

    class MyController {
        private MyService myService;
    
        public MyController(MyService myService) {
            this.myService = myService;
        }
    
        public String getData() {
            // 从 myService 获取数据并处理
            String data = myService.getData();
            return data;
        }
    }
    
    interface MyService {
        String getData();
    }

    在测试中,您可以模拟 MyService,并在 MyController 中使用它。如果要测试 MyController 的 getData() 方法的返回值,您可以直接序列化返回值,而不是 MyController 或模拟的 MyService 对象:

    import com.fasterxml.jackson.databind.ObjectMapper;
    import org.junit.jupiter.api.Test;
    import org.mockito.Mockito;
    
    import static org.junit.jupiter.api.Assertions.assertEquals;
    import static org.mockito.Mockito.when;
    
    public class MyControllerTest {
    
        @Test
        public void testGetData() throws Exception {
            // 模拟 MyService
            MyService myServiceMock = Mockito.mock(MyService.class);
            when(myServiceMock.getData()).thenReturn("test data");
    
            // 创建 MyController 实例,并注入模拟的 MyService
            MyController myController = new MyController(myServiceMock);
    
            // 调用 getData() 方法
            String data = myController.getData();
    
            // 序列化返回值
            ObjectMapper objectMapper = new ObjectMapper();
            String jsonData = objectMapper.writeValueAsString(data);
    
            // 断言
            assertEquals("\"test data\"", jsonData);
        }
    }

    在这个例子中,我们只序列化了 getData() 方法的返回值 data,避免了直接序列化模拟对象 myServiceMock,从而避免了 InvalidDefinitionException 异常。

  3. 自定义序列化器(不推荐): 虽然可以为 Mockito 的内部类型创建自定义序列化器,但这通常不是一个好的解决方案。这会引入额外的复杂性,并且需要深入了解 Mockito 的内部实现,而这些实现可能会在未来的版本中发生变化。此外,自定义序列化器可能无法完全处理模拟对象的所有状态,导致序列化结果不完整或不正确。

总结

当使用 Mockito 模拟对象时,请记住它们并非设计用于序列化。避免直接使用 ObjectMapper 序列化模拟对象,而是专注于序列化所需的数据,例如从模拟对象中提取相关属性并构建新的对象进行序列化。 这样可以避免 InvalidDefinitionException 异常,并保持代码的简洁和可维护性。