Mockito-PowerMock。模拟方法调用

Question

@Resource
private Bar bar;

public int methodToTest(String arg1) {
    String crucialValue = SomeClass.methodToMock(bar.getterToMock());
        Method Logic Here 
        ....
}

private BarContext barContext = mock(BarContext.class)
@Mock
private Bar bar;

@Test
public void testTheMethod(){
    PowerMockito.mockStatic(SomeClass.class);
    PowerMockito.doReturn(barContext).when(bar).getterToMock();
    PowerMockito.doReturn("Bingo").when(SomeClass.methodToMock(barContext));
    Foo foo = new Foo();
    foo.methodToTest("foo");
}

@WebService
public class FileUploadServiceImpl implements FileUploadService {

private final String soapFilePath = "target/;
private final static Map<String, String> userFileMap = new HashMap<>();
private final static AtomicInteger id = new AtomicInteger();

@Resource
private WebServiceContext webServiceContext;

@Override
public int createFile(@WebParam String name) {
    String username = Utils.getUsername(webServiceContext.getMessageContext());

    try {
        File userDir = new File(filePath + username);
        userDir.mkdir();
        File file = new File(userDir.getPath() + "/" + name);
        file.createNewFile();
        userFileMap.put(username + "_" + id.get(), name);
    } catch (IOException e) {
        LOG.error(e.getMessage(), e);
        e.printStackTrace();
    }
    return id.getAndIncrement();
   }
}

public static String getUsername(MessageContext context){
            Map httpHeaders = (Map) context.get(MessageContext.HTTP_REQUEST_HEADERS);
            if(httpHeaders != null){
                List<String> auth = (List<String>) httpHeaders.get("Authorization");
                if (auth != null && auth.get(0).startsWith("Basic")){
                    String base64Credentials = auth.get(0).substring("Basic".length()).trim();
                    String credentials = new String(Base64.getDecoder().decode(
                            base64Credentials.getBytes(Charset.forName("UTF-8"))));
                    return credentials.split(":", 2)[0];
                }
            }
        return null;
      }

@RunWith(PowerMockRunner.class)
@PrepareForTest(FileUploadServiceImpl.class)
public class ServiceTest {
private String username = "10237832";
private FileUploadService service;
private MessageContext messageContext = mock(MessageContext.class);

@Mock
private WebServiceContext webServiceContext;

@Test
public void startUploadFileTest(){
    service = new FileUploadServiceImpl();
    mockStatic(Utils.class);
    when(webServiceContext.getMessageContext()).thenReturn(messageContext);
    when(Utils.getUsername(messageContext)).thenReturn(username);
    service.startUpload("test_file");
  }
}

异常堆栈跟踪

java.lang.NullPointerException
    at org.example.utils.Utils.getUsername(Utils.java:47)
    at org.example.services.ServiceTest.startUploadFileTest(ServiceTest.java:42)
    at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
    at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)
    at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
    at org.junit.internal.runners.TestMethod.invoke(TestMethod.java:68)
    at org.powermock.modules.junit4.internal.impl.PowerMockJUnit44RunnerDelegateImpl$PowerMockJUnit44MethodRunner.runTestMethod(PowerMockJUnit44RunnerDelegateImpl.java:310)
    at org.junit.internal.runners.MethodRoadie$2.run(MethodRoadie.java:88)
    at org.junit.internal.runners.MethodRoadie.runBeforesThenTestThenAfters(MethodRoadie.java:96)
    at org.powermock.modules.junit4.internal.impl.PowerMockJUnit44RunnerDelegateImpl$PowerMockJUnit44MethodRunner.executeTest(PowerMockJUnit44RunnerDelegateImpl.java:294)
    at org.powermock.modules.junit4.internal.impl.PowerMockJUnit47RunnerDelegateImpl$PowerMockJUnit47MethodRunner.executeTestInSuper(PowerMockJUnit47RunnerDelegateImpl.java:127)
    at org.powermock.modules.junit4.internal.impl.PowerMockJUnit47RunnerDelegateImpl$PowerMockJUnit47MethodRunner.executeTest(PowerMockJUnit47RunnerDelegateImpl.java:82)
    at org.powermock.modules.junit4.internal.impl.PowerMockJUnit44RunnerDelegateImpl$PowerMockJUnit44MethodRunner.runBeforesThenTestThenAfters(PowerMockJUnit44RunnerDelegateImpl.java:282)
    at org.junit.internal.runners.MethodRoadie.runTest(MethodRoadie.java:86)
    at org.junit.internal.runners.MethodRoadie.run(MethodRoadie.java:49)
    at org.powermock.modules.junit4.internal.impl.PowerMockJUnit44RunnerDelegateImpl.invokeTestMethod(PowerMockJUnit44RunnerDelegateImpl.java:207)
    at org.powermock.modules.junit4.internal.impl.PowerMockJUnit44RunnerDelegateImpl.runMethods(PowerMockJUnit44RunnerDelegateImpl.java:146)
    at org.powermock.modules.junit4.internal.impl.PowerMockJUnit44RunnerDelegateImpl$1.run(PowerMockJUnit44RunnerDelegateImpl.java:120)
    at org.junit.internal.runners.ClassRoadie.runUnprotected(ClassRoadie.java:33)
    at org.junit.internal.runners.ClassRoadie.runProtected(ClassRoadie.java:45)
    at org.powermock.modules.junit4.internal.impl.PowerMockJUnit44RunnerDelegateImpl.run(PowerMockJUnit44RunnerDelegateImpl.java:122)
    at org.powermock.modules.junit4.common.internal.impl.JUnit4TestSuiteChunkerImpl.run(JUnit4TestSuiteChunkerImpl.java:106)
    at org.powermock.modules.junit4.common.internal.impl.AbstractCommonPowerMockRunner.run(AbstractCommonPowerMockRunner.java:53)
    at org.powermock.modules.junit4.PowerMockRunner.run(PowerMockRunner.java:59)
    at org.junit.runner.JUnitCore.run(JUnitCore.java:160)
    at com.intellij.junit4.JUnit4IdeaTestRunner.startRunnerWithArgs(JUnit4IdeaTestRunner.java:78)
    at com.intellij.rt.execution.junit.JUnitStarter.prepareStreamsAndStart(JUnitStarter.java:212)
    at com.intellij.rt.execution.junit.JUnitStarter.main(JUnitStarter.java:68)
    at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
    at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)
    at com.intellij.rt.execution.application.AppMain.main(AppMain.java:140)

Answer 1

为什么不将用户名的获取抽象到一个类中呢？

    class UsernameRetriever {

        private final WebServiceContext webServiceContext;

        UsernameRetriever(WebServiceContext webServiceContext) {
           this.webServiceContext = webServiceContext;
        }


        String getUsernameFromContext() {
            return Utils.getUsername(webServiceContext.getMessageContext());
        }

    }

然后通过构造函数将其注入到类中（我不记得@inject和构造函数的语法，所以您自己查看一下）。

@WebService
public class FileUploadServiceImpl implements FileUploadService {

// SHOULD BE STATIC FINAL IF IT'S A CONSTANT - IT SHOULD BE PARAMETRIZED BTW
private static final String soapFilePath = "target/;
// YOU'RE GOING TO HAVE A RACE CONDITION - HASH MAP IS NOT THREAD SAFE
private final static Map<String, String> userFileMap = new HashMap<>();
private final static AtomicInteger id = new AtomicInteger();

private final UsernameRetriver usernameRetriever;

@Inject
public FileUploadServiceImpl(WebServiceContext webServiceContext) {
    this.usernameRetriever = new UsernameRetriever(webServiceContext);
}


@Inject
// for testing purposes
FileUploadServiceImpl(UsernameRetriever usernameRetriever) {
    this.usernameRetriever = usernameRetriever
}

@Override
public int createFile(@WebParam String name) {
    String username = usernameRetriever.getUsernameFromContext();

    try {
        File userDir = new File(filePath + username);
        userDir.mkdir();
        File file = new File(userDir.getPath() + "/" + name);
        file.createNewFile();
        userFileMap.put(username + "_" + id.get(), name);
    } catch (IOException e) {
        LOG.error(e.getMessage(), e);
        e.printStackTrace();
    }
    return id.getAndIncrement();
   }
}

这样:*不需要Powermock*您只负责*您已经正确地编写了代码；）