/*
* Copyright 2004,2005 The Apache Software Foundation.
*
* Licensed 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.axiom.attachments;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.Enumeration;
import javax.mail.Header;
import org.apache.axiom.om.AbstractTestCase;
/**
* Test the PartOnFile class
*/
public class PartOnFileTest extends AbstractTestCase {
public PartOnFileTest(String testName) {
super(testName);
}
String inMimeFileName = "mtom/MTOMAttachmentStream.bin";
String contentTypeString = "multipart/related; boundary=MIMEBoundaryurn:uuid:A3ADBAEE51A1A87B2A11443668160701; type=\"application/xop+xml\"; start=\"<0.urn:uuid:A3ADBAEE51A1A87B2A11443668160702@apache.org>\"; start-info=\"application/soap+xml\"; charset=UTF-8;action=\"mtomSample\"";
File temp;
public void setUp() throws Exception {
createTemporaryDirectory();
}
public void tearDown() throws Exception {
deleteTemporaryDirectory();
}
public void testHeaderGetSet() throws Exception {
InputStream inStream = new FileInputStream(getTestResourceFile(inMimeFileName));
Attachments attachments = new Attachments(inStream, contentTypeString, true, temp.getPath(), "1");
Part p = attachments.getPart("1.urn:uuid:A3ADBAEE51A1A87B2A11443668160943@apache.org");
if (!(p instanceof PartOnFile)) {
fail("Expected PartOnFile");
}
assertEquals("<1.urn:uuid:A3ADBAEE51A1A87B2A11443668160943@apache.org>", p.getContentID());
assertEquals("image/jpeg", p.getContentType());
p.addHeader("Some-New-Header", "TestNH");
assertEquals(p.getHeader("Some-New-Header"), "TestNH");
}
public void testGetAllheaders() throws Exception {
InputStream inStream = new FileInputStream(getTestResourceFile(inMimeFileName));
Attachments attachments = new Attachments(inStream, contentTypeString, true, temp.getPath(), "1");
Part p = attachments.getPart("1.urn:uuid:A3ADBAEE51A1A87B2A11443668160943@apache.org");
if (!(p instanceof PartOnFile)) {
fail("Expected PartOnFile");
}
assertEquals("<1.urn:uuid:A3ADBAEE51A1A87B2A11443668160943@apache.org>", p.getContentID());
// Check if the enumeration works
p.addHeader("Some-New-Header", "TestNH");
Enumeration e = p.getAllHeaders();
boolean cTypeFound = false;
boolean cTransferEncFound = false;
boolean cIdFound = false;
boolean snhFound = false;
while (e.hasMoreElements()) {
Header h = (Header) e.nextElement();
if (h.getName().toLowerCase().equals("content-type")) {
cTypeFound = true;
}
if (h.getName().toLowerCase().equals("content-transfer-encoding")) {
cTransferEncFound = true;
}
if (h.getName().toLowerCase().equals("content-id")) {
cIdFound = true;
}
if (h.getName().toLowerCase().equals("some-new-header")) {
snhFound = true;
}
}
if (!cTypeFound || !cTransferEncFound || !cIdFound || !snhFound) {
fail("Header enumeration failed");
}
}
private void createTemporaryDirectory() throws Exception {
temp = File.createTempFile("partOnFileTest", ".tmp");
if (!temp.delete()) {
fail("Cannot delete from temporary directory. File: " + temp.toURL());
}
if (!temp.mkdir()) {
fail("Cannot create a temporary location for part files");
}
}
private void deleteTemporaryDirectory() throws Exception {
String[] fileList = temp.list();
for (int i=0; i < fileList.length; i++) {
if (!(new File(temp, fileList[i])).delete()) {
System.err.println("WARNING: temporary directory removal failed.");
}
}
if (!temp.delete()) {
System.err.println("WARNING: temporary directory removal failed.");
}
}
}