/**************************************************************** | |
* 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.james.mime4j.io; | |
import java.io.ByteArrayInputStream; | |
import java.io.IOException; | |
import java.io.UnsupportedEncodingException; | |
import org.apache.james.mime4j.io.EOLConvertingInputStream; | |
import junit.framework.TestCase; | |
public class EOLConvertingInputStreamTest extends TestCase { | |
public void testRead() throws IOException { | |
testConvertBoth("Line 1\r\nLine 2\r\n", "Line 1\r\nLine 2\r\n"); | |
testConvertCR("Line 1\r\nLine 2\r\n", "Line 1\r\nLine 2\r\n"); | |
testConvertLF("Line 1\r\nLine 2\r\n", "Line 1\r\nLine 2\r\n"); | |
testConvertBoth("Line 1\n\rLine 2\n\r", "Line 1\r\n\r\nLine 2\r\n\r\n"); | |
testConvertCR("Line 1\n\rLine 2\n\r", "Line 1\n\r\nLine 2\n\r\n"); | |
testConvertLF("Line 1\n\rLine 2\n\r", "Line 1\r\n\rLine 2\r\n\r"); | |
testConvertBoth("Line 1\nLine 2\n", "Line 1\r\nLine 2\r\n"); | |
testConvertCR("Line 1\nLine 2\n", "Line 1\nLine 2\n"); | |
testConvertLF("Line 1\nLine 2\n", "Line 1\r\nLine 2\r\n"); | |
testConvertBoth("Line 1\rLine 2\r", "Line 1\r\nLine 2\r\n"); | |
testConvertCR("Line 1\rLine 2\r", "Line 1\r\nLine 2\r\n"); | |
testConvertLF("Line 1\rLine 2\r", "Line 1\rLine 2\r"); | |
testConvertBoth("\r\n", "\r\n"); | |
testConvertCR("\r\n", "\r\n"); | |
testConvertLF("\r\n", "\r\n"); | |
testConvertBoth("\n", "\r\n"); | |
testConvertCR("\n", "\n"); | |
testConvertLF("\n", "\r\n"); | |
testConvertBoth("\r", "\r\n"); | |
testConvertCR("\r", "\r\n"); | |
testConvertLF("\r", "\r"); | |
testConvertBoth("", ""); | |
testConvertCR("", ""); | |
testConvertLF("", ""); | |
} | |
private void testConvertBoth(String s1, String s2) throws IOException { | |
byte[] bytes = new byte[1024]; | |
ByteArrayInputStream bais = new ByteArrayInputStream(fromString(s1)); | |
EOLConvertingInputStream in = | |
new EOLConvertingInputStream(bais, | |
EOLConvertingInputStream.CONVERT_BOTH); | |
int n = in.read(bytes); | |
assertEquals(s2, toString(bytes, n)); | |
} | |
private void testConvertCR(String s1, String s2) throws IOException { | |
byte[] bytes = new byte[1024]; | |
ByteArrayInputStream bais = new ByteArrayInputStream(fromString(s1)); | |
EOLConvertingInputStream in = | |
new EOLConvertingInputStream(bais, | |
EOLConvertingInputStream.CONVERT_CR); | |
int n = in.read(bytes); | |
assertEquals(s2, toString(bytes, n)); | |
} | |
private void testConvertLF(String s1, String s2) throws IOException { | |
byte[] bytes = new byte[1024]; | |
ByteArrayInputStream bais = new ByteArrayInputStream(fromString(s1)); | |
EOLConvertingInputStream in = | |
new EOLConvertingInputStream(bais, | |
EOLConvertingInputStream.CONVERT_LF); | |
int n = in.read(bytes); | |
assertEquals(s2, toString(bytes, n)); | |
} | |
private String toString(byte[] b, int len) { | |
try { | |
if (len == -1) { | |
return ""; | |
} | |
return new String(b, 0, len, "US-ASCII"); | |
} catch (UnsupportedEncodingException e) { | |
fail(e.getMessage()); | |
return null; | |
} | |
} | |
private byte[] fromString(String s) { | |
try { | |
return s.getBytes("US-ASCII"); | |
} catch (UnsupportedEncodingException e) { | |
fail(e.getMessage()); | |
return null; | |
} | |
} | |
} |