如何使用JMockit模拟Date类的默认构造函数？

Question

我想模拟 java.util.date 的默认构造函数，因此它不会构造 一个 Date 对象，表示它的创建时间，但始终是相同的 Date 对象（在我的示例中为 2010 年 12 月 31 日）。我尝试使用 JMockit 和 JUnit 执行此操作，但在执行下面的测试时，输出始终为 Thu Jan 01 01:00:00 CET 1970。那么我的 Date() 模拟有什么问题吗？

import java.util.Date;

import org.junit.*;
import mockit.*;

public class AppTest {

    @Before
    public void setUp() {
        Mockit.setUpMocks(MockedDate.class);
    }

    @After
    public void tearDown() {
        Mockit.tearDownMocks();
    }  

   @Test
    public void testDate() {
        Date today=new Date();
        System.out.println(today.toString());
    }

    @MockClass(realClass=Date.class)
    public static class MockedDate {

        @Mock
        public void $init() {
            // Now should be always 31.12.2010!
            new Date(110,11,31);  //110 = 2010! 11 = December! This is sick!
        }
    }
}

Answer 1

al nik 的回答对我来说是一个很好的提示。最好模拟 System 类而不是 Date 类来生成假时间。我自己的解决方案最终只是简单地模拟 System.currentTimeMillis() 方法（该方法由 Date() 内部调用）。

JMockit 1.5 及更高版本

new MockUp<System>(){

    @Mock
    public long currentTimeMillis() {

        // Now is always 11/11/2011
        Date fake = new Date(111,10,11);
        return fake.getTime();
    }
};

JMockit 1.4 及更早版本

@MockClass(realClass = System.class)
public static class MockedSystem {

    @Mock
    public long currentTimeMillis() {

        // Now is always 11/11/2011
        Date fake = new Date(111,10,11);
        return fake.getTime();
    }
}

Answer 2

正如测试驱动一书中所建议的，在您的应用程序中使用 SystemTime 抽象是一种很好的做法。 java 类。
将方法调用（System#currentTimeMillis 和 Calendar#getInstance）和直接构造（new Date()）替换为静态方法调用，例如：

long time = SystemTime.asMillis();
Calendar calendar = SystemTime.asCalendar();
Date date = SystemTime.asDate();

要伪造时间，您只需修改 SystemTime 类返回的内容即可。
SystemTime 使用 TimeSource 接口，默认情况下委托给 System.currentTimeMillis()

public interface TimeSource {
    long millis();
}

一个可配置的 SystemTime 实现可能是这样的，

public class SystemTime {
    private static final TimeSource defaultSrc =
            new TimeSource() {
                public long millis() {
                    return System.currentTimeMillis();
                }
            };

    private static TimeSource source = null;
    public static long asMillis() {
        return getTimeSource().millis();
    }

    public static Date asDate() {
        return new Date(asMillis());
    }
    public static void reset() {
        setTimeSource(null);
    }
    public static void setTimeSource(TimeSource source) {
        SystemTime.source = source;
    }
    private static TimeSource getTimeSource() {
        return (source != null ? source : defaultSrc);
    }
}

并且要伪造返回的时间，您只需执行

@Test
public void clockReturnsFakedTimeInMilliseconds() throws Exception {
    final long fakeTime = 123456790L;
    SystemTime.setTimeSource(new TimeSource() {
        public long millis() {
                return fakeTime;
        }
    });
    long clock = SystemTime.asMillis();
    assertEquals("Should return fake time", fakeTime, clock);
}

Joda-Time 库 简化了 Java 中的日期处理，并为您提供了开箱即用的功能

Answer 3

您模拟了构造函数，并在内部创建了一个 Date 实例（与构造的实例无关），然后将其丢弃。由于默认构造函数是模拟出来的，因此日期不会初始化为当前时间，因此您得到的时间为零（代表 1970-01-01）。

要修改返回的日期，您需要使用神奇的“it”属性，如下所示：

@MockClass(realClass=Date.class)
public static class MockedDate {

    public Date it;
    @Mock
    public void $init() {
        // This is sick!
        it.setDate(31);
        it.setYear(110); // 110 = 2010!
        it.setMonth(11); // 11 = December!
    }
}

Answer 4

这是一个基于 @asmaier 的精彩答案的完整 JUnit 示例：

    @Test
    public void dateConstructorReturnsMockedDate() throws ParseException {
        final SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd");
        final Date mockDate = dateFormat.parse("2002-02-02");

        new MockUp<System>(){
            @Mock
            public long currentTimeMillis() {
                return mockDate.getTime();
            }
        };

        final Date actualDate = new Date();
        assertThat(actualDate).isEqualTo(mockDate); // using AssertJ
    }

当使用 Maven 时，在 pom.xml 中按如下方式配置 JMockit：

    <dependencies>
        <dependency>
            <groupId>org.jmockit</groupId>
            <artifactId>jmockit</artifactId>
            <version>${jmockit.version}</version>
            <scope>test</scope>
        </dependency>
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <artifactId>maven-surefire-plugin</artifactId>
                <version>${maven-surefire-plugin.version}</version>
                <configuration>
                    <argLine>
                        -javaagent:${settings.localRepository}/org/jmockit/jmockit/${jmockit.version}/jmockit-${jmockit.version}.jar
                    </argLine>
                    <disableXmlReport>true</disableXmlReport>
                </configuration>
            </plugin>
        </plugins>
    </build>
    <properties>
        <jmockit.version>1.44</jmockit.version>
    </properties>