@Test public void testcreateS3ProfileToken() throws JSONObjectAdapterException, ServletException, IOException, DatastoreException { S3AttachmentToken startToken = new S3AttachmentToken(); startToken.setFileName("someImage.jpg"); startToken.setMd5(TEST_MD5); String fileName = "images/squarish.png"; URL toUpUrl = S3TokenControllerTest.class.getClassLoader().getResource(fileName); assertNotNull("Failed to find: " + fileName + " on the classpath", toUpUrl); File toUpload = new File(toUpUrl.getFile()); // Create the token S3AttachmentToken resultToken = testHelper.createS3AttachmentToken( TestUserDAO.TEST_USER_NAME, ServiceConstants.AttachmentType.USER_PROFILE, testUser.getId(), startToken); System.out.println(resultToken); assertNotNull(resultToken); assertNotNull(resultToken.getTokenId()); assertNotNull(resultToken.getPresignedUrl()); // Upload it String path = S3TokenManagerImpl.createAttachmentPathNoSlash(testUser.getId(), resultToken.getTokenId()); s3Utility.uploadToS3(toUpload, path); // Make sure we can get a signed download URL for this attachment. long now = System.currentTimeMillis(); long oneMinuteFromNow = now + (60 * 1000); PresignedUrl url = testHelper.getUserProfileAttachmentUrl( TestUserDAO.TEST_USER_NAME, testUser.getId(), resultToken.getTokenId()); System.out.println(url); assertNotNull(url); assertNotNull(url.getPresignedUrl()); URL urlReal = new URL(url.getPresignedUrl()); // Check that it expires quickly (not as important when it's the public user profile picture) String[] split = urlReal.getQuery().split("&"); assertTrue(split.length > 1); String[] expiresSplit = split[0].split("="); assertEquals("Expires", expiresSplit[0]); // It should expire within a minute max Long expirsInt = Long.parseLong(expiresSplit[1]); long expiresMil = expirsInt.longValue() * 1000l; System.out.println("Now: " + new Date(now)); System.out.println("Expires: " + new Date(expiresMil)); assertTrue("This URL should expire in under a minute!", expiresMil < oneMinuteFromNow); assertTrue("This URL should expire after now!", now <= expiresMil); // Delete the file s3Utility.deleteFromS3(path); }
@Test public void testGetAttachmentUrl() throws Exception { Long tokenId = new Long(456); String entityId = "132"; String expectedPath = S3TokenManagerImpl.createAttachmentPathSlash(entityId, tokenId.toString()); String expectePreSigneUrl = "I am a presigned url! whooot!"; when(mockUuserManager.getUserInfo(userId)).thenReturn(mockUser); when(mockPermissionsManager.hasAccess(entityId, ACCESS_TYPE.READ, mockUser)).thenReturn(true); when(mocKLocationHelper.presignS3GETUrlShortLived(userId, expectedPath)) .thenReturn(expectePreSigneUrl); when(mockS3Utilitiy.doesExist(any(String.class))).thenReturn(true); // Make the actual call when(mockUuserManager.getUserInfo(userId)).thenReturn(mockUser); PresignedUrl url = manager.getAttachmentUrl(userId, entityId, tokenId.toString()); assertNotNull(url); assertEquals(expectePreSigneUrl, url.getPresignedUrl()); assertEquals(URLStatus.READ_FOR_DOWNLOAD, url.getStatus()); }