1/*
2 *  Licensed to the Apache Software Foundation (ASF) under one or more
3 *  contributor license agreements.  See the NOTICE file distributed with
4 *  this work for additional information regarding copyright ownership.
5 *  The ASF licenses this file to You under the Apache License, Version 2.0
6 *  (the "License"); you may not use this file except in compliance with
7 *  the License.  You may obtain a copy of the License at
8 *
9 *     http://www.apache.org/licenses/LICENSE-2.0
10 *
11 *  Unless required by applicable law or agreed to in writing, software
12 *  distributed under the License is distributed on an "AS IS" BASIS,
13 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 *  See the License for the specific language governing permissions and
15 *  limitations under the License.
16 */
17
18package org.apache.harmony.tests.java.io;
19
20import java.io.ByteArrayInputStream;
21import java.io.IOException;
22import java.io.InputStreamReader;
23import java.io.LineNumberReader;
24import java.io.Reader;
25import java.io.StringReader;
26
27import junit.framework.TestCase;
28
29public class LineNumberReaderTest extends TestCase {
30
31    String text = "0\n1\n2\n3\n4\n5\n6\n7\n8\n9\n10\n11\n12\n13\n14\n15\n16\n17\n18\n19\n20\n21\n22\n23\n24\n25\n26\n27\n28\n29\n30\n31\n32\n33\n34\n35\n36\n37\n38\n39\n40\n41\n42\n43\n44\n45\n46\n47\n48\n49\n50\n51\n52\n53\n54\n55\n56\n57\n58\n59\n60\n61\n62\n63\n64\n65\n66\n67\n68\n69\n70\n71\n72\n73\n74\n75\n76\n77\n78\n79\n80\n81\n82\n83\n84\n85\n86\n87\n88\n89\n90\n91\n92\n93\n94\n95\n96\n97\n98\n99\n100\n101\n102\n103\n104\n105\n106\n107\n108\n109\n110\n111\n112\n113\n114\n115\n116\n117\n118\n119\n120\n121\n122\n123\n124\n125\n126\n127\n128\n129\n130\n131\n132\n133\n134\n135\n136\n137\n138\n139\n140\n141\n142\n143\n144\n145\n146\n147\n148\n149\n150\n151\n152\n153\n154\n155\n156\n157\n158\n159\n160\n161\n162\n163\n164\n165\n166\n167\n168\n169\n170\n171\n172\n173\n174\n175\n176\n177\n178\n179\n180\n181\n182\n183\n184\n185\n186\n187\n188\n189\n190\n191\n192\n193\n194\n195\n196\n197\n198\n199\n200\n201\n202\n203\n204\n205\n206\n207\n208\n209\n210\n211\n212\n213\n214\n215\n216\n217\n218\n219\n220\n221\n222\n223\n224\n225\n226\n227\n228\n229\n230\n231\n232\n233\n234\n235\n236\n237\n238\n239\n240\n241\n242\n243\n244\n245\n246\n247\n248\n249\n250\n251\n252\n253\n254\n255\n256\n257\n258\n259\n260\n261\n262\n263\n264\n265\n266\n267\n268\n269\n270\n271\n272\n273\n274\n275\n276\n277\n278\n279\n280\n281\n282\n283\n284\n285\n286\n287\n288\n289\n290\n291\n292\n293\n294\n295\n296\n297\n298\n299\n300\n301\n302\n303\n304\n305\n306\n307\n308\n309\n310\n311\n312\n313\n314\n315\n316\n317\n318\n319\n320\n321\n322\n323\n324\n325\n326\n327\n328\n329\n330\n331\n332\n333\n334\n335\n336\n337\n338\n339\n340\n341\n342\n343\n344\n345\n346\n347\n348\n349\n350\n351\n352\n353\n354\n355\n356\n357\n358\n359\n360\n361\n362\n363\n364\n365\n366\n367\n368\n369\n370\n371\n372\n373\n374\n375\n376\n377\n378\n379\n380\n381\n382\n383\n384\n385\n386\n387\n388\n389\n390\n391\n392\n393\n394\n395\n396\n397\n398\n399\n400\n401\n402\n403\n404\n405\n406\n407\n408\n409\n410\n411\n412\n413\n414\n415\n416\n417\n418\n419\n420\n421\n422\n423\n424\n425\n426\n427\n428\n429\n430\n431\n432\n433\n434\n435\n436\n437\n438\n439\n440\n441\n442\n443\n444\n445\n446\n447\n448\n449\n450\n451\n452\n453\n454\n455\n456\n457\n458\n459\n460\n461\n462\n463\n464\n465\n466\n467\n468\n469\n470\n471\n472\n473\n474\n475\n476\n477\n478\n479\n480\n481\n482\n483\n484\n485\n486\n487\n488\n489\n490\n491\n492\n493\n494\n495\n496\n497\n498\n499\n500\n";
32
33    LineNumberReader lnr;
34
35    /**
36     * java.io.LineNumberReader#LineNumberReader(java.io.Reader)
37     */
38    public void test_ConstructorLjava_io_Reader() {
39        lnr = new LineNumberReader(new StringReader(text), 4092);
40        assertEquals("Failed to create reader", 0, lnr.getLineNumber());
41    }
42
43    /**
44     * java.io.LineNumberReader#LineNumberReader(java.io.Reader, int)
45     */
46    public void test_ConstructorLjava_io_ReaderI() {
47        lnr = new LineNumberReader(new StringReader(text));
48        assertEquals("Failed to create reader", 0, lnr.getLineNumber());
49    }
50
51    /**
52     * java.io.LineNumberReader#getLineNumber()
53     */
54    public void test_getLineNumber() throws IOException {
55        lnr = new LineNumberReader(new StringReader(text));
56        assertEquals("Returned incorrect line number--expected 0, got ", 0, lnr
57                .getLineNumber());
58
59        lnr.readLine();
60        lnr.readLine();
61
62        assertTrue("Returned incorrect line number--expected 2, got: "
63                + lnr.getLineNumber(), lnr.getLineNumber() == 2);
64    }
65
66    /**
67     * java.io.LineNumberReader#mark(int)
68     */
69    public void test_markI() throws IOException {
70        lnr = new LineNumberReader(new StringReader(text));
71        String line;
72        lnr.skip(80);
73        lnr.mark(100);
74        line = lnr.readLine();
75        lnr.reset();
76        assertTrue("Failed to return to marked position", line.equals(lnr
77                .readLine()));
78        // The spec does not say the mark has to be invalidated
79    }
80
81    /**
82     * java.io.LineNumberReader#read()
83     */
84    public void test_read() throws IOException {
85        lnr = new LineNumberReader(new StringReader(text));
86
87        int c = lnr.read();
88        assertEquals("Read returned incorrect character", '0', c);
89
90        lnr.read();
91        assertEquals("Read failed to inc lineNumber", 1, lnr.getLineNumber());
92    }
93
94    /**
95     * java.io.LineNumberReader#read(char[], int, int)
96     */
97    public void test_read$CII() throws IOException {
98        lnr = new LineNumberReader(new StringReader(text));
99        char[] c = new char[100];
100        lnr.read(c, 0, 4);
101        assertTrue("Read returned incorrect characters", "0\n1\n"
102                .equals(new String(c, 0, 4)));
103        assertEquals("Read failed to inc lineNumber", 2, lnr.getLineNumber());
104    }
105
106    /**
107     * java.io.LineNumberReader#readLine()
108     */
109    public void test_readLine() throws IOException {
110        lnr = new LineNumberReader(new StringReader(text));
111        assertEquals("Returned incorrect line number", 0, lnr.getLineNumber());
112        String line = null;
113        lnr.readLine();
114        line = lnr.readLine();
115
116        assertEquals("Returned incorrect string", "1", line);
117        assertTrue("Returned incorrect line number :" + lnr.getLineNumber(),
118                lnr.getLineNumber() == 2);
119
120        // Regression for HARMONY-4294
121        byte[] buffer = new byte[] { '\r', '\n' };
122        LineNumberReader reader = new LineNumberReader(new InputStreamReader(
123                new ByteArrayInputStream(buffer), "UTF-8"));
124        assertEquals('\n', reader.read());
125        assertEquals(-1, reader.read());
126        reader = new LineNumberReader(new InputStreamReader(
127                new ByteArrayInputStream(buffer), "UTF-8"));
128        assertNotNull(reader.readLine());
129        assertNull(reader.readLine());
130    }
131
132    /**
133     * java.io.LineNumberReader#reset()
134     */
135    public void test_reset() throws IOException {
136        lnr = new LineNumberReader(new StringReader(text));
137        assertEquals("Returned incorrect line number", 0, lnr.getLineNumber());
138        String line = null;
139        lnr.mark(100);
140        lnr.readLine();
141        lnr.reset();
142        line = lnr.readLine();
143
144        assertEquals("Failed to reset reader", "0", line);
145    }
146
147    public void testReadLineSourceThrows() throws IOException {
148        lnr = new LineNumberReader(new Reader() {
149            private StringReader delegate = new StringReader("hello\nworld");
150            private int calls = 0;
151
152            @Override
153            public void close() throws IOException {
154            }
155
156            @Override
157            public int read(char[] buf, int offset, int len) throws IOException {
158                if (calls++ < 2) {
159                    throw new IOException();
160                } else {
161                    return delegate.read(buf, offset, len);
162                }
163            }
164        });
165
166        assertEquals(0, lnr.getLineNumber());
167        try {
168            lnr.readLine();
169            fail();
170        } catch (IOException expected) {
171        }
172
173        assertEquals(0, lnr.getLineNumber());
174        try {
175            lnr.readLine();
176            fail();
177        } catch (IOException expected) {
178        }
179
180        assertEquals(0, lnr.getLineNumber());
181        assertEquals("hello", lnr.readLine());
182        assertEquals(1, lnr.getLineNumber());
183        assertEquals("world", lnr.readLine());
184        assertEquals(2, lnr.getLineNumber());
185    }
186
187    public void testGetLineNumberAfterEnd() throws IOException {
188        lnr = new LineNumberReader(new StringReader("hello\nworld"));
189        assertEquals(0, lnr.getLineNumber());
190        assertEquals("hello", lnr.readLine());
191        assertEquals(1, lnr.getLineNumber());
192        assertEquals("world", lnr.readLine());
193        assertEquals(2, lnr.getLineNumber());
194        assertEquals(null, lnr.readLine());
195        assertEquals(2, lnr.getLineNumber());
196        assertEquals(null, lnr.readLine());
197        assertEquals(2, lnr.getLineNumber());
198    }
199
200    /**
201     * java.io.LineNumberReader#setLineNumber(int)
202     */
203    public void test_setLineNumberI() {
204        lnr = new LineNumberReader(new StringReader(text));
205        lnr.setLineNumber(1001);
206        assertEquals("set incorrect line number", 1001, lnr.getLineNumber());
207    }
208
209    /**
210     * java.io.LineNumberReader#skip(long)
211     */
212    public void test_skipJ() throws IOException {
213        lnr = new LineNumberReader(new StringReader(text));
214        char[] c = new char[100];
215        lnr.skip(80);
216        lnr.read(c, 0, 100);
217
218        assertTrue("Failed to skip to correct position", text
219                .substring(80, 180).equals(new String(c, 0, c.length)));
220    }
221}
222