11d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/* 21d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Copyright (C) 2007 The Guava Authors 31d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 41d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Licensed under the Apache License, Version 2.0 (the "License"); 51d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * you may not use this file except in compliance with the License. 61d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * You may obtain a copy of the License at 71d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 81d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * http://www.apache.org/licenses/LICENSE-2.0 91d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 101d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Unless required by applicable law or agreed to in writing, software 111d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * distributed under the License is distributed on an "AS IS" BASIS, 121d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 131d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * See the License for the specific language governing permissions and 141d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * limitations under the License. 151d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 161d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 171d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpackage com.google.common.io; 181d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 191d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.io.ByteArrayInputStream; 201d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.io.IOException; 211d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.io.InputStream; 221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 231d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/** 241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @author Charles Fry 251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpublic class LimitInputStreamTest extends IoTestCase { 271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 281d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testLimit() throws Exception { 291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert byte[] big = newPreFilledByteArray(5); 301d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert InputStream bin = new ByteArrayInputStream(big); 311d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert InputStream lin = new LimitInputStream(bin, 2); 321d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 331d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // also test available 341d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.mark(2); 351d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(2, lin.available()); 361d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert int read = lin.read(); 371d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(big[0], read); 381d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(1, lin.available()); 391d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert read = lin.read(); 401d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(big[1], read); 411d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(0, lin.available()); 421d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert read = lin.read(); 431d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(-1, read); 441d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 451d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.reset(); 461d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert byte[] small = new byte[5]; 471d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert read = lin.read(small); 481d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(2, read); 491d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(big[0], small[0]); 501d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(big[1], small[1]); 511d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 521d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.reset(); 531d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert read = lin.read(small, 2, 3); 541d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(2, read); 551d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(big[0], small[2]); 561d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(big[1], small[3]); 571d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 581d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 591d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testMark() throws Exception { 601d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert byte[] big = newPreFilledByteArray(5); 611d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert InputStream bin = new ByteArrayInputStream(big); 621d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert InputStream lin = new LimitInputStream(bin, 2); 631d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 641d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert int read = lin.read(); 651d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(big[0], read); 661d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.mark(2); 671d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 681d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert read = lin.read(); 691d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(big[1], read); 701d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert read = lin.read(); 711d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(-1, read); 721d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 731d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.reset(); 741d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert read = lin.read(); 751d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(big[1], read); 761d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert read = lin.read(); 771d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(-1, read); 781d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 791d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 801d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testSkip() throws Exception { 811d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert byte[] big = newPreFilledByteArray(5); 821d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert InputStream bin = new ByteArrayInputStream(big); 831d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert InputStream lin = new LimitInputStream(bin, 2); 841d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 851d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // also test available 861d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.mark(2); 871d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(2, lin.available()); 881d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.skip(1); 891d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(1, lin.available()); 901d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 911d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.reset(); 921d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(2, lin.available()); 931d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.skip(3); 941d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(0, lin.available()); 951d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 961d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 971d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 981d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testMarkNotSet() { 991d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert byte[] big = newPreFilledByteArray(5); 1001d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert InputStream bin = new ByteArrayInputStream(big); 1011d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert InputStream lin = new LimitInputStream(bin, 2); 1021d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 1031d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert try { 1041d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.reset(); 1051d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert fail(); 1061d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } catch (IOException expected) { 1071d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals("Mark not set", expected.getMessage()); 1081d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 1091d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 1101d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 1111d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testMarkNotSupported() { 1121d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert InputStream lin = new LimitInputStream(new UnmarkableInputStream(), 2); 1131d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 1141d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert try { 1151d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert lin.reset(); 1161d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert fail(); 1171d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } catch (IOException expected) { 1181d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals("Mark not supported", expected.getMessage()); 1191d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 1201d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 1211d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 1221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private static class UnmarkableInputStream extends InputStream { 1231d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 1241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public int read() throws IOException { 1251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return 0; 1261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 1271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 1281d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 1291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public boolean markSupported() { 1301d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return false; 1311d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 1321d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 1331d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert} 134