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.ByteArrayOutputStream; 201d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 211d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/** 221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Unit tests for {@link CountingOutputStream}. 231d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @author Chris Nokleberg 251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpublic class CountingOutputStreamTest extends IoTestCase { 271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 281d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testCount() throws Exception { 291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert int written = 0; 301d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ByteArrayOutputStream out = new ByteArrayOutputStream(); 311d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert CountingOutputStream counter = new CountingOutputStream(out); 321d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, out.size()); 331d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, counter.getCount()); 341d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 351d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert counter.write(0); 361d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert written += 1; 371d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, out.size()); 381d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, counter.getCount()); 391d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 401d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert byte[] data = new byte[10]; 411d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert counter.write(data); 421d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert written += 10; 431d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, out.size()); 441d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, counter.getCount()); 451d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 461d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert counter.write(data, 0, 5); 471d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert written += 5; 481d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, out.size()); 491d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, counter.getCount()); 501d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 511d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert counter.write(data, 2, 5); 521d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert written += 5; 531d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, out.size()); 541d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, counter.getCount()); 551d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 561d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // Test that illegal arguments do not affect count 571d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert try { 581d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert counter.write(data, 0, data.length + 1); 591d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert fail("expected exception"); 601d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } catch (IndexOutOfBoundsException expected) { 611d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 621d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, out.size()); 631d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(written, counter.getCount()); 641d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 651d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert} 66