我写了一个工厂来产生java.sql.Connection对象:
public class MySQLDatabaseConnectionFactory implements DatabaseConnectionFactory {
@Override public Connection getConnection() {
try {
return DriverManager.getConnection(...);
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
}
我想验证传递给DriverManager的参数。getConnection，但我不知道如何模拟静态方法。我的测试用例使用JUnit 4和Mockito。是否有一个好的方法来模拟/验证这个特定的用例?
当前回答
避免使用无法避免的静态方法的典型策略是创建包装对象并使用包装对象。
包装器对象成为真正静态类的外观，您不需要测试它们。
包装器对象可以是这样的
public class Slf4jMdcWrapper {
public static final Slf4jMdcWrapper SINGLETON = new Slf4jMdcWrapper();
public String myApisToTheSaticMethodsInSlf4jMdcStaticUtilityClass() {
return MDC.getWhateverIWant();
}
}
最后，被测试的类可以通过以下方式使用这个单例对象，例如: 在现实生活中使用默认构造函数:
public class SomeClassUnderTest {
final Slf4jMdcWrapper myMockableObject;
/** constructor used by CDI or whatever real life use case */
public myClassUnderTestContructor() {
this.myMockableObject = Slf4jMdcWrapper.SINGLETON;
}
/** constructor used in tests*/
myClassUnderTestContructor(Slf4jMdcWrapper myMock) {
this.myMockableObject = myMock;
}
}
这里您有一个可以很容易测试的类，因为您不直接使用带有静态方法的类。
如果您正在使用CDI，并且可以使用@Inject注释，那么就更容易了。 只需将Wrapper bean设置为@ApplicationScoped，将它作为协作器注入(甚至不需要复杂的构造函数进行测试)，然后继续模拟。
其他回答
观察:当您在静态实体中调用静态方法时，您需要更改@PrepareForTest中的类。
例如:
securityAlgo = MessageDigest.getInstance(SECURITY_ALGORITHM);
对于上面的代码，如果您需要模拟MessageDigest类，请使用
@PrepareForTest(MessageDigest.class)
而如果你有如下内容:
public class CustomObjectRule {
object = DatatypeConverter.printHexBinary(MessageDigest.getInstance(SECURITY_ALGORITHM)
.digest(message.getBytes(ENCODING)));
}
然后，您需要准备这个代码所在的类。
@PrepareForTest(CustomObjectRule.class)
然后模拟这个方法:
PowerMockito.mockStatic(MessageDigest.class);
PowerMockito.when(MessageDigest.getInstance(Mockito.anyString()))
.thenThrow(new RuntimeException());
避免使用无法避免的静态方法的典型策略是创建包装对象并使用包装对象。
包装器对象成为真正静态类的外观，您不需要测试它们。
包装器对象可以是这样的
public class Slf4jMdcWrapper {
public static final Slf4jMdcWrapper SINGLETON = new Slf4jMdcWrapper();
public String myApisToTheSaticMethodsInSlf4jMdcStaticUtilityClass() {
return MDC.getWhateverIWant();
}
}
最后，被测试的类可以通过以下方式使用这个单例对象，例如: 在现实生活中使用默认构造函数:
public class SomeClassUnderTest {
final Slf4jMdcWrapper myMockableObject;
/** constructor used by CDI or whatever real life use case */
public myClassUnderTestContructor() {
this.myMockableObject = Slf4jMdcWrapper.SINGLETON;
}
/** constructor used in tests*/
myClassUnderTestContructor(Slf4jMdcWrapper myMock) {
this.myMockableObject = myMock;
}
}
这里您有一个可以很容易测试的类，因为您不直接使用带有静态方法的类。
如果您正在使用CDI，并且可以使用@Inject注释，那么就更容易了。 只需将Wrapper bean设置为@ApplicationScoped，将它作为协作器注入(甚至不需要复杂的构造函数进行测试)，然后继续模拟。
使用JMockit框架。这对我很管用。您不必为模拟dbconnection . getconnection()方法编写语句。只要下面的代码就足够了。
下面的@Mock是mockit。模拟方案
Connection jdbcConnection = Mockito.mock(Connection.class);
MockUp<DBConnection> mockUp = new MockUp<DBConnection>() {
DBConnection singleton = new DBConnection();
@Mock
public DBConnection getInstance() {
return singleton;
}
@Mock
public Connection getConnection() {
return jdbcConnection;
}
};
我还编写了Mockito和AspectJ的组合:https://github.com/iirekm/varia/tree/develop/ajmock
你的例子是:
when(() -> DriverManager.getConnection(...)).thenReturn(...);
我也遇到过类似的问题。根据PowerMock的mockStatic文档，接受的答案对我不起作用，直到我做出了更改:@PrepareForTest(classthatcontainsstaticmethod .class)。
我不需要使用BDDMockito。
我的类:
public class SmokeRouteBuilder {
public static String smokeMessageId() {
try {
return InetAddress.getLocalHost().getHostAddress();
} catch (UnknownHostException e) {
log.error("Exception occurred while fetching localhost address", e);
return UUID.randomUUID().toString();
}
}
}
我的测试类:
@RunWith(PowerMockRunner.class)
@PrepareForTest(SmokeRouteBuilder.class)
public class SmokeRouteBuilderTest {
@Test
public void testSmokeMessageId_exception() throws UnknownHostException {
UUID id = UUID.randomUUID();
mockStatic(InetAddress.class);
mockStatic(UUID.class);
when(InetAddress.getLocalHost()).thenThrow(UnknownHostException.class);
when(UUID.randomUUID()).thenReturn(id);
assertEquals(id.toString(), SmokeRouteBuilder.smokeMessageId());
}
}
