/*

 * Copyright (c) 2015, 2017, 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.IOException;

import java.net.InetSocketAddress;

import java.net.ServerSocket;

import java.net.URI;

import java.net.http.HttpClient;

import java.net.http.HttpRequest;

import java.net.http.HttpResponse;

import java.net.http.HttpResponse.BodyHandlers;

import java.net.http.HttpTimeoutException;

import jdk.testlibrary.SimpleSSLContext;

import javax.net.ServerSocketFactory;

import javax.net.ssl.SSLContext;

import javax.net.ssl.SSLServerSocketFactory;

import java.time.Duration;

import java.util.Arrays;

import java.util.List;

import java.util.concurrent.CompletionException;

import java.util.function.Function;

import static java.lang.System.out;

/**

 * @test

 * @library /lib/testlibrary

 * @build jdk.testlibrary.SimpleSSLContext

 * @summary Basic tests for response timeouts

 * @run main/othervm TimeoutBasic

 */

public class TimeoutBasic {

    static List<Duration> TIMEOUTS = List.of(Duration.ofSeconds(1),

                                             Duration.ofMillis(100),

                                             Duration.ofNanos(99),

                                             Duration.ofNanos(1));

    static final List<Function<HttpRequest.Builder, HttpRequest.Builder>> METHODS =

            Arrays.asList(HttpRequest.Builder::GET,

                          TimeoutBasic::DELETE,

                          TimeoutBasic::PUT,

                          TimeoutBasic::POST,

                          null);

    static final List<HttpClient.Version> VERSIONS =

            Arrays.asList(HttpClient.Version.HTTP_2, HttpClient.Version.HTTP_1_1, null);

    static final List<String> SCHEMES = List.of("https", "http");

    static {

        try {

            SSLContext.setDefault(new SimpleSSLContext().get());

        } catch (IOException x) {

            throw new ExceptionInInitializerError(x);

        }

    }

    public static void main(String[] args) throws Exception {

        for (Function<HttpRequest.Builder, HttpRequest.Builder> m : METHODS) {

            for (HttpClient.Version version : List.of(HttpClient.Version.HTTP_1_1)) {

                for (HttpClient.Version reqVersion : VERSIONS) {

                    for (String scheme : SCHEMES) {

                        ServerSocketFactory ssf;

                        if (scheme.equalsIgnoreCase("https")) {

                            ssf = SSLServerSocketFactory.getDefault();

                        } else {

                            ssf = ServerSocketFactory.getDefault();

                        }

                        test(version, reqVersion, scheme, m, ssf);

                    }

                }

            }

        }

    }

    static HttpRequest.Builder DELETE(HttpRequest.Builder builder) {

        HttpRequest.BodyPublisher noBody = HttpRequest.BodyPublishers.noBody();

        return builder.DELETE(noBody);

    }

    static HttpRequest.Builder PUT(HttpRequest.Builder builder) {

        HttpRequest.BodyPublisher noBody = HttpRequest.BodyPublishers.noBody();

        return builder.PUT(noBody);

    }

    static HttpRequest.Builder POST(HttpRequest.Builder builder) {

        HttpRequest.BodyPublisher noBody = HttpRequest.BodyPublishers.noBody();

        return builder.POST(noBody);

    }

    static HttpRequest newRequest(URI uri,

                                  Duration duration,

                                  HttpClient.Version reqVersion,

                                  Function<HttpRequest.Builder, HttpRequest.Builder> method) {

        HttpRequest.Builder reqBuilder = HttpRequest.newBuilder(uri)

                .timeout(duration);

        if (method != null) reqBuilder = method.apply(reqBuilder);

        if (reqVersion != null) reqBuilder = reqBuilder.version(reqVersion);

        HttpRequest request = reqBuilder.build();

        if (duration.compareTo(Duration.ofSeconds(1)) >= 0) {

            if (method == null || !request.method().equalsIgnoreCase("get")) {

                out.println("Skipping " + duration + " for " + request.method());

                return null;

            }

        }

        return request;

    }

    public static void test(HttpClient.Version version,

                            HttpClient.Version reqVersion,

                            String scheme,

                            Function<HttpRequest.Builder, HttpRequest.Builder> method,

                            ServerSocketFactory ssf)

            throws Exception

    {

        HttpClient.Builder builder = HttpClient.newBuilder()

                .proxy(HttpClient.Builder.NO_PROXY);

        if (version != null) builder.version(version);

        HttpClient client = builder.build();

        out.printf("%ntest(version=%s, reqVersion=%s, scheme=%s)%n", version, reqVersion, scheme);

        try (ServerSocket ss = ssf.createServerSocket()) {

            ss.setReuseAddress(false);

            int port = ss.getLocalPort();

            out.println("--- TESTING Async");

            int count = 0;

            for (Duration duration : TIMEOUTS) {

                out.println("  with duration of " + duration);

                HttpRequest request = newRequest(uri, duration, reqVersion, method);

                if (request == null) continue;

                count++;

                try {

                    HttpResponse<?> resp = client.sendAsync(request, BodyHandlers.discarding()).join();

                    out.println("Unexpected response for: " + request);

                    out.println("\t from " + ss.getLocalSocketAddress());

                    out.println("Response is: " + resp);

                    out.println("Headers: " + resp.headers().map());

                    out.println("Body (should be null): " + resp.body());

                    throw new RuntimeException("Unexpected response: " + resp.statusCode());

                } catch (CompletionException e) {

                    if (!(e.getCause() instanceof HttpTimeoutException)) {

                        throw new RuntimeException("Unexpected exception: " + e.getCause());

                    } else {

                        out.println("Caught expected timeout: " + e.getCause());

                    }

                }

            }

            assert count >= TIMEOUTS.size() -1;

            out.println("--- TESTING Sync");

            count = 0;

            for (Duration duration : TIMEOUTS) {

                out.println("  with duration of " + duration);

                HttpRequest request = newRequest(uri, duration, reqVersion, method);

                if (request == null) continue;

                count++;

                try {

                    client.send(request, BodyHandlers.discarding());

                } catch (HttpTimeoutException e) {

                    out.println("Caught expected timeout: " + e);

                }

            }

            assert count >= TIMEOUTS.size() -1;

        }

    }

}