Updated tests according to fix.

dataquest-dev · Nov 15, 2023 · ed83394 · ed83394
1 parent ab71ae3
commit ed83394
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/...ce-server-webapp/src/test/java/org/dspace/app/rest/ClarinBitstreamImportControllerIT.java b/...ce-server-webapp/src/test/java/org/dspace/app/rest/ClarinBitstreamImportControllerIT.java
@@ -95,7 +95,7 @@ public void setup() throws Exception {
                 .withName("TESTINGBUNDLE")
                 .build();
         token = getAuthToken(admin.getEmail(), password);
-        bitstreamFormat = BitstreamFormatBuilder.createBitstreamFormat(context).build();
+        bitstreamFormat = BitstreamFormatBuilder.createBitstreamFormat(context).withMimeType("application/pdf").build();
         String input = "Hello, World!";
         MockMultipartFile file = new MockMultipartFile("file", "hello.txt",
                 MediaType.TEXT_PLAIN_VALUE,
@@ -147,7 +147,7 @@ public void importBitstreamForExistingFileWithBundleTest() throws Exception {
                         .contentType(contentType)
                         .param("internal_id", internalId)
                         .param("storeNumber", Integer.toString(storeNumber))
-                        .param("bitstreamFormat", Integer.toString(bitstreamFormat.getID()))
+                        .param("bitstreamFormat", bitstreamFormat.getMIMEType())
                         .param("deleted", Boolean.toString(deleted))
                         .param("sequenceId", Integer.toString(sequence))
                         .param("primaryBundle_id", "")
@@ -156,7 +156,7 @@ public void importBitstreamForExistingFileWithBundleTest() throws Exception {
                         .andReturn().getResponse().getContentAsString(),
                 "$.id"));
 
-        checkCreatedBitstream(uuid, internalId, storeNumber, bitstreamFormat.getID(), sequence, deleted, sizeBytes,
+        checkCreatedBitstream(uuid, internalId, storeNumber, bitstreamFormat.getMIMEType(), sequence, deleted, sizeBytes,
                 checkSum);
 
         //clean all
@@ -182,7 +182,7 @@ public void importBitstreamForExistingFileWithoutBundleTest() throws Exception {
                                 .contentType(contentType)
                                 .param("internal_id", internalId)
                                 .param("storeNumber", Integer.toString(storeNumber))
-                                .param("bitstreamFormat", Integer.toString(bitstreamFormat.getID()))
+                                .param("bitstreamFormat", bitstreamFormat.getMIMEType())
                                 .param("deleted", Boolean.toString(deleted))
                                 .param("sequenceId", Integer.toString(sequence))
                                 .param("primaryBundle_id", "")
@@ -191,7 +191,7 @@ public void importBitstreamForExistingFileWithoutBundleTest() throws Exception {
                         .andReturn().getResponse().getContentAsString(),
                 "$.id"));
 
-        checkCreatedBitstream(uuid, internalId, storeNumber, bitstreamFormat.getID(), sequence, deleted, sizeBytes,
+        checkCreatedBitstream(uuid, internalId, storeNumber, bitstreamFormat.getMIMEType(), sequence, deleted, sizeBytes,
                 checkSum);
 
         //clean all
@@ -217,7 +217,7 @@ public void importBitstreamForExistingFileAsPrimaryBitstreamOfBundleTest() throw
                                 .contentType(contentType)
                                 .param("internal_id", internalId)
                                 .param("storeNumber", Integer.toString(storeNumber))
-                                .param("bitstreamFormat", Integer.toString(bitstreamFormat.getID()))
+                                .param("bitstreamFormat", bitstreamFormat.getMIMEType())
                                 .param("deleted", Boolean.toString(deleted))
                                 .param("sequenceId", Integer.toString(sequence))
                                 .param("primaryBundle_id", bundle.getID().toString())
@@ -226,7 +226,7 @@ public void importBitstreamForExistingFileAsPrimaryBitstreamOfBundleTest() throw
                         .andReturn().getResponse().getContentAsString(),
                 "$.id"));
 
-        checkCreatedBitstream(uuid, internalId, storeNumber, bitstreamFormat.getID(), sequence, deleted, sizeBytes,
+        checkCreatedBitstream(uuid, internalId, storeNumber, bitstreamFormat.getMIMEType(), sequence, deleted, sizeBytes,
                 checkSum);
         bundle = bundleService.find(context, bundle.getID());
         assertEquals(bundle.getPrimaryBitstream().getID(), bitstream.getID());
@@ -257,7 +257,7 @@ public void importBitstreamForExistingFileValidationErrorTest() throws Exception
                         .contentType(contentType)
                         .param("internal_id", internalId)
                         .param("storeNumber", Integer.toString(storeNumber))
-                        .param("bitstreamFormat", Integer.toString(bitstreamFormat.getID()))
+                        .param("bitstreamFormat", bitstreamFormat.getMIMEType())
                         .param("deleted", Boolean.toString(deleted))
                         .param("sequenceId", Integer.toString(sequence))
                         .param("primaryBundle_id", "")
@@ -271,12 +271,12 @@ public void importBitstreamForExistingFileValidationErrorTest() throws Exception
     }
 
     private void checkCreatedBitstream(UUID uuid, String internalId, int storeNumber,
-                                       Integer bitstreamFormat, int sequence, boolean deleted, long sizeBytes,
+                                       String bitstreamFormat, int sequence, boolean deleted, long sizeBytes,
                                        String checkSum) throws SQLException {
         bitstream = bitstreamService.find(context, uuid);
         assertEquals(bitstream.getChecksum(), checkSum);
         assertEquals(bitstream.getSizeBytes(), sizeBytes);
-        assertEquals(bitstream.getFormat(context).getID(), bitstreamFormat);
+        assertEquals(bitstream.getFormat(context).getMIMEType(), bitstreamFormat);
         assertEquals(bitstream.getInternalId(), internalId);
         assertEquals(bitstream.getStoreNumber(), storeNumber);
         assertEquals(bitstream.getSequenceID(), sequence);