/*

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

import java.util.concurrent.CompletableFuture;

import javax.net.ssl.SSLContext;

import javax.net.ServerSocketFactory;

import javax.net.ssl.SSLServerSocketFactory;

import jdk.incubator.http.HttpClient;

import jdk.incubator.http.HttpClient.Version;

import jdk.incubator.http.HttpRequest;

import jdk.incubator.http.HttpResponse;

import jdk.testlibrary.SimpleSSLContext;

import static java.lang.System.out;

import static java.lang.String.format;

import static jdk.incubator.http.HttpResponse.BodyHandler.asString;

/**

 * @test

 * @bug 8087112

 * @library /lib/testlibrary

 * @build jdk.testlibrary.SimpleSSLContext

 * @build MockServer

 * @run main/othervm -Djdk.internal.httpclient.debug=true -Djdk.httpclient.HttpClient.log=all SplitResponse

 */

/**

 * Similar test to QuickResponses except that each byte of the response

 * is sent in a separate packet, which tests the stability of the implementation

 * for receiving unusual packet sizes. Additionally, tests scenarios there

 * connections that are retrieved from the connection pool may reach EOF before

 * being reused.

 */

public class SplitResponse {

    static String response(String body, boolean serverKeepalive) {

        StringBuilder sb = new StringBuilder();

        sb.append("HTTP/1.1 200 OK\r\n");

        if (!serverKeepalive)

            sb.append("Connection: Close\r\n");

        sb.append("Content-length: ").append(body.length()).append("\r\n");

        sb.append("\r\n");

        sb.append(body);

        return sb.toString();

    }

    static final String responses[] = {

        "Lorem ipsum",

        "dolor sit amet",

        "consectetur adipiscing elit, sed do eiusmod tempor",

        "quis nostrud exercitation ullamco",

        "laboris nisi",

        "ut",

        "aliquip ex ea commodo consequat." +

        "Duis aute irure dolor in reprehenderit in voluptate velit esse" +

        "cillum dolore eu fugiat nulla pariatur.",

        "Excepteur sint occaecat cupidatat non proident."

    };

    final ServerSocketFactory factory;

    final SSLContext context;

    final boolean useSSL;

    SplitResponse(boolean useSSL) throws IOException {

        this.useSSL = useSSL;

        context = new SimpleSSLContext().get();

        SSLContext.setDefault(context);

        factory = useSSL ? SSLServerSocketFactory.getDefault()

                         : ServerSocketFactory.getDefault();

    }

    public HttpClient newHttpClient() {

        HttpClient client;

        if (useSSL) {

            client = HttpClient.newBuilder()

                               .sslContext(context)

                               .build();

        } else {

            client = HttpClient.newHttpClient();

        }

        return client;

    }

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

        boolean useSSL = false;

        if (args != null && args.length == 1) {

            useSSL = "SSL".equals(args[0]);

        }

        SplitResponse sp = new SplitResponse(useSSL);

        for (Version version : Version.values()) {

            for (boolean serverKeepalive : new boolean[]{ true, false }) {

                // Note: the mock server doesn't support Keep-Alive, but

                // pretending that it might exercises code paths in and out of

                // the connection pool, and retry logic

                for (boolean async : new boolean[]{ true, false }) {

                    sp.test(version, serverKeepalive, async);

                }

            }

        }

    }

    // @Test

    void test(Version version, boolean serverKeepalive, boolean async)

        throws Exception

    {

        out.println(format("*** version %s, serverKeepAlive: %s, async: %s ***",

                           version, serverKeepalive, async));

        MockServer server = new MockServer(0, factory);

        URI uri = new URI(server.getURL());

        out.println("server is: " + uri);

        server.start();

        // The following code can be uncommented to verify that the

        // MockServer will reject rogue requests whose URI does not

        // contain "/foo/".

        //

        //        Thread rogue = new Thread() {

        //            public void run() {

        //                try {

        //                    HttpClient client = newHttpClient();

        //                    URI uri2 = URI.create(uri.toString().replace("/foo/","/"));

        //                    HttpRequest request = HttpRequest

        //                        .newBuilder(uri2).version(version).build();

        //                    while (true) {

        //                        try {

        //                            client.send(request, HttpResponse.BodyHandler.asString());

        //                        } catch (IOException ex) {

        //                            System.out.println("Client rejected " + request);

        //                        }

        //                        sleep(250);

        //                    }

        //                } catch ( Throwable x) {

        //                }

        //            }

        //        };

        //        rogue.setDaemon(true);

        //        rogue.start();

        HttpClient client = newHttpClient();

        HttpRequest request = HttpRequest.newBuilder(uri).version(version).build();

        HttpResponse<String> r;

        CompletableFuture<HttpResponse<String>> cf1;

        try {

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

                out.println("----- iteration " + i + " -----");

                String body = responses[i];

                Thread t = sendSplitResponse(response(body, serverKeepalive), server);

                if (async) {

                    out.println("send async: " + request);

                    cf1 = client.sendAsync(request, asString());

                    r = cf1.get();

                } else { // sync

                    out.println("send sync: " + request);

                    r = client.send(request, asString());

                }

                if (r.statusCode() != 200)

                    throw new RuntimeException("Failed");

                String rxbody = r.body();

                out.println("received " + rxbody);

                if (!rxbody.equals(body))

                    throw new RuntimeException(format("Expected:%s, got:%s", body, rxbody));

                t.join();

                conn.close();

            }

        } finally {

            server.close();

        }

        System.out.println("OK");

    }

    // required for cleanup

    volatile MockServer.Connection conn;

    // Sends the response, mostly, one byte at a time with a small delay

    // between bytes, to encourage that each byte is read in a separate read

    Thread sendSplitResponse(String s, MockServer server) {

        System.out.println("Sending: ");

        Thread t = new Thread(() -> {

            System.out.println("Waiting for server to receive headers");

            conn = server.activity();

            System.out.println("Start sending response");

            try {

                int len = s.length();

                out.println("sending " + s);

                for (int i = 0; i < len; i++) {

                    String onechar = s.substring(i, i + 1);

                    conn.send(onechar);

                    Thread.sleep(10);

                }

                out.println("sent " + s);

            } catch (IOException | InterruptedException e) {

                throw new RuntimeException(e);

            }

        });

        t.setDaemon(true);

        t.start();

        return t;

    }

}