/*
 * Copyright (c) 2018 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

import java.lang.*;
import java.io.*;
import java.net.*;

import com.sun.ejte.ccl.reporter.*;

public class WebTest extends Thread
{
    
    private static int count = 0;
    private static int EXPECTED_COUNT = 3;
    
    static SimpleReporterAdapter stat=
        new SimpleReporterAdapter("appserv-tests");
    private static final String HOST = "localhost";
    private static final int PORT = 7007;
    private ServerSocket ss = null;

    public WebTest() {
        try {
           ss = new ServerSocket(PORT, 0, InetAddress.getByName(HOST));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void run() {
        try {
            stat.addDescription("SocketImpl Test");
            Socket s = ss.accept();
            s.setSoLinger(false, 10);
            InputStream is = s.getInputStream();
            OutputStream os = s.getOutputStream();
            int count = 1;
            os.write(count);        
            
            while (true) {                
                if (count > 10) {
                    s.close();
                    break;
                }
                System.out.println("Count is :" + count);
                int i = is.read();
                if (count != i) throw new RuntimeException ("Data Wrong: Expected :" + count + " Obtained:" + i);                
                if (count == 5) { 
                    Thread.sleep(6000);                    
                    count++;
                    int j = is.read();
                    if (count != j) throw new RuntimeException ("Data Wrong: Expected :" + count + " Obtained:" + i);
                }
                os.write(++count);
            }        
        } catch (RuntimeException rex) {
            rex.printStackTrace();
            stat.addStatus("SocketImpl SocketImpl", stat.FAIL);
        } catch (Exception ex) {
            stat.addStatus("SocketImpl SocketImpl", stat.FAIL);
            ex.printStackTrace();
        }
        stat.addStatus("SocketImpl SocketImpl", stat.PASS);
    }

    public static void main(String args[])
    {

        // The stat reporter writes out the test info and results
        // into the top-level quicklook directory during a run.
      
        stat.addDescription("Standalone Servlet/Filter war test");

        String host = args[0];
        String portS = args[1];
        String contextRoot = args[2];

        int port = new Integer(portS).intValue();
        String name;

        WebTest test = new WebTest();
        test.start();
        
        try {
            goGet(host, port, "FILTER", contextRoot + "/ServletTest" );
            Thread.sleep(2000);
            //Thread.currentThread().join();
        } catch (Throwable t) {
            System.out.println(t.getMessage());
        }

        if (count != EXPECTED_COUNT){
            stat.addStatus("Test standalonewarUNPREDICTED-FAILURE",
                            stat.FAIL);
        }           
        stat.printSummary("web/standalonewar---> expect " + EXPECTED_COUNT);
    }

    private static void goGet(String host, int port,
                              String result, String contextPath)
         throws Exception
    {
        long time = System.currentTimeMillis();
        Socket s = new Socket(host, port);
        OutputStream os = s.getOutputStream();

        System.out.println(("GET " + contextPath + " HTTP/1.1\n"));
        os.write(("GET " + contextPath + " HTTP/1.1\n").getBytes());
        os.write(("Host: localhost\n").getBytes());
        os.write("\n".getBytes());
        
        InputStream is = s.getInputStream();
        System.out.println("Time: " + (System.currentTimeMillis() - time));
        BufferedReader bis = new BufferedReader(new InputStreamReader(is));
        String line = null;

        try{
            int index;
            while ((line = bis.readLine()) != null) {
                index = line.indexOf(result);
                System.out.println(line);
                if (index != -1) {
                    index = line.indexOf(":");
                    String status = line.substring(index+1);
                    
                    if (status.equalsIgnoreCase("PASS")){
                        stat.addStatus("web-requestdispatcher: " + line.substring(0,index), stat.PASS);
                    } else {
                        stat.addStatus("web-requestdispatcher: " + line.substring(0,index), stat.FAIL);                       
                    }
                    count++;
                } 
            }
        } catch( Exception ex){
            ex.printStackTrace();   
            throw new Exception("Test UNPREDICTED-FAILURE");
         }
   }
  
}
