/**************************************************************** | |
* 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.stream; | |
import java.io.ByteArrayInputStream; | |
import java.io.IOException; | |
import org.apache.james.mime4j.stream.RootInputStream; | |
import junit.framework.TestCase; | |
public class RootInputStreamTest extends TestCase { | |
public void testTruncate() throws IOException { | |
String s = "Yada yada yada"; | |
RootInputStream is = | |
new RootInputStream(new ByteArrayInputStream(s.getBytes())); | |
assertEquals(s.charAt(0), (char) is.read()); | |
assertEquals(s.charAt(1), (char) is.read()); | |
assertEquals(s.charAt(2), (char) is.read()); | |
assertEquals(s.charAt(3), (char) is.read()); | |
is.truncate(); | |
assertEquals(-1, is.read()); | |
byte[] buf = new byte[100]; | |
assertEquals(-1, is.read(buf)); | |
assertEquals(-1, is.read(buf, 2, 5)); | |
} | |
/** | |
* Tests that reading single bytes updates the line number appropriately. | |
*/ | |
public void testReadSingleByte() throws IOException { | |
String s = "Yada\r\nyada\r\nyada\r\n"; | |
RootInputStream is = | |
new RootInputStream(new ByteArrayInputStream(s.getBytes())); | |
for (int i = 0; i < 6; i++) { | |
assertEquals(1, is.getLineNumber()); | |
is.read(); | |
} | |
for (int i = 6; i < 12; i++) { | |
assertEquals(2, is.getLineNumber()); | |
is.read(); | |
} | |
for (int i = 12; i < 18; i++) { | |
assertEquals(3, is.getLineNumber()); | |
is.read(); | |
} | |
assertEquals(4, is.getLineNumber()); | |
assertEquals(-1, is.read()); | |
} | |
/** | |
* Tests that reading multiple bytes at once | |
* updates the line number appropriately. | |
*/ | |
public void testReadManyBytes() throws IOException { | |
String s = "Yada\r\nyada\r\nyada\r\n"; | |
RootInputStream is = | |
new RootInputStream(new ByteArrayInputStream(s.getBytes())); | |
byte[] buf = new byte[4]; | |
assertEquals(1, is.getLineNumber()); | |
is.read(buf); | |
assertEquals(1, is.getLineNumber()); | |
is.read(buf); | |
assertEquals(2, is.getLineNumber()); | |
is.read(buf); | |
assertEquals(3, is.getLineNumber()); | |
is.read(buf); | |
assertEquals(3, is.getLineNumber()); | |
is.read(buf); | |
assertEquals(4, is.getLineNumber()); | |
assertEquals(-1, is.read()); | |
} | |
} |