private void mockContentImport(Owner owner, final Map<String, Content> contents) { when(contentManager.importContent(eq(owner), any(Map.class), any(Set.class))) .thenAnswer( new Answer<ImportResult<Content>>() { @Override public ImportResult<Content> answer(InvocationOnMock invocation) throws Throwable { Object[] args = invocation.getArguments(); Map<String, ContentData> contentData = (Map<String, ContentData>) args[1]; ImportResult<Content> importResult = new ImportResult<Content>(); Map<String, Content> output = importResult.getCreatedEntities(); if (contentData != null) { for (String pid : contentData.keySet()) { Content content = contents.get(pid); if (content != null) { output.put(content.getId(), content); } } } return importResult; } }); }