poi/src/testcases/org/apache/poi/ss/usermodel/TestExcelStyleDateFormatter.java

/* ====================================================================
   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.poi.ss.usermodel;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import java.text.DateFormatSymbols;
import java.text.FieldPosition;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import org.apache.poi.util.LocaleUtil;
import org.junit.Test;

public class TestExcelStyleDateFormatter {
    private static final String EXCEL_DATE_FORMAT = "MMMMM";
    private static final SimpleDateFormat DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd", Locale.ROOT);
    private final int jreVersion;

    public TestExcelStyleDateFormatter() {
        jreVersion = Integer.parseInt(System.getProperty("java.version")
            .replace("1.8", "8").replaceAll("(\\d+).*", "$1"));
    }

    /**
     * [Bug 60369] Month format 'MMMMM' issue with TEXT-formula and Java 8
     */
    @Test
    public void test60369() {
        Map<Locale, String> testMap = initializeLocales();

        // We have to set up dates as well.
        List<Date> testDates = Stream.of("1980-01-12", "1995-02-11", "2045-03-10", "2016-04-09", "2017-05-08",
            "1945-06-07", "1998-07-06", "2099-08-05", "1988-09-04", "2023-10-03", "1978-11-02", "1890-12-01")
            .map(this::parseDate).collect(Collectors.toList());

        // Let's iterate over the test setup.
        final String provider = System.getProperty("java.locale.providers");
        final FieldPosition fp = new FieldPosition(java.text.DateFormat.MONTH_FIELD);
        final ExcelStyleDateFormatter formatter = new ExcelStyleDateFormatter(EXCEL_DATE_FORMAT);
        final StringBuffer sb = new StringBuffer();

        for (Map.Entry<Locale,String> me : testMap.entrySet()) {
            final Locale locale = me.getKey();
            final String expected = me.getValue();
            formatter.setDateFormatSymbols(DateFormatSymbols.getInstance(locale));
            int month = 0;
            for (Date d : testDates) {
                sb.setLength(0);
                String result = formatter.format(d, sb, fp).toString();
                String msg = "Failed testDates for locale " + locale + ", provider: " + provider +
                        " and date " + d + ", having: " + result;

                int actIdx = (Locale.CHINESE.equals(locale) && jreVersion >= 12) ? 1 : 0;

                assertNotNull(msg, result);
                assertTrue(msg, result.length() > actIdx);
                assertEquals(msg, expected.charAt(month), result.charAt(actIdx));
                month++;
            }
        }
    }

    private Date parseDate(String dateStr) {
        try {
            return DATE_FORMAT.parse(dateStr);
        } catch (ParseException e) {
            return new Date(0);
        }
    }

    /**
     * Setting up the locale to be tested together with a list of asserted
     * unicode-formatted results and put them in a map.
     */
    private Map<Locale, String> initializeLocales() {
        Map<Locale, String> testMap = new HashMap<>();

        testMap.put(Locale.GERMAN, "JFMAMJJASOND");
        testMap.put(new Locale("de", "AT"), "JFMAMJJASOND");
        testMap.put(Locale.UK, "JFMAMJJASOND");
        testMap.put(new Locale("en", "IN"), "JFMAMJJASOND");
        testMap.put(new Locale("in", "ID"), "JFMAMJJASOND");
        testMap.put(Locale.FRENCH, "jfmamjjasond");

        testMap.put(new Locale("ru", "RU"),
            "\u044f\u0444\u043c\u0430\u043c\u0438\u0438\u0430\u0441\u043e\u043d\u0434");

        testMap.put(Locale.CHINESE, jreVersion < 12
            ? "\u4e00\u4e8c\u4e09\u56db\u4e94\u516d\u4e03\u516b\u4e5d\u5341\u5341\u5341"
            : "123456789111");

        testMap.put(new Locale("tr", "TR"),
            "\u004f\u015e\u004d\u004e\u004d\u0048\u0054\u0041\u0045\u0045\u004b\u0041");

        testMap.put(new Locale("hu", "HU"),
            "\u006a\u0066\u006d\u00e1\u006d\u006a\u006a\u0061\u0073\u006f\u006e\u0064");

        return testMap;
    }

    @Test
    public void testConstruct() {
        new ExcelStyleDateFormatter(EXCEL_DATE_FORMAT, LocaleUtil.getUserLocale());
        new ExcelStyleDateFormatter(EXCEL_DATE_FORMAT);
    }

    @Test
    public void testWithLocale() throws ParseException {
        Locale before = LocaleUtil.getUserLocale();
        try {
            LocaleUtil.setUserLocale(Locale.GERMAN);
            String dateStr = new ExcelStyleDateFormatter(EXCEL_DATE_FORMAT).format(
                    DATE_FORMAT.parse("2016-03-26"));
            assertEquals("M", dateStr);
        } finally {
            LocaleUtil.setUserLocale(before);
        }
    }

    @Test
    public void testWithPattern() throws ParseException {
        String dateStr = new ExcelStyleDateFormatter("yyyy|" + EXCEL_DATE_FORMAT + "|").format(
                DATE_FORMAT.parse("2016-03-26"));
        assertEquals("2016|M|", dateStr);
    }
}