예제 #1
0
  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;
              }
            });
  }