PowerMockito使用示例

71 阅读 0 评论 47 点赞

我是靠谱客的博主着急镜子，最近开发中收集的这篇文章主要介绍PowerMockito使用示例，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

1.mock私有函数或者静态私有函数

public class User{
	private String getUser(String id){
		return "abc";
	}
}

@RunWith(PowerMockRunner.class)
public class Test1{

	@Mock
	User user;
	@Test
	private String test(){
		Method method = PowerMockito.method(User.class, "getUser", String.class);
		try {
            String userName = method.invoke(user, "123");  //如果是静态私有函数，也是要一个实例化的对象
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        }
	}
}

2.禁止执行类的static代码段

package com.abc.test;

public class User{
	static {
        System.out.println("run static");
    }
    
	private String getUser(String id){
		return "abc";
	}
}


@RunWith(PowerMockRunner.class)
@SuppressStaticInitializationFor("com.abc.test.User")   //加上这个注解，就不会执行User类的static代码段
public class Test1{
	@Test
	private String test(){
		//...
	}
}

3.mock静态函数

package com.abc.test;

public class User{
	public static String getUser(String id){
		return "abc";
	}
}


@RunWith(PowerMockRunner.class)
@PrepareForTest({ User.class})           //第一步需要把类加到PrepareForTest注解里，多个可以用逗号隔开
public class Test1{
	@Test
	private String test(){
		PowerMockito.mockStatic(User.class);      //第二步
		PowerMockito.when(User.getUser(Mockito.anyString())).thenReturn("edf");   //第三步指定执行该函数时要返回的想要的值
		//...
	}
}

4 代码中针对某个类的所有实例化动作都返回指定的实例

package com.abc.test;

public class User{
	public static String getUser(String id){
		Op op = new Op();
		return op.getUser(id);
	}
}

public class Op{
	public String getUser(String id){
		return "abc";
	}
}


@RunWith(PowerMockRunner.class)
//重点
@PrepareForTest({User.class})
public class Test1{
	@Mock
	Op op1;
	@Test
	private String test(){
		 PowerMockito.whenNew(Op.class).withNoArguments().thenReturn(op1);    
		 String name = User.getUser("123");      //getUser()函数里实例化的op实例，将是op1这个实例
		//...
	}
}

5.修改类实例私有字段


public class User{
    private String str;
	public static String getUser(){
		return str;
	}
}


@RunWith(PowerMockRunner.class)
public class Test1{
	@Test
	private String test(){
		 User user = Mockito.mock(User.class);
		 MemberModifier.field(User.class, "str").set(user, "mock string");   //修改指定实例的指定私有字段
		 String name = User.getUser();      //将获取"mock string"
		//...
	}
}

6.修改类的静态私有字段


public class User{
    private static String str;
	public static String getUser(){
		return str;
	}
}


@RunWith(PowerMockRunner.class)
public class Test1{
	@Test
	private String test(){
		 Whitebox.setInternalState(User.class, "str",  "mock string");   //修改指定实例的指定私有字段
		 String name = User.getUser();      //将获取"mock string"
		//...
	}
}