/** ******************************************************************************** */
  private void assertContentItem(String data, String mimeTypeRawData, String expectedFileSuffix)
      throws Exception {
    // Simulates what ContentFrameworkImpl would do
    String uuid = UUID.randomUUID().toString().replaceAll("-", "");
    ContentItem contentItem =
        new IncomingContentItem(uuid, IOUtils.toInputStream(data), mimeTypeRawData);
    CreateRequest createRequest = new CreateRequestImpl(contentItem, null);
    CreateResponse createResponse = provider.create(createRequest);
    ContentItem createdContentItem = createResponse.getCreatedContentItem();

    assertNotNull(createdContentItem);
    String id = createdContentItem.getId();
    assertNotNull(id);
    assertThat(id, equalTo(uuid));

    String contentUri = createdContentItem.getUri();
    LOGGER.debug("contentUri = {}", contentUri);
    assertNotNull(contentUri);
    String expectedContentUri = FileSystemProvider.CONTENT_URI_PREFIX + uuid;
    assertThat(contentUri, equalTo(expectedContentUri));

    File file = createdContentItem.getFile();
    assertNotNull(file);
    assertTrue(file.exists());
    assertTrue(createdContentItem.getSize() > 0);
    assertEquals(mimeTypeRawData, createdContentItem.getMimeTypeRawData());
    assertEquals(data, IOUtils.toString(createdContentItem.getInputStream()));
  }
Exemplo n.º 2
0
  protected ContentItem getMockGoodContentItem(
      String content, String fileName, String contentId, Long size, String mimeType)
      throws IOException, MimeTypeParseException {
    ContentItem item = mock(ContentItem.class);
    when(item.getInputStream()).thenReturn(new ByteArrayInputStream(content.getBytes()));
    when(item.getFilename()).thenReturn(fileName);
    when(item.getId()).thenReturn(contentId);
    if (size != null) {
      when(item.getSize()).thenReturn(size);
    } else {
      when(item.getSize()).thenThrow(new IOException("IOException"));
    }
    when(item.getMimeTypeRawData()).thenReturn(mimeType);
    if (mimeType != null) {
      when(item.getMimeType()).thenReturn(new MimeType(mimeType));
    }

    return item;
  }
  @Test
  public void testRead() throws Exception {
    CreateResponse createResponse =
        storeContentItem(TEST_INPUT_CONTENTS, NITF_MIME_TYPE, TEST_INPUT_FILENAME);
    String id = createResponse.getCreatedContentItem().getId();
    ReadRequest readRequest = new ReadRequestImpl(id, null);

    ReadResponse readResponse = provider.read(readRequest);
    ContentItem item = readResponse.getContentItem();

    LOGGER.debug("Item retrieved: {}", item);
    assertEquals(id, item.getId());
    assertEquals(NITF_MIME_TYPE, item.getMimeTypeRawData());

    String expectedFilePath = BASE_DIR + File.separator + id + File.separator + item.getFilename();
    assertThat(item.getFile().getAbsolutePath(), endsWith(expectedFilePath));
    assertTrue(item.getSize() > 0);
    assertTrue(item.getFile().exists());
  }
  @Test
  public void testUpdate() throws Exception {
    CreateResponse createResponse =
        storeContentItem(TEST_INPUT_CONTENTS, NITF_MIME_TYPE, TEST_INPUT_FILENAME);
    String id = createResponse.getCreatedContentItem().getId();
    ContentItem updateItem =
        new IncomingContentItem(id, IOUtils.toInputStream("Updated NITF"), NITF_MIME_TYPE);
    UpdateRequest updateRequest = new UpdateRequestImpl(updateItem);

    UpdateResponse updateResponse = provider.update(updateRequest);
    ContentItem item = updateResponse.getUpdatedContentItem();

    LOGGER.debug("Item retrieved: {}", item);
    assertEquals(id, item.getId());
    assertEquals(NITF_MIME_TYPE, item.getMimeTypeRawData());

    String expectedFilePath = BASE_DIR + File.separator + id + File.separator + item.getFilename();
    assertThat(item.getFile().getAbsolutePath(), endsWith(expectedFilePath));
    assertTrue(item.getSize() > 0);
    assertTrue(item.getFile().exists());

    String updatedFileContents = getFileContentsAsString(item.getFile().getAbsolutePath());
    assertEquals("Updated NITF", updatedFileContents);
  }