UploadFileTest.java

/*
 * 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;
  }

}