/*

 * Copyright (c) 2012, 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.

 */

/*

 * @test

 * @bug 8005263

 * @run testng LoggerSupplierAPIsTest

 */

import java.util.logging.Logger;

import java.util.logging.Level;

import java.util.logging.Handler;

import java.util.logging.LogRecord;

import java.util.function.Supplier;

import java.util.concurrent.atomic.AtomicInteger;

import java.util.ArrayList;

import java.util.List;

import java.util.Collections;

import org.testng.annotations.Test;

import static org.testng.Assert.assertEquals;

import static org.testng.Assert.assertTrue;

@Test(groups="unit")

public class LoggerSupplierAPIsTest {

    static class CountingSupplier implements Supplier<String> {

        AtomicInteger sno = new AtomicInteger();

        @Override

        public String get() {

            return "Log message " + sno.getAndIncrement();

        }

        public int getCount() { return sno.get(); }

        public void reset() { sno.set(0); }

    }

    static class CountingHandler extends Handler {

        AtomicInteger count = new AtomicInteger();

        ArrayList<LogRecord> ar = new ArrayList<>();

        @Override

        public void close() { reset(); }

        @Override

        public void flush() { reset(); }

        @Override

        public void publish(LogRecord lr) {

            // Can do some more assertion here?

            // System.out.println(lr.getMessage());

            count.incrementAndGet();

        }

        public int getCount() {

            return count.get();

        }

        public List<LogRecord> getLogs() {

            return Collections.unmodifiableList(ar);

        }

        public void reset() {

            count.set(0);

            ar.clear();

        }

    }

    static class HelperEx extends Exception {

        final Level level;

        HelperEx(Level l) { level = l; }

        Level getLevel() { return level; }

    }

    static class SystemInfo {

        public static String Java() {

            return "Java: " + System.getProperty("java.version") +

                   " installed at " + System.getProperty("java.home");

        }

        public static String OS() {

            return "OS: " + System.getProperty("os.name") +

                   " " + System.getProperty("os.version") +

                   " " + System.getProperty("os.arch");

        }

    }

    static final CountingSupplier supplier = new CountingSupplier();

    static final CountingHandler handler = new CountingHandler();

    static final Logger logger;

    static final Level[] levels = { Level.ALL, Level.OFF, Level.SEVERE,

                                    Level.WARNING, Level.INFO, Level.CONFIG,

                                    Level.FINE, Level.FINER, Level.FINEST };

    static final int[] invokes = { 7, 0, 1, 2, 3, 4, 5, 6, 7 };

    static final int[] log_count = { 10, 0, 1, 2, 3, 4, 6, 8, 10 };

    static {

        logger = Logger.getLogger("LoggerSupplierApisTest");

        logger.setUseParentHandlers(false);

        logger.addHandler(handler);

    }

    public void setup() {

        supplier.reset();

        handler.reset();

    }

    private void testLog() {

        logger.log(Level.SEVERE, supplier);

        logger.log(Level.WARNING, supplier);

        logger.log(Level.INFO, supplier);

        logger.log(Level.CONFIG, supplier);

        logger.log(Level.FINE, supplier);

        logger.log(Level.FINER, supplier);

        logger.log(Level.FINEST, supplier);

        // Lambda expression

        logger.log(Level.FINE, () ->

            "Timestamp: " + System.currentTimeMillis() +

            ", user home directory:  " + System.getProperty("user.home"));

        // Method reference

        logger.log(Level.FINER, SystemInfo::Java);

        logger.log(Level.FINEST, SystemInfo::OS);

    }

    private void testLogThrown() {

        logger.log(Level.SEVERE, new HelperEx(Level.SEVERE), supplier);

        logger.log(Level.WARNING, new HelperEx(Level.WARNING), supplier);

        logger.log(Level.INFO, new HelperEx(Level.INFO), supplier);

        logger.log(Level.CONFIG, new HelperEx(Level.CONFIG), supplier);

        logger.log(Level.FINE, new HelperEx(Level.FINE), supplier);

        logger.log(Level.FINER, new HelperEx(Level.FINER), supplier);

        logger.log(Level.FINEST, new HelperEx(Level.FINEST), supplier);

        // Lambda expression

        logger.log(Level.FINE, new HelperEx(Level.FINE), () ->

            "Timestamp: " + System.currentTimeMillis() +

            ", user home directory:  " + System.getProperty("user.home"));

        // Method reference

        logger.log(Level.FINER, new HelperEx(Level.FINER), SystemInfo::Java);

        logger.log(Level.FINEST, new HelperEx(Level.FINEST), SystemInfo::OS);

    }

    private void testLogp() {

        final String cls = getClass().getName();

        final String mtd = "testLogp";

        logger.logp(Level.SEVERE, cls, mtd, supplier);

        logger.logp(Level.WARNING, cls, mtd, supplier);

        logger.logp(Level.INFO, cls, mtd, supplier);

        logger.logp(Level.CONFIG, cls, mtd, supplier);

        logger.logp(Level.FINE, cls, mtd, supplier);

        logger.logp(Level.FINER, cls, mtd, supplier);

        logger.logp(Level.FINEST, cls, mtd, supplier);

        // Lambda expression

        logger.logp(Level.FINE, cls, mtd, () ->

            "Timestamp: " + System.currentTimeMillis() +

            ", user home directory:  " + System.getProperty("user.home"));

        // Method reference

        logger.logp(Level.FINER, cls, mtd, SystemInfo::Java);

        logger.logp(Level.FINEST, cls, mtd, SystemInfo::OS);

    }

    private void testLogpThrown() {

        final String cls = getClass().getName();

        final String mtd = "testLogpThrown";

        logger.logp(Level.SEVERE, cls, mtd, new HelperEx(Level.SEVERE), supplier);

        logger.logp(Level.WARNING, cls, mtd, new HelperEx(Level.WARNING), supplier);

        logger.logp(Level.INFO, cls, mtd, new HelperEx(Level.INFO), supplier);

        logger.logp(Level.CONFIG, cls, mtd, new HelperEx(Level.CONFIG), supplier);

        logger.logp(Level.FINE, cls, mtd, new HelperEx(Level.FINE), supplier);

        logger.logp(Level.FINER, cls, mtd, new HelperEx(Level.FINER), supplier);

        logger.logp(Level.FINEST, cls, mtd, new HelperEx(Level.FINEST), supplier);

        // Lambda expression

        logger.logp(Level.FINE, cls, mtd, new HelperEx(Level.FINE), () ->

            "Timestamp: " + System.currentTimeMillis() +

            ", user home directory:  " + System.getProperty("user.home"));

        // Method reference

        logger.logp(Level.FINER, cls, mtd, new HelperEx(Level.FINER), SystemInfo::Java);

        logger.logp(Level.FINEST, cls, mtd, new HelperEx(Level.FINEST), SystemInfo::OS);

    }

    private void testLevelConvenientMethods() {

        logger.severe(supplier);

        logger.warning(supplier);

        logger.info(supplier);

        logger.config(supplier);

        logger.fine(supplier);

        logger.finer(supplier);

        logger.finest(supplier);

        // Lambda expression

        logger.fine(() ->

            "Timestamp: " + System.currentTimeMillis() +

            ", user home directory:  " + System.getProperty("user.home"));

        // Method reference

        logger.finer(SystemInfo::Java);

        logger.finest(SystemInfo::OS);

    }

    private void validate(int index, boolean thrown, String methodName) {

        assertEquals(supplier.getCount(), invokes[index]);

        assertEquals(handler.getCount(), log_count[index]);

        // Verify associated Throwable is right

        if (thrown) {

            for (LogRecord r: handler.getLogs()) {

                assertTrue(r.getThrown() instanceof HelperEx, "Validate Thrown");

                HelperEx e = (HelperEx) r.getThrown();

                assertEquals(r.getLevel(), e.getLevel(), "Validate Thrown Log Level");

            }

        }

        if (methodName != null) {

            for (LogRecord r: handler.getLogs()) {

                assertEquals(r.getSourceClassName(), getClass().getName());

                assertEquals(r.getSourceMethodName(), methodName);

            }

        }

    }

    public void verifyLogLevel() {

        for (int i = 0; i < levels.length; i++) {

            logger.setLevel(levels[i]);

            setup();

            testLog();

            validate(i, false, null);

            setup();

            testLogThrown();

            validate(i, true, null);

            setup();

            testLogp();

            validate(i, false, "testLogp");

            setup();

            testLogpThrown();

            validate(i, true, "testLogpThrown");

            setup();

            testLevelConvenientMethods();

            validate(i, false, null);

        }

    }

}