1 ///////////////////////////////////////////////////////////////////////////////////////////////
2 // checkstyle: Checks Java source code and other text files for adherence to a set of rules.
3 // Copyright (C) 2001-2024 the original author or authors.
4 //
5 // This library is free software; you can redistribute it and/or
6 // modify it under the terms of the GNU Lesser General Public
7 // License as published by the Free Software Foundation; either
8 // version 2.1 of the License, or (at your option) any later version.
9 //
10 // This library is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 // Lesser General Public License for more details.
14 //
15 // You should have received a copy of the GNU Lesser General Public
16 // License along with this library; if not, write to the Free Software
17 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 ///////////////////////////////////////////////////////////////////////////////////////////////
19
20 package com.puppycrawl.tools.checkstyle.internal.utils;
21
22 import java.io.ByteArrayOutputStream;
23 import java.io.IOException;
24
25 /**
26 * CloseAndFlushTestByteArrayOutputStream.
27 *
28 * @noinspection ClassOnlyUsedInOnePackage
29 * @noinspectionreason ClassOnlyUsedInOnePackage - class is internal tool, and only used in testing
30 */
31 public final class CloseAndFlushTestByteArrayOutputStream extends ByteArrayOutputStream {
32
33 private int closeCount;
34 private int flushCount;
35
36 @Override
37 public void flush() throws IOException {
38 super.flush();
39 flushCount++;
40 }
41
42 @Override
43 public void close() throws IOException {
44 super.close();
45 closeCount++;
46 }
47
48 public int getCloseCount() {
49 return closeCount;
50 }
51
52 public int getFlushCount() {
53 return flushCount;
54 }
55
56 }