/****************************************************************
* 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;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import org.apache.james.mime4j.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());
}
}