Mockito模拟静态方法– PowerMock
Mockito允许我们创建模拟对象。
由于静态方法属于该类,因此Mockito中无法模拟静态方法。
但是,我们可以结合使用PowerMock和Mockito框架来模拟静态方法。
使用PowerMock的Mockito Mock静态方法
PowerMock提供了不同的模块来扩展Mockito框架并运行JUnit和TestNG测试用例。
请注意,PowerMock还不支持JUnit 5,因此我们将创建JUnit 4测试用例。
我们还将学习如何将TestNG与Mockito和PowerMock集成。
PowerMock依赖项
我们需要遵循PowerMock依赖关系才能在Mockito中模拟静态方法。
powermock-api-mockito2:这是PowerMock的核心依赖项,用于扩展Mockito2模拟框架。
如果您使用的是Mockito 1.x版本,则使用powermock-api-mockito
模块。powermock-module-junit4
:用于使用PowerMock运行JUnit 4测试用例。powermock-module-testng
:用于运行TestNG测试用例并支持PowerMock。
以下是我们项目中的最终pom.xml。
<project xmlns="https://maven.apache.org/POM/4.0.0" xmlns:xsi="https://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="https://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd"> <modelVersion>4.0.0</modelVersion> <groupId>com.theitroad.powermock</groupId> <artifactId>PowerMock-Examples</artifactId> <version>0.0.1-SNAPSHOT</version> <properties> <testng.version>6.14.3</testng.version> <junit4.version>4.12</junit4.version> <mockito-core.version>2.19.0</mockito-core.version> <powermock.version>2.0.0-beta.5</powermock.version> <java.version>10</java.version> </properties> <dependencies> <!-- TestNG --> <dependency> <groupId>org.testng</groupId> <artifactId>testng</artifactId> <version>${testng.version}</version> <scope>test</scope> </dependency> <!-- JUnit 4 --> <dependency> <groupId>junit</groupId> <artifactId>junit</artifactId> <version>${junit4.version}</version> <scope>test</scope> </dependency> <!-- Mockito 2 --> <dependency> <groupId>org.mockito</groupId> <artifactId>mockito-core</artifactId> <version>${mockito-core.version}</version> <scope>test</scope> </dependency> <!-- PowerMock TestNG Module --> <dependency> <groupId>org.powermock</groupId> <artifactId>powermock-module-testng</artifactId> <version>${powermock.version}</version> <scope>test</scope> </dependency> <!-- PowerMock JUnit 4.4+ Module --> <dependency> <groupId>org.powermock</groupId> <artifactId>powermock-module-junit4</artifactId> <version>${powermock.version}</version> <scope>test</scope> </dependency> <!-- PowerMock Mockito2 API --> <dependency> <groupId>org.powermock</groupId> <artifactId>powermock-api-mockito2</artifactId> <version>${powermock.version}</version> <scope>test</scope> </dependency> </dependencies> <build> <plugins> <plugin> <artifactId>maven-compiler-plugin</artifactId> <version>3.7.0</version> <configuration> <source>${java.version}</source> <target>${java.version}</target> </configuration> </plugin> <plugin> <groupId>org.apache.maven.plugins</groupId> <artifactId>maven-surefire-plugin</artifactId> <version>2.22.0</version> <dependencies> <dependency> <groupId>org.apache.maven.surefire</groupId> <artifactId>surefire-junit47</artifactId> <version>2.22.0</version> </dependency> <dependency> <groupId>org.apache.maven.surefire</groupId> <artifactId>surefire-testng</artifactId> <version>2.22.0</version> </dependency> </dependencies> <configuration> <additionalClasspathElements> <additionalClasspathElement>src/test/java/</additionalClasspathElement> </additionalClasspathElements> <!-- TestNG Test Fails when executed from command line with message "Cannot use a threadCount parameter less than 1" Works when threadCount is explicitly specified https://gist.github.com/juherr/6eb3e93e2db33979b7e90b63ddadc888--> <threadCount>5</threadCount> </configuration> </plugin> </plugins> </build> </project>
请注意,我使用的是PowerMock的2.0.0-beta.5版本。
此版本支持Java 10,但是仍处于测试阶段,因此在复杂情况下可能会出现一些问题。
当我尝试使用当前的稳定版本1.7.x
时,出现以下错误。
java.lang.NoSuchMethodError: org.mockito.internal.handler.MockHandlerFactory.createMockHandler(Lorg/mockito/mock/MockCreationSettings;)Lorg/mockito/internal/InternalMockHandler; at org.powermock.api.mockito.internal.mockcreation.DefaultMockCreator.createMethodInvocationControl(DefaultMockCreator.java:114) at org.powermock.api.mockito.internal.mockcreation.DefaultMockCreator.createMock(DefaultMockCreator.java:69) at org.powermock.api.mockito.internal.mockcreation.DefaultMockCreator.mock(DefaultMockCreator.java:46) at org.powermock.api.mockito.PowerMockito.mockStatic(PowerMockito.java:73)
以下是针对此例外而开放的GitHub问题-issue1和issue2。
让我们用静态方法创建一个简单的类。
package com.theitroad.mockito.staticmethod; public class Utils { public static boolean print(String msg) { System.out.println("Printing "+msg); return true; } }
JUnit Mockito PowerMock示例
我们需要执行以下操作以将PowerMock与Mockito和JUnit 4集成。
用@RunWith(PowerMockRunner.class)批注注释测试类。
用
@ PrepareForTest
注释测试类,并提供使用PowerMock模拟的分类。使用
PowerMockito.mockStatic()
通过静态方法模拟类。使用
PowerMockito.verifyStatic()
来验证使用Mockito的模拟方法。
这是在JUnit测试案例中使用Mockito和PowerMock模拟静态方法的完整示例。
package com.theitroad.mockito.staticmethod; import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertTrue; import static org.mockito.ArgumentMatchers.anyString; import static org.mockito.Mockito.atLeast; import static org.mockito.Mockito.when; import org.junit.Test; import org.junit.runner.RunWith; import org.powermock.api.mockito.PowerMockito; import org.powermock.core.classloader.annotations.PrepareForTest; import org.powermock.modules.junit4.PowerMockRunner; @RunWith(PowerMockRunner.class) @PrepareForTest(Utils.class) public class JUnit4PowerMockitoStaticTest{ @Test public void test_static_mock_methods() { PowerMockito.mockStatic(Utils.class); when(Utils.print("Hello")).thenReturn(true); when(Utils.print("Wrong Message")).thenReturn(false); assertTrue(Utils.print("Hello")); assertFalse(Utils.print("Wrong Message")); PowerMockito.verifyStatic(Utils.class, atLeast(2)); Utils.print(anyString()); } }
TestNG Mockito PowerMock示例
对于TestNG测试用例,我们不需要使用@ RunWith
注释。
我们需要测试类来扩展" PowerMockTestCase",以便使用" PowerMockObjectFactory"来创建测试类实例。
package com.theitroad.mockito.staticmethod; import static org.testng.Assert.assertFalse; import static org.testng.Assert.assertTrue; import static org.mockito.Mockito.*; import org.powermock.api.mockito.PowerMockito; import org.powermock.core.classloader.annotations.PrepareForTest; import org.powermock.modules.testng.PowerMockTestCase; import org.testng.annotations.Test; @PrepareForTest(Utils.class) public class TestNGPowerMockitoStaticTest extends PowerMockTestCase{ @Test public void test_static_mock_methods() { PowerMockito.mockStatic(Utils.class); when(Utils.print("Hello")).thenReturn(true); when(Utils.print("Wrong Message")).thenReturn(false); assertTrue(Utils.print("Hello")); assertFalse(Utils.print("Wrong Message")); PowerMockito.verifyStatic(Utils.class); Utils.print("Hello"); PowerMockito.verifyStatic(Utils.class, times(2)); Utils.print(anyString()); } }