src/main/java/org/apache/sling/launchpad/webapp/integrationtest/UploadFileTest.java - sling-org-apache-sling-launchpad-integration-tests - Git at Google

 /*
  * Licensed to the Apache Software Foundation (ASF) under one or more
  * contributor license agreements.  See the NOTICE file distributed with
  * this work for additional information regarding copyright ownership.
  * The ASF licenses this file to You under the Apache License, Version 2.0
  * (the "License"); you may not use this file except in compliance with
  * the License.  You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
 package org.apache.sling.launchpad.webapp.integrationtest;

 import org.apache.commons.httpclient.HttpException;
 import org.apache.commons.httpclient.methods.GetMethod;
 import org.apache.sling.commons.testing.integration.HttpTestBase;

 import java.io.File;
 import java.io.FileWriter;
 import java.io.IOException;

 /**
  * Test uploading files using the Sling post servlet (SLING-168)
  */
 public class UploadFileTest extends HttpTestBase {


     private static final String UPLOAD_CONTENT = "This file is used by UploadFileTest.java.\n" +
     		"\n" +
     		"It must contain http://www.apache.org/licenses/LICENSE-2.0 for tests.\n" +
     		"\n" +
     		"Lorem ipsum dolor sit amet, consectetuer adipiscing elit. Fusce semper ipsum et lorem. In hac habitasse \n" +
     		"platea dictumst. Donec dictum tincidunt purus. Aenean quis nunc. Aliquam rhoncus. Proin sed risus. \n" +
     		"Maecenas porta arcu in nisi. Pellentesque quis sapien quis lectus vehicula aliquet. Cras ornare elit \n" +
     		"eget massa. Cum sociis natoque penatibus et magnis dis parturient montes, nascetur ridiculus mus. Sed \n" +
     		"ut justo. Integer porttitor quam. Aliquam aliquet. Nulla interdum arcu vitae nibh. Morbi dapibus odio \n" +
     		"a sem. Integer dictum. Praesent vel eros nec ipsum venenatis malesuada. Vestibulum rutrum mi ac ligula. \n" +
     		"Ut nisl ligula, vehicula dignissim, accumsan quis, faucibus sed, sapien. Quisque purus tellus, euismod \n" +
     		"id, auctor quis, rutrum non, augue.\n";

     public void testDistinctResource() throws IOException {
         String folderPath = "/UploadFileTest_1_" + System.currentTimeMillis();
         final String url = HTTP_BASE_URL + folderPath;

         // create new node
         String urlOfNewNode = null;
         try {
             urlOfNewNode = testClient.createNode(url, null);
         } catch(IOException ioe) {
             fail("createNode failed: " + ioe);
         }

         // upload local file
         File localFile = getTestFile();
         testClient.uploadToFileNode(urlOfNewNode, localFile, "./file", null);

         // get and check URL of created file
         String urlOfFileNode = urlOfNewNode + "/file";
         final GetMethod get = new GetMethod(urlOfFileNode);
         final int status = httpClient.executeMethod(get);
         assertEquals(urlOfFileNode + " must be accessible after createNode",200,status);

         /*
         We should check the data, but nt:resources are not handled yet
         // compare data with local file (just length)
         final byte[] data = get.getResponseBody();
         assertEquals("size of file must be same", localFile.length(), data.length);
         */
         String data = get.getResponseBodyAsString();
         assertTrue("checking for content", data.contains("http://www.apache.org/licenses/LICENSE-2.0"));

         // download structure
         String json = getContent(urlOfFileNode + ".json", CONTENT_TYPE_JSON);
         // just check for some strings
         assertTrue("checking primary type", json.contains("\"jcr:primaryType\":\"nt:resource\""));
         assertTrue("checking mime type", json.contains("\"jcr:mimeType\":\"text/plain\""));

     }


     public void testDistinctResourceWithType() throws IOException {
         String folderPath = "/UploadFileTest_1_" + System.currentTimeMillis();
         final String url = HTTP_BASE_URL + folderPath;

         // create new node
         String urlOfNewNode = null;
         try {
             urlOfNewNode = testClient.createNode(url, null);
         } catch(IOException ioe) {
             fail("createNode failed: " + ioe);
         }

         // upload local file
         File localFile = getTestFile();
         testClient.uploadToFileNode(urlOfNewNode, localFile, "./file", "nt:unstructured");

         // get and check URL of created file
         String urlOfFileNode = urlOfNewNode + "/file";
         final GetMethod get = new GetMethod(urlOfFileNode);
         final int status = httpClient.executeMethod(get);
         assertEquals(urlOfFileNode + " must be accessible after createNode",200,status);

         /*
         We should check the data, but nt:resources are not handled yet
         // compare data with local file (just length)
         final byte[] data = get.getResponseBody();
         assertEquals("size of file must be same", localFile.length(), data.length);
         */
         String data = get.getResponseBodyAsString();
         assertTrue("checking for content", data.contains("http://www.apache.org/licenses/LICENSE-2.0"));

         // download structure
         String json = getContent(urlOfFileNode + ".100.json", CONTENT_TYPE_JSON);

         // just check for some strings
         assertTrue("checking primary type", json.contains("\"jcr:primaryType\":\"nt:unstructured\""));
         assertTrue("checking mime type", json.contains("\"jcr:mimeType\":\"text/plain\""));
     }

     public void testDistinctFile() throws IOException {
         String folderPath = "/UploadFileTest_1_" + System.currentTimeMillis();
         testClient.mkdirs(WEBDAV_BASE_URL, folderPath);
         final String url = HTTP_BASE_URL + folderPath;


         // upload local file
         File localFile = getTestFile();
         testClient.uploadToFileNode(url, localFile, "./file", null);

         // get and check URL of created file
         String urlOfFileNode = url + "/file";

         /*
         TODO: does not work, since no nt:file resource type handler present ???

         final GetMethod get = new GetMethod(urlOfFileNode);
         final int status = httpClient.executeMethod(get);
         assertEquals(urlOfFileNode + " must be accessible after createNode",200,status);

         // compare data with local file (just length)
         final byte[] data = get.getResponseBody();
         assertEquals("size of file must be same", localFile.length(), data.length);
         */

         String webdavUrl = WEBDAV_BASE_URL + folderPath + "/file";
         final GetMethod get = new GetMethod(webdavUrl);
         final int status = httpClient.executeMethod(get);
         assertEquals(urlOfFileNode + " must be accessible after createNode",200,status);

         // compare data with local file (just length)
         final byte[] data = get.getResponseBody();
         assertEquals("size of file must be same", localFile.length(), data.length);

         // download structure
         String json = getContent(urlOfFileNode + ".json", CONTENT_TYPE_JSON);
         // just check for some strings
         assertTrue("checking primary type", json.contains("\"jcr:primaryType\":\"nt:file\""));
     }


     public void testMultiFileUpload() throws IOException {
         String folderPath = "/UploadFileTest_1_" + System.currentTimeMillis();
         final String url = HTTP_BASE_URL + folderPath;

         // create new node
         String urlOfNewNode = null;
         try {
             urlOfNewNode = testClient.createNode(url, null);
         } catch(IOException ioe) {
             fail("createNode failed: " + ioe);
         }
         // upload local file
         File tempFile = getTestFile();
         File [] localFiles = new File[] {
         	tempFile,
         	tempFile,
         	tempFile
         };
         String [] fieldNames = new String [] {
         		"./file1.txt",
         		"./file2.txt",
         		"./*"
         };
         testClient.uploadToFileNodes(urlOfNewNode, localFiles, fieldNames, null);

         // get and check URL of created file1
         String urlOfFileNode = urlOfNewNode + "/file1.txt";
         checkUploadedFileState(urlOfFileNode);

         // get and check URL of created file2
         String urlOfFileNode2 = urlOfNewNode + "/file2.txt";
         checkUploadedFileState(urlOfFileNode2);

         // get and check URL of created file3
         String urlOfFileNode3 = urlOfNewNode + "/file-to-upload.txt";
         checkUploadedFileState(urlOfFileNode3);
     }

 	private void checkUploadedFileState(String urlOfFileNode) throws IOException,
 			HttpException {
 		final GetMethod get = new GetMethod(urlOfFileNode);
         final int status = httpClient.executeMethod(get);
         assertEquals(urlOfFileNode + " must be accessible after createNode",200,status);

         /*
         We should check the data, but nt:resources are not handled yet
         // compare data with local file (just length)
         final byte[] data = get.getResponseBody();
         assertEquals("size of file must be same", localFile.length(), data.length);
         */
         String data = get.getResponseBodyAsString();
         assertTrue("checking for content", data.contains("http://www.apache.org/licenses/LICENSE-2.0"));

         // download structure
         String json = getContent(urlOfFileNode + ".json", CONTENT_TYPE_JSON);
         // just check for some strings
         assertTrue("checking primary type", json.contains("\"jcr:primaryType\":\"nt:file\""));

         String content_json = getContent(urlOfFileNode + "/jcr:content.json", CONTENT_TYPE_JSON);
         // just check for some strings
         assertTrue("checking primary type", content_json.contains("\"jcr:primaryType\":\"nt:resource\""));
         assertTrue("checking mime type", content_json.contains("\"jcr:mimeType\":\"text/plain\""));
 	}

   /**
    * @return
    * @throws IOException
    */
   private File getTestFile() throws IOException {
     File tempFile = new File("target/file-to-upload.txt");
     if ( !tempFile.exists() || tempFile.length() == 0 ) {
       FileWriter out = new FileWriter(tempFile);
       out.write(UPLOAD_CONTENT); // we do rather than from a resource since surefire has fun with classloaders.
       out.close();
     }
     return tempFile;
   }

 }
	/*
	* Licensed to the Apache Software Foundation (ASF) under one or more
	* contributor license agreements. See the NOTICE file distributed with
	* this work for additional information regarding copyright ownership.
	* The ASF licenses this file to You under the Apache License, Version 2.0
	* (the "License"); you may not use this file except in compliance with
	* the License. You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/
	package org.apache.sling.launchpad.webapp.integrationtest;

	import org.apache.commons.httpclient.HttpException;
	import org.apache.commons.httpclient.methods.GetMethod;
	import org.apache.sling.commons.testing.integration.HttpTestBase;

	import java.io.File;
	import java.io.FileWriter;
	import java.io.IOException;

	/**
	* Test uploading files using the Sling post servlet (SLING-168)
	*/
	public class UploadFileTest extends HttpTestBase {


	private static final String UPLOAD_CONTENT = "This file is used by UploadFileTest.java.\n" +
	"\n" +
	"It must contain http://www.apache.org/licenses/LICENSE-2.0 for tests.\n" +
	"\n" +
	"Lorem ipsum dolor sit amet, consectetuer adipiscing elit. Fusce semper ipsum et lorem. In hac habitasse \n" +
	"platea dictumst. Donec dictum tincidunt purus. Aenean quis nunc. Aliquam rhoncus. Proin sed risus. \n" +
	"Maecenas porta arcu in nisi. Pellentesque quis sapien quis lectus vehicula aliquet. Cras ornare elit \n" +
	"eget massa. Cum sociis natoque penatibus et magnis dis parturient montes, nascetur ridiculus mus. Sed \n" +
	"ut justo. Integer porttitor quam. Aliquam aliquet. Nulla interdum arcu vitae nibh. Morbi dapibus odio \n" +
	"a sem. Integer dictum. Praesent vel eros nec ipsum venenatis malesuada. Vestibulum rutrum mi ac ligula. \n" +
	"Ut nisl ligula, vehicula dignissim, accumsan quis, faucibus sed, sapien. Quisque purus tellus, euismod \n" +
	"id, auctor quis, rutrum non, augue.\n";

	public void testDistinctResource() throws IOException {
	String folderPath = "/UploadFileTest_1_" + System.currentTimeMillis();
	final String url = HTTP_BASE_URL + folderPath;

	// create new node
	String urlOfNewNode = null;
	try {
	urlOfNewNode = testClient.createNode(url, null);
	} catch(IOException ioe) {
	fail("createNode failed: " + ioe);
	}

	// upload local file
	File localFile = getTestFile();
	testClient.uploadToFileNode(urlOfNewNode, localFile, "./file", null);

	// get and check URL of created file
	String urlOfFileNode = urlOfNewNode + "/file";
	final GetMethod get = new GetMethod(urlOfFileNode);
	final int status = httpClient.executeMethod(get);
	assertEquals(urlOfFileNode + " must be accessible after createNode",200,status);

	/*
	We should check the data, but nt:resources are not handled yet
	// compare data with local file (just length)
	final byte[] data = get.getResponseBody();
	assertEquals("size of file must be same", localFile.length(), data.length);
	*/
	String data = get.getResponseBodyAsString();
	assertTrue("checking for content", data.contains("http://www.apache.org/licenses/LICENSE-2.0"));

	// download structure
	String json = getContent(urlOfFileNode + ".json", CONTENT_TYPE_JSON);
	// just check for some strings
	assertTrue("checking primary type", json.contains("\"jcr:primaryType\":\"nt:resource\""));
	assertTrue("checking mime type", json.contains("\"jcr:mimeType\":\"text/plain\""));

	}


	public void testDistinctResourceWithType() throws IOException {
	String folderPath = "/UploadFileTest_1_" + System.currentTimeMillis();
	final String url = HTTP_BASE_URL + folderPath;

	// create new node
	String urlOfNewNode = null;
	try {
	urlOfNewNode = testClient.createNode(url, null);
	} catch(IOException ioe) {
	fail("createNode failed: " + ioe);
	}

	// upload local file
	File localFile = getTestFile();
	testClient.uploadToFileNode(urlOfNewNode, localFile, "./file", "nt:unstructured");

	// get and check URL of created file
	String urlOfFileNode = urlOfNewNode + "/file";
	final GetMethod get = new GetMethod(urlOfFileNode);
	final int status = httpClient.executeMethod(get);
	assertEquals(urlOfFileNode + " must be accessible after createNode",200,status);

	/*
	We should check the data, but nt:resources are not handled yet
	// compare data with local file (just length)
	final byte[] data = get.getResponseBody();
	assertEquals("size of file must be same", localFile.length(), data.length);
	*/
	String data = get.getResponseBodyAsString();
	assertTrue("checking for content", data.contains("http://www.apache.org/licenses/LICENSE-2.0"));

	// download structure
	String json = getContent(urlOfFileNode + ".100.json", CONTENT_TYPE_JSON);

	// just check for some strings
	assertTrue("checking primary type", json.contains("\"jcr:primaryType\":\"nt:unstructured\""));
	assertTrue("checking mime type", json.contains("\"jcr:mimeType\":\"text/plain\""));
	}

	public void testDistinctFile() throws IOException {
	String folderPath = "/UploadFileTest_1_" + System.currentTimeMillis();
	testClient.mkdirs(WEBDAV_BASE_URL, folderPath);
	final String url = HTTP_BASE_URL + folderPath;


	// upload local file
	File localFile = getTestFile();
	testClient.uploadToFileNode(url, localFile, "./file", null);

	// get and check URL of created file
	String urlOfFileNode = url + "/file";

	/*
	TODO: does not work, since no nt:file resource type handler present ???

	final GetMethod get = new GetMethod(urlOfFileNode);
	final int status = httpClient.executeMethod(get);
	assertEquals(urlOfFileNode + " must be accessible after createNode",200,status);

	// compare data with local file (just length)
	final byte[] data = get.getResponseBody();
	assertEquals("size of file must be same", localFile.length(), data.length);
	*/

	String webdavUrl = WEBDAV_BASE_URL + folderPath + "/file";
	final GetMethod get = new GetMethod(webdavUrl);
	final int status = httpClient.executeMethod(get);
	assertEquals(urlOfFileNode + " must be accessible after createNode",200,status);

	// compare data with local file (just length)
	final byte[] data = get.getResponseBody();
	assertEquals("size of file must be same", localFile.length(), data.length);

	// download structure
	String json = getContent(urlOfFileNode + ".json", CONTENT_TYPE_JSON);
	// just check for some strings
	assertTrue("checking primary type", json.contains("\"jcr:primaryType\":\"nt:file\""));
	}


	public void testMultiFileUpload() throws IOException {
	String folderPath = "/UploadFileTest_1_" + System.currentTimeMillis();
	final String url = HTTP_BASE_URL + folderPath;

	// create new node
	String urlOfNewNode = null;
	try {
	urlOfNewNode = testClient.createNode(url, null);
	} catch(IOException ioe) {
	fail("createNode failed: " + ioe);
	}
	// upload local file
	File tempFile = getTestFile();
	File [] localFiles = new File[] {
	tempFile,
	tempFile,
	tempFile
	};
	String [] fieldNames = new String [] {
	"./file1.txt",
	"./file2.txt",
	"./*"
	};
	testClient.uploadToFileNodes(urlOfNewNode, localFiles, fieldNames, null);

	// get and check URL of created file1
	String urlOfFileNode = urlOfNewNode + "/file1.txt";
	checkUploadedFileState(urlOfFileNode);

	// get and check URL of created file2
	String urlOfFileNode2 = urlOfNewNode + "/file2.txt";
	checkUploadedFileState(urlOfFileNode2);

	// get and check URL of created file3
	String urlOfFileNode3 = urlOfNewNode + "/file-to-upload.txt";
	checkUploadedFileState(urlOfFileNode3);
	}

	private void checkUploadedFileState(String urlOfFileNode) throws IOException,
	HttpException {
	final GetMethod get = new GetMethod(urlOfFileNode);
	final int status = httpClient.executeMethod(get);
	assertEquals(urlOfFileNode + " must be accessible after createNode",200,status);

	/*
	We should check the data, but nt:resources are not handled yet
	// compare data with local file (just length)
	final byte[] data = get.getResponseBody();
	assertEquals("size of file must be same", localFile.length(), data.length);
	*/
	String data = get.getResponseBodyAsString();
	assertTrue("checking for content", data.contains("http://www.apache.org/licenses/LICENSE-2.0"));

	// download structure
	String json = getContent(urlOfFileNode + ".json", CONTENT_TYPE_JSON);
	// just check for some strings
	assertTrue("checking primary type", json.contains("\"jcr:primaryType\":\"nt:file\""));

	String content_json = getContent(urlOfFileNode + "/jcr:content.json", CONTENT_TYPE_JSON);
	// just check for some strings
	assertTrue("checking primary type", content_json.contains("\"jcr:primaryType\":\"nt:resource\""));
	assertTrue("checking mime type", content_json.contains("\"jcr:mimeType\":\"text/plain\""));
	}

	/**
	* @return
	* @throws IOException
	*/
	private File getTestFile() throws IOException {
	File tempFile = new File("target/file-to-upload.txt");
	if ( !tempFile.exists() \|\| tempFile.length() == 0 ) {
	FileWriter out = new FileWriter(tempFile);
	out.write(UPLOAD_CONTENT); // we do rather than from a resource since surefire has fun with classloaders.
	out.close();
	}
	return tempFile;
	}

	}