/*
 * 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.chemistry.opencmis.commons.impl.misc;

import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.Closeable;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.List;
import java.util.Map;

import org.apache.chemistry.opencmis.commons.data.ContentStream;
import org.apache.chemistry.opencmis.commons.impl.IOUtils;
import org.junit.Test;

public class IOUtilsTest {

    @Test
    public void testUTF8() {
        StringBuilder sb = new StringBuilder();
        for (char c = ' '; c <= 0xfff; c++) {
            sb.append(c);
        }

        assertEquals(sb.toString(), IOUtils.toUTF8String(IOUtils.toUTF8Bytes(sb.toString())));
        assertNull(IOUtils.toUTF8Bytes(null));
        assertNull(IOUtils.toUTF8String(null));
    }

    @Test
    public void testEncoding() {
        String url = "http://example.com/abc?key=äöü @!§$%&/";

        assertEquals(url, IOUtils.decodeURL(IOUtils.encodeURL(url)));
        assertNull(IOUtils.encodeURL(null));
        assertNull(IOUtils.decodeURL(null));
    }

    @Test
    public void testCheckForBytes() throws Exception {
        assertFalse(IOUtils.checkForBytes(new ByteArrayInputStream(new byte[0])));
        assertTrue(IOUtils.checkForBytes(new ByteArrayInputStream(IOUtils.toUTF8Bytes("Hello World!"))));

        assertNull(IOUtils.checkForBytes(new ByteArrayInputStream(new byte[0]), 1024));
        assertTrue(IOUtils.checkForBytes(new ByteArrayInputStream(IOUtils.toUTF8Bytes("Hello World!")), 1024) instanceof ByteArrayInputStream);

        assertNull(IOUtils.checkForBytes(new ByteArrayInputStream(new byte[0]) {
            @Override
            public boolean markSupported() {

                return false;
            }
        }, 1024));
        assertTrue(IOUtils.checkForBytes(new ByteArrayInputStream(IOUtils.toUTF8Bytes("Hello World!")) {
            @Override
            public boolean markSupported() {
                return false;
            }
        }, 1024) instanceof BufferedInputStream);
    }

    @Test
    public void testConsumeAndClose() {
        ByteArrayInputStream stream = new ByteArrayInputStream(IOUtils.toUTF8Bytes("test"));

        IOUtils.consumeAndClose(stream);
        assertEquals(0, stream.available());

        ByteArrayInputStream stream2 = new ByteArrayInputStream(IOUtils.toUTF8Bytes("test"));
        InputStreamReader reader = new InputStreamReader(stream2);

        IOUtils.consumeAndClose(reader);
        assertEquals(0, stream2.available());

        try {
            reader.read();
            fail("Reader should be closed.");
        } catch (IOException e) {
            // expected
        }

        IOUtils.closeQuietly(reader);
        IOUtils.closeQuietly((Closeable) null);
        IOUtils.closeQuietly((ContentStream) null);
    }

    @Test
    public void testCopy() throws Exception {
        byte[] input = IOUtils.toUTF8Bytes("test");
        ByteArrayInputStream in = new ByteArrayInputStream(input);
        ByteArrayOutputStream out = new ByteArrayOutputStream();

        IOUtils.copy(in, out);

        assertArrayEquals(input, out.toByteArray());

        IOUtils.closeQuietly(in);
        IOUtils.closeQuietly(out);
    }

    @Test
    public void testFirstLine() throws IOException {
        String content = "line 1\nline 2\nline 3";
        InputStream stream = createStream(content);

        assertEquals("line 1", IOUtils.readFirstLine(stream));
    }

    @Test
    public void testAllLines() throws IOException {
        String content = "line 1\nline 2\nline 3";
        InputStream stream = createStream(content);

        assertEquals("line 1\nline 2\nline 3\n", IOUtils.readAllLines(stream));
    }

    @Test
    public void testAllLinesAndSkipHeader1() throws IOException {
        String content = "#header\n\nline 1\nline 2";
        InputStream stream = createStream(content);

        assertEquals("line 1\nline 2\n", IOUtils.readAllLinesAndRemoveHeader(stream));
    }

    @Test
    public void testAllLinesAndSkipHeader2() throws IOException {
        String content = "/* header 1\n * header 2\n*/\n\nline 1\nline 2";
        InputStream stream = createStream(content);

        assertEquals("line 1\nline 2\n", IOUtils.readAllLinesAndRemoveHeader(stream));
    }

    @Test
    public void testIgnoreComments() throws IOException {
        String content = "#comment1\n\nline 1\n  # comment 2\nline 2\nline 3";
        InputStream stream = createStream(content);

        assertEquals("line 1\nline 2\nline 3\n", IOUtils.readAllLinesAndIgnoreComments(stream));
    }

    @Test
    public void testAsList() throws IOException {
        String content = "#comment1\n\nline 1\n  # comment 2\nline 2\nline 3";
        InputStream stream = createStream(content);

        List<String> list = IOUtils.readAllLinesAsList(stream);

        assertNotNull(list);
        assertEquals(3, list.size());
        assertEquals("line 1", list.get(0));
        assertEquals("line 2", list.get(1));
        assertEquals("line 3", list.get(2));
    }

    @Test
    public void testAsMap() throws IOException {
        String content = "#comment1\nkey1\n  # comment 2\nkey2=value2\nkey3 = value3";
        InputStream stream = createStream(content);

        Map<String, String> map = IOUtils.readAllLinesAsMap(stream);

        assertNotNull(map);
        assertEquals(3, map.size());
        assertEquals("", map.get("key1"));
        assertEquals("value2", map.get("key2"));
        assertEquals("value3", map.get("key3"));
    }

    private InputStream createStream(String content) {
        return new ByteArrayInputStream(IOUtils.toUTF8Bytes(content));
    }
}