如何在部分Spring测试中注入特定的Clock模拟实例？

这个场景我太熟了！Spring测试里局部替换Bean其实有好几种灵活的方式，我给你列几个最常用的，你可以根据自己的需求选：

方法1：用@TestConfiguration局部覆盖Bean #

在需要特殊Clock的测试类里，定义一个内部的@TestConfiguration类，注册你需要的自定义Clock，并且加上@Primary确保它会优先被注入（覆盖全局的那个模拟Clock）。示例代码如下：

public class MyClassTest {
    @Autowired
    private MyClass myClass;

    // 这个内部配置只会作用于当前测试类，不会影响其他测试
    @TestConfiguration
    static class CustomClockConfig {
        @Bean
        @Primary
        public Clock customClock() {
            // 返回你需要的特定模拟Clock，比如固定某个时间点
            return Clock.fixed(Instant.parse("2024-01-01T00:00:00Z"), ZoneId.systemDefault());
        }
    }

    @Test
    public void test1(){
        // 这里的myClass已经注入了上面定义的customClock
        // 编写你的测试逻辑...
    }
}

这种方式的优势是隔离性强，只影响当前测试类，而且可以完全自定义Clock的行为。

方法2：用@MockBean动态自定义（配合Mockito） #

如果你的项目用了Mockito，直接在测试类里用@MockBean替换原来的Clock，然后在测试方法或者初始化阶段设置它的行为，适合需要动态调整Clock输出的场景：

import org.mockito.Mockito;
import org.springframework.boot.test.mock.mockito.MockBean;
import java.time.Instant;
import java.time.ZoneId;

public class MyClassTest {
    @Autowired
    private MyClass myClass;

    @MockBean
    private Clock clock;

    @org.junit.jupiter.api.BeforeEach // 或者Junit4的@Before
    void setUp() {
        // 自定义Clock的返回值，比如固定某个时间
        Mockito.when(clock.instant()).thenReturn(Instant.parse("2024-01-01T00:00:00Z"));
        Mockito.when(clock.getZone()).thenReturn(ZoneId.systemDefault());
    }

    @Test
    public void test1(){
        // 这里的myClass使用的是你mock的Clock
        // 测试逻辑...
    }

    @Test
    public void test2(){
        // 甚至可以在单个测试方法里重新设置Clock行为
        Mockito.when(clock.instant()).thenReturn(Instant.parse("2024-12-31T23:59:59Z"));
        // 测试逻辑...
    }
}

方法3：用@ContextConfiguration指定专属配置类 #

如果你的自定义Clock配置比较复杂，或者想让多个测试类复用这个配置，可以单独写一个配置类，然后在测试类里指定使用它：

首先写一个独立的配置类：

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import java.time.Clock;
import java.time.Instant;
import java.time.ZoneId;

@Configuration
public class TestCustomClockConfig {
    @Bean
    public Clock customClock() {
        return Clock.fixed(Instant.parse("2024-01-01T00:00:00Z"), ZoneId.systemDefault());
    }
}

然后在测试类里指定这个配置：

import org.springframework.test.context.ContextConfiguration;

@ContextConfiguration(classes = TestCustomClockConfig.class)
public class MyClassTest {
    @Autowired
    private MyClass myClass;

    @Test
    public void test1(){
        // 测试逻辑...
    }
}

这种方式适合多测试类复用同一种自定义Clock的场景，把配置逻辑抽离出来更整洁。

小提醒 #

如果你的全局模拟Clock已经标记了@Primary，那自定义的Clock Bean也需要加上@Primary才能确保覆盖成功，避免出现Bean冲突的问题。

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