/*

 * 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.

 */

/*

 * @test

 * @bug 8213418

 * @summary Ensure correct impl supported socket options

 * @run testng ImplSupportedOptions

 */

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.net.InetAddress;

import java.net.ServerSocket;

import java.net.Socket;

import java.net.SocketAddress;

import java.net.SocketException;

import java.net.SocketImpl;

import java.net.SocketOption;

import java.net.StandardSocketOptions;

import java.util.Set;

import org.testng.annotations.Test;

import static org.testng.Assert.assertEquals;

import static org.testng.Assert.assertTrue;

public class ImplSupportedOptions {

    @Test

    public void socketSupportedOptions() throws Exception {

        Socket s = new Socket();

        Set<?> standardOptions = s.supportedOptions();

        assertTrue(standardOptions.contains(StandardSocketOptions.SO_LINGER),

                   "Expected SO_LINGER, in:" + standardOptions);

        assertEquals(standardOptions, s.supportedOptions());

        assertEquals(standardOptions, s.supportedOptions());

        s = new DummySocket();

        Set<?> dummyOptions = s.supportedOptions();

        assertEquals(dummyOptions.size(), 1);

        assertTrue(dummyOptions.contains(DummySocketImpl.SOCKET_OPT));

        assertEquals(dummyOptions, s.supportedOptions());

        assertEquals(dummyOptions, s.supportedOptions());

        s = new Socket();

        standardOptions = s.supportedOptions();

        assertTrue(standardOptions.contains(StandardSocketOptions.SO_LINGER),

                   "Expected SO_LINGER, in:" + standardOptions);

        assertEquals(standardOptions, s.supportedOptions());

        assertEquals(standardOptions, s.supportedOptions());

        s = new DummySocket();

        dummyOptions = s.supportedOptions();

        assertEquals(dummyOptions.size(), 1);

        assertTrue(dummyOptions.contains(DummySocketImpl.SOCKET_OPT));

        assertEquals(dummyOptions, s.supportedOptions());

        assertEquals(dummyOptions, s.supportedOptions());

    }

    @Test

    public void serverSocketSupportedOptions() throws Exception {

        ServerSocket s = new ServerSocket();

        Set<?> standardOptions = s.supportedOptions();

        assertTrue(standardOptions.contains(StandardSocketOptions.SO_REUSEADDR),

                   "Expected SO_REUSEADDR, in:" + standardOptions);

        assertEquals(standardOptions, s.supportedOptions());

        assertEquals(standardOptions, s.supportedOptions());

        s = new DummyServerSocket();

        Set<?> dummyOptions = s.supportedOptions();

        assertEquals(dummyOptions.size(), 1);

        assertTrue(dummyOptions.contains(DummySocketImpl.SOCKET_OPT));

        assertEquals(dummyOptions, s.supportedOptions());

        assertEquals(dummyOptions, s.supportedOptions());

        s = new ServerSocket();

        standardOptions = s.supportedOptions();

        assertTrue(standardOptions.contains(StandardSocketOptions.SO_REUSEADDR),

                   "Expected SO_REUSEADDR, in:" + standardOptions);

        assertEquals(standardOptions, s.supportedOptions());

        assertEquals(standardOptions, s.supportedOptions());

        s = new DummyServerSocket();

        dummyOptions = s.supportedOptions();

        assertEquals(dummyOptions.size(), 1);

        assertTrue(dummyOptions.contains(DummySocketImpl.SOCKET_OPT));

        assertEquals(dummyOptions, s.supportedOptions());

        assertEquals(dummyOptions, s.supportedOptions());

    }

    static class DummySocket extends Socket {

        DummySocket() throws IOException  {

            super(new DummySocketImpl());

        }

    }

    static class DummyServerSocket extends ServerSocket {

        DummyServerSocket() throws IOException  {

            super(new DummySocketImpl());

        }

    }

    static class DummySocketImpl extends SocketImpl {

        public static final SocketOption<String> SOCKET_OPT =

                new ImplSocketOption<>("SOCKET_OPT", String.class);

        private static class ImplSocketOption<T> implements SocketOption<T> {

            private final String name;

            private final Class<T> type;

            ImplSocketOption(String name, Class<T> type) {

                this.name = name;

                this.type = type;

            }

            @Override public String name() { return name; }

            @Override public Class<T> type() { return type; }

            @Override public String toString() { return name; }

        }

        private final Set<SocketOption<?>> SO = Set.of(SOCKET_OPT);

        @Override

        public Set<SocketOption<?>> supportedOptions() { return SO; }

        // ---

        @Override

        protected void create(boolean stream) throws IOException { }

        @Override

        protected void connect(String host, int port) throws IOException { }

        @Override

        protected void connect(InetAddress address, int port) throws IOException { }

        @Override

        protected void connect(SocketAddress address, int timeout) throws IOException { }

        @Override

        protected void bind(InetAddress host, int port) throws IOException { }

        @Override

        protected void listen(int backlog) throws IOException { }

        @Override

        protected void accept(SocketImpl s) throws IOException { }

        @Override

        protected InputStream getInputStream() throws IOException { return null; }

        @Override

        protected OutputStream getOutputStream() throws IOException { return null; }

        @Override

        protected int available() throws IOException { return 0; }

        @Override

        protected void close() throws IOException { }

        @Override

        protected void sendUrgentData(int data) throws IOException { }

        @Override

        public void setOption(int optID, Object value) throws SocketException { }

        @Override

        public Object getOption(int optID) throws SocketException { return null; }

    }

}