Question

我必须为某些方法进行一些jUnit测试，但我无法更改源代码。是否有可能在没有更改源代码的情况下更改函数的行为？看一个直截了当的例子：A类和B类是源代码（不能改变它们）。我想在Junit测试中通过testing（）在B中调用run（）方法时改变run（）方法的行为。有什么想法吗？

public class A {
    public String run(){
        return "test";
    } 
}

public class B {
    public void testing() {
        String fromA = new A().run(); //I want a mocked result here
        System.out.println(fromA);
    }
}

public class C {
    @Test
    public void jUnitTest() {
        new B().testing();
        // And here i want to call testing method from B but with a "mock return" from run()         
    }
}

Answer 1

您可以使用Mockito和PowerMock：

import org.junit.Before;
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;

import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.when;

@RunWith(PowerMockRunner.class)
@PrepareForTest(B.class)
public class C {
    @Before
    public void setUp() throws Exception {
        A a = spy(new A());
        when(a.run()).thenReturn("mock return");
        PowerMockito.whenNew(A.class).withNoArguments().thenReturn(a);
    }

    @Test
    public void jUnitTest() {
        new B().testing();
    }
}

Answer 2

您无法以您想要的方式测试，也无法更改源代码。你不能模拟局部变量。我建议您向B.class添加新方法，该方法将使用Mockito返回new A()和spy。

@Test
public void test(){
    final B spy = Mockito.spy(new B());    
    Mockito.doReturn(new C()).when(spy).getA();
} 

class C extends A {
    @Override
    public String run(){return "new String";}
}

在测试中使用字节码库并不是一个好主意。

jUnit更改方法

2 个答案: