import java.io.File;
import java.util.HashMap;
import java.util.Map;
import org.junit.Before;
import org.junit.Test;
import yalp.Logger;
import yalp.Yalp;
import yalp.data.MemoryUpload;
import yalp.data.Upload;
import yalp.mvc.Http;
import yalp.mvc.Http.Response;
import yalp.mvc.results.Redirect;
import yalp.test.Fixtures;
import yalp.test.FunctionalTest;
import controllers.Binary;
public class BinaryTest extends FunctionalTest {
@Before
public void setUp() {
Fixtures.deleteAll(); // see Bug #491403
Fixtures.deleteDirectory("attachments");
URL deleteURL = reverse(); {
Binary.deleteAll();
}
Response deletedResponse = GET(deleteURL);
assertStatus(200, deletedResponse);
}
@Test
public void testUploadSomething() {
URL imageURL = reverse(); {
Binary.showAvatar(1l);
}
Response getResponse = GET(imageURL);
assertStatus(404, getResponse);
URL url = reverse(); {
Binary.save(null);
}
Map<String,String> parameters= new HashMap<String,String>();
parameters.put("user.username", "username");
Map<String, File> files= new HashMap<String, File>();
File f = Yalp.getFile("test/fond1.png");
assertTrue(f.exists());
files.put("user.avatar", f);
Response uploadResponse = POST(url, parameters, files);
assertStatus(302, uploadResponse);
String id = uploadResponse.getHeader("Location").split("=")[1];
imageURL = reverse(); {
Binary.showAvatar(new Long(id));
}
getResponse = GET(imageURL);
assertStatus(200, getResponse);
}
@Test
public void testUploadBigFile() {
Map<String,String> parameters= new HashMap<String,String>();
Map<String, File> files= new HashMap<String, File>();
File file = Yalp.getFile("test/winie.jpg");
assertTrue(file.exists());
files.put("file", file);
Response uploadResponse = POST("/Binary/uploadFile", parameters, files);
assertStatus(200, uploadResponse);
String size = uploadResponse.getHeader("Content-Length");
assertEquals("Size does not match", "1366949", size);
}
@Test
public void testUploadBigFile2() {
Map<String,String> parameters= new HashMap<String,String>();
Map<String, File> files= new HashMap<String, File>();
File file = Yalp.getFile("test/winie.jpg");
assertTrue(file.exists());
files.put("upload", file);
Response uploadResponse = POST("/Binary/upload", parameters, files);
assertStatus(200, uploadResponse);
String size = uploadResponse.getHeader("Content-Length");
assertEquals("Size does not match", "1366949", size);
}
@Test
public void testUploadSmallFile() {
Map<String,String> parameters= new HashMap<String,String>();
Map<String, File> files= new HashMap<String, File>();
File file = Yalp.getFile("test/angel.gif");
assertTrue(file.exists());
files.put("file", file);
Response uploadResponse = POST("/Binary/uploadFile", parameters, files);
assertStatus(200, uploadResponse);
String size = uploadResponse.getHeader("Content-Length");
assertEquals("Size does not match", "2440", size);
}
@Test
public void testUploadSmallFile2() {
Map<String,String> parameters= new HashMap<String,String>();
Map<String, File> files= new HashMap<String, File>();
File file = Yalp.getFile("test/angel.gif");
assertTrue(file.exists());
files.put("upload", file);
Response uploadResponse = POST("/Binary/upload", parameters, files);
assertStatus(200, uploadResponse);
String size = uploadResponse.getHeader("Content-Length");
assertEquals("Size does not match", "2440", size);
}
// TODO: Missing possibility to upload multiple files at once
// See: http://yalp.lighthouseapp.com/projects/57987-play-framework/tickets/472-functionaltest-and-ws-client-library-dont-allow-upload-of-multiple-file#ticket-472-2
// @Test
// public void testMultipleUpload() {
//
// Map<String,String> parameters= new HashMap<String,String>();
//
// Map<String, File[]> files= new HashMap<String, File[]>();
// File file1 = Yalp.getFile("test/angel.gif");
// assertTrue(file1.exists());
//
// File file2 = Yalp.getFile("test/winie .gif");
// assertTrue(file1.exists());
//
// files.put("upload", new File[] {file1, file2 });
// Response uploadResponse = POST("/Binary/uploadMultiple", parameters, files);
//
// assertStatus(200, uploadResponse);
//
// String size = uploadResponse.getHeader("Content-Length");
//
// assertEquals("Size does not match", "2440", size);
// }
@Test
public void testGetBinaryWithCustomContentType() {
Response response = GET("/binary/getBinaryWithCustomContentType");
assertIsOk(response);
assertContentType("custom/contentType", response);
}
// Tests to check whether input streams to renderBinary are closed.
@Test
public void testGetEmptyBinary() {
Response response = GET("/binary/getEmptyBinary");
assertIsOk(response);
assertTrue(Binary.emptyInputStreamClosed);
}
@Test
public void testGetErrorBinary() {
Response response = GET("/binary/getErrorBinary");
// This does not work. See Lighthouse ticket #1637.
// assertStatus(500, response);
assertTrue(Binary.errorInputStreamClosed);
}
}