unit tests for creating fcr:content

fcrepo · Jun 14, 2013 · 410a733 · 410a733
1 parent 59cf6f8
commit 410a733
Showing 1 changed file with 56 additions and 0 deletions.
diff --git a/fcrepo-http-api/src/test/java/org/fcrepo/api/FedoraContentTest.java b/fcrepo-http-api/src/test/java/org/fcrepo/api/FedoraContentTest.java
@@ -20,13 +20,15 @@
 import javax.jcr.RepositoryException;
 import javax.jcr.Session;
 import javax.ws.rs.core.EntityTag;
+import javax.ws.rs.core.MediaType;
 import javax.ws.rs.core.Request;
 import javax.ws.rs.core.Response;
 import javax.ws.rs.core.Response.Status;
 
 import org.apache.commons.io.IOUtils;
 import org.fcrepo.Datastream;
 import org.fcrepo.exception.InvalidChecksumException;
+import org.fcrepo.identifiers.PidMinter;
 import org.fcrepo.services.DatastreamService;
 import org.fcrepo.services.NodeService;
 import org.fcrepo.test.util.TestHelpers;
@@ -87,6 +89,60 @@ public void testPutContent() throws RepositoryException, IOException,
         verify(mockSession).save();
     }
 
+    @Test
+    public void testCreateContent() throws RepositoryException, IOException,
+                                                            InvalidChecksumException {
+        final String pid = "FedoraDatastreamsTest1";
+        final String dsId = "xyz";
+        final String dsContent = "asdf";
+        final String dsPath = "/" + pid + "/" + dsId;
+        final InputStream dsContentStream = IOUtils.toInputStream(dsContent);
+        final Node mockNode = mock(Node.class);
+
+        when(mockNode.isNew()).thenReturn(true);
+        when(mockNodeService.exists(mockSession, dsPath)).thenReturn(false);
+        when(
+                    mockDatastreams.createDatastreamNode(any(Session.class),
+                                                                eq(dsPath), anyString(), any(InputStream.class)))
+                .thenReturn(mockNode);
+        when(mockDatastreams.exists(mockSession, dsPath)).thenReturn(true);
+        final Response actual =
+                testObj.create(createPathList(pid, dsId), null,  null, MediaType.TEXT_PLAIN_TYPE,
+                                      dsContentStream);
+        assertEquals(Status.CREATED.getStatusCode(), actual.getStatus());
+        verify(mockDatastreams).createDatastreamNode(mockSession,dsPath, "text/plain", dsContentStream, null, null);
+        verify(mockSession).save();
+    }
+
+    @Test
+    public void testCreateContentAtMintedPath() throws RepositoryException, IOException,
+                                                InvalidChecksumException {
+        final String pid = "FedoraDatastreamsTest1";
+        final String dsId = "fcr:new";
+        final String dsContent = "asdf";
+        final String dsPath = "/" + pid + "/" + dsId;
+        final InputStream dsContentStream = IOUtils.toInputStream(dsContent);
+        final Node mockNode = mock(Node.class);
+
+        PidMinter mockMinter = mock(PidMinter.class);
+        when(mockMinter.mintPid()).thenReturn("xyz");
+        testObj.setPidMinter(mockMinter);
+        when(mockNode.isNew()).thenReturn(true);
+        when(mockNodeService.exists(mockSession, dsPath)).thenReturn(false);
+        when(
+                    mockDatastreams.createDatastreamNode(any(Session.class),
+                                                                eq(dsPath), anyString(), any(InputStream.class)))
+                .thenReturn(mockNode);
+        when(mockDatastreams.exists(mockSession, dsPath)).thenReturn(true);
+        final Response actual =
+                testObj.create(createPathList(pid, dsId), null,  null, MediaType.TEXT_PLAIN_TYPE,
+                                             dsContentStream);
+        assertEquals(Status.CREATED.getStatusCode(), actual.getStatus());
+        verify(mockDatastreams).createDatastreamNode(mockSession,
+                                                            "/" + pid + "/xyz", "text/plain", dsContentStream, null, null);
+        verify(mockSession).save();
+    }
+
     @Test
     public void testModifyContent() throws RepositoryException, IOException,
             InvalidChecksumException {