PowerMock使用遇到的问题——2

如果在测一个类的某一个方法时，这个方法还调用了此类的其他方法，那么如何指定其他方法的返回值呢？

Partial mock local private method or public method in the class and suppress static initial block

public class Calc {

1. static {
2. System.out.println("hahaha");
3. }
4. public int add(int a, int b) {
5. return interADD(a, b);
6. }
7. private int interADD(int a, int b) {
8. return a + b;
9. }
10. public int minus(int a, int b) {
11. return interMinus(a, b);
12. }
13. public int interMinus(int a, int b) {
14. return a-b;
15. }
16. }
17. package com.kevin.util;
18. import static org.junit.Assert.*;
19. import junit.framework.Assert;
20. import org.easymock.EasyMock;
21. import org.junit.Test;
22. import org.junit.runner.RunWith;
23. import org.powermock.api.easymock.PowerMock;
24. import org.powermock.core.classloader.annotations.PrepareForTest;
25. import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor;
26. import org.powermock.modules.junit4.PowerMockRunner;
27. @RunWith(PowerMockRunner.class)
28. @PrepareForTest({Calc.class})
29. //suppress static initial
30. @SuppressStaticInitializationFor({"Calc.class"})
31. public class CalcTest {
32. //mock private method in class that be testing
33. @Test
34. public void testAdd() {
35. Calc createPartialMock = PowerMock.createPartialMock(Calc.class, "interADD");
36. try {
37. PowerMock.expectPrivate(createPartialMock, "interADD",2,3).andReturn(1000);
38. PowerMock.replay(createPartialMock);
39. int result = createPartialMock.add(2, 3);
40. Assert.assertEquals(1000, result);
41. } catch (Exception e) {
42. // TODO Auto-generated catch block
43. fail();
44. }
45. }
46. //mock public method int class that be testing
47. @Test
48. public void testMinus() {
49. Calc createPartialMock = PowerMock.createPartialMock(Calc.class, "interMinus");
50. try {
51. EasyMock.expect(createPartialMock.interMinus(5, 4)).andReturn(1000);
52. EasyMock.replay(createPartialMock);
53. int result = createPartialMock.minus(5, 4);
54. Assert.assertEquals(1000, result);
55. } catch (Exception e) {
56. // TODO Auto-generated catch block
57. fail();
58. }
59. }
60. }