@Test(expected = InvalidModelException.class)
public void testCreateS3AttachmentTokenBadMD5()
throws NumberFormatException, DatastoreException, NotFoundException, UnauthorizedException,
InvalidModelException {
S3AttachmentToken startToken = new S3AttachmentToken();
startToken.setFileName("SomeFile.jpg");
String almostMd5 = "79054025255fb1a26e4bc422aef";
startToken.setMd5(almostMd5);
Long tokenId = new Long(456);
String entityId = "132";
String userId = "007";
String expectedPath = entityId + "/" + tokenId.toString();
String expectePreSigneUrl = "I am a presigned url! whooot!";
when(mocIdGenerator.generateNewId()).thenReturn(tokenId);
Credentials mockCreds = Mockito.mock(Credentials.class);
when(mockUuserManager.getUserInfo(userId)).thenReturn(mockUser);
when(mockPermissionsManager.hasAccess(entityId, ACCESS_TYPE.UPDATE, mockUser)).thenReturn(true);
when(mocKLocationHelper.createFederationTokenForS3(userId, HttpMethod.PUT, expectedPath))
.thenReturn(mockCreds);
when(mocKLocationHelper.presignS3PUTUrl(mockCreds, expectedPath, almostMd5, "image/jpeg"))
.thenReturn(expectePreSigneUrl);
// Make the actual call
S3AttachmentToken endToken = manager.createS3AttachmentToken(userId, entityId, startToken);
assertNotNull(endToken);
assertEquals(expectePreSigneUrl, endToken.getPresignedUrl());
}
@Before
public void setUp() throws DatastoreException, InvalidModelException, NotFoundException {
// create a node
Node toCreate = NodeTestUtils.createNew(name, userId);
toCreate.setVersionComment("This is the first version of the first node ever!");
toCreate.setVersionLabel("0.0.1");
nodeId = nodeDAO.createNew(toCreate).substring(3); // trim "syn" from node ID
// Initialize a new Evaluation
EvaluationDBO evaluation = new EvaluationDBO();
evaluation.setId(evalId);
evaluation.seteTag("etag");
evaluation.setName("name");
evaluation.setOwnerId(userId);
evaluation.setContentSource(KeyFactory.ROOT_ID);
evaluation.setCreatedOn(System.currentTimeMillis());
evaluation.setStatusEnum(EvaluationStatus.PLANNED);
evalId = dboBasicDao.createNew(evaluation).getId();
// Initialize a new Participant
ParticipantDBO participant = new ParticipantDBO();
participant.setUserId(userId);
participant.setEvalId(evalId);
participant.setCreatedOn(System.currentTimeMillis());
participant.setId(idGenerator.generateNewId(TYPE.PARTICIPANT_ID));
dboBasicDao.createNew(participant);
// Initialize a new Submission
SubmissionDBO submission = new SubmissionDBO();
submission.setId(submissionId);
submission.setName(name);
submission.setEntityId(Long.parseLong(nodeId));
submission.setVersionNumber(1L);
submission.setUserId(userId);
submission.setEvalId(evalId);
submission.setCreatedOn(System.currentTimeMillis());
dboBasicDao.createNew(submission);
}
