test/jdk/java/io/FileWriter/ConstructorTest.java

/*
 * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Reader;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import org.testng.Assert;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

/**
 * @test
 * @bug 8183554
 * @summary Test to verify the new Constructors that take a Charset.
 * @run testng ConstructorTest
 */
public class ConstructorTest {
    static String USER_DIR = System.getProperty("user.dir", ".");

    public static enum ConstructorType {
        STRING,
        FILE,
        STRING_APPEND,
        FILE_APPEND
    }

    static final String TEST_STRING = "abc \u0100 \u0101 \u0555 \u07FD \u07FF";
    static final int BUFFER_SIZE = 8192;

    @DataProvider(name = "parameters")
    public Object[][] getParameters() throws IOException {
        File file1 = new File(USER_DIR, "FileWriterTest1.txt");
        File file2 = new File(USER_DIR, "FileWriterTest2.txt");

        return new Object[][]{
            {ConstructorType.STRING, file1, file2, StandardCharsets.UTF_8},
            {ConstructorType.FILE, file1, file2, StandardCharsets.UTF_8},
            {ConstructorType.STRING_APPEND, file1, file2, StandardCharsets.UTF_8},
            {ConstructorType.FILE_APPEND, file1, file2, StandardCharsets.UTF_8},
            {ConstructorType.STRING, file1, file2, StandardCharsets.ISO_8859_1},
            {ConstructorType.FILE, file1, file2, StandardCharsets.ISO_8859_1},
            {ConstructorType.STRING_APPEND, file1, file2, StandardCharsets.ISO_8859_1},
            {ConstructorType.FILE_APPEND, file1, file2, StandardCharsets.ISO_8859_1},
        };
    }

    /**
     * Verifies that the new constructors that take a Charset function the same
     * as an OutputStreamWriter on a FileOutputStream as was recommended before
     * this change.
     *
     * @param type the type of the constructor
     * @param file1 file1 to be written with a FileWriter
     * @param file2 file2 to be written  with an OutputStreamWriter
     * @param charset the charset
     * @throws IOException
     */
    @Test(dataProvider = "parameters")
    void test(ConstructorType type, File file1, File file2, Charset charset)
            throws Exception {
        writeWithFileWriter(type, file1, TEST_STRING, charset);
        writeWithOutputStreamWriter(type, file2, TEST_STRING, charset);

        try (
                FileReader r1 = getFileReader(type, file1, charset);
                FileReader r2 = getFileReader(type, file2, charset);
            ) {
            String result1 = readAll(r1, BUFFER_SIZE);
            String result2 = readAll(r2, BUFFER_SIZE);
            Assert.assertEquals(result1, result2);
        }
    }

    public String readAll(Reader reader, int bufferSize) throws IOException {
        StringBuilder sb = new StringBuilder();
        char[] buf = new char[bufferSize];
        int numRead;
        while ((numRead = reader.read(buf)) != -1) {
            if (numRead == buf.length) {
                sb.append(buf);
            } else {
                sb.append(String.valueOf(buf, 0, numRead));
            }
        }
        return sb.toString();
    }

    /*
     * Creates a FileReader over the given input file.
     */
    FileReader getFileReader(ConstructorType type, File file, Charset charset)
            throws IOException {
        switch (type) {
            case STRING:
            case STRING_APPEND:
                return new FileReader(file.getPath(), charset);
            case FILE:
            case FILE_APPEND:
                return new FileReader(file, charset);
        }

        return null;
    }

    /*
     * Creates a FileWriter using the constructor as specified.
     */
    FileWriter getFileWriter(ConstructorType type, File file, Charset charset)
            throws IOException {
        switch (type) {
            case STRING:
                return new FileWriter(file.getPath(), charset);
            case FILE:
                return new FileWriter(file, charset);
            case STRING_APPEND:
                return new FileWriter(file.getPath(), charset, true);
            case FILE_APPEND:
                return new FileWriter(file, charset, true);
        }

        return null;
    }

    void writeWithFileWriter(ConstructorType type, File file, String content, Charset charset)
            throws IOException {
        if (type == ConstructorType.STRING_APPEND || type == ConstructorType.FILE_APPEND) {
            try (FileWriter writer = getFileWriter(ConstructorType.FILE, file, charset);) {
                writer.write(content);
            }
        }
        try (FileWriter writer = getFileWriter(type, file, charset);) {
            writer.write(content);
        }
    }

    void writeWithOutputStreamWriter(ConstructorType type, File file, String content, Charset charset)
            throws IOException {
        try (OutputStreamWriter writer = new OutputStreamWriter(new FileOutputStream(file), charset)) {
            writer.write(content);
            if (type == ConstructorType.STRING_APPEND || type == ConstructorType.FILE_APPEND) {
                writer.write(content);
            }
        }
    }
}