/* @test

 * @summary Unit test for java.net.ResponseCacheHandler

 * @bug 4837267

 * @author Yingxian Wang

 */

import java.net.*;

import java.util.*;

import java.io.*;

import java.nio.*;

import java.nio.channels.ReadableByteChannel;

import sun.net.www.ParseUtil;

/**

 * Request should get serviced by the cache handler. Response get

 * saved through the cache handler.

 */

public class ResponseCacheHandlerTest implements Runnable {

    ServerSocket ss;

    static URL url1;

    static URL url2;

    static boolean flag = false;

    /* 

     * Our "http" server to return a 404 */

    public void run() {

	try {

	    Socket s = ss.accept();

	    // check request contains "Cookie"

	    InputStream is = s.getInputStream ();

	    BufferedReader r = new BufferedReader(new InputStreamReader(is));

	    boolean flag = false;

	    String x;

	    while ((x=r.readLine()) != null) {

		if (x.length() ==0) {

		    break;

		}

	    }

	    PrintStream out = new PrintStream( 

				 new BufferedOutputStream(

				    s.getOutputStream() ));

	    /* send file2.1 */

	    File file2 = new File("file2.1");

	    out.print("HTTP/1.1 200 OK\r\n");

	    out.print("Content-Type: text/html; charset=iso-8859-1\r\n");

	    out.print("Content-Length: "+file2.length()+"\r\n");

	    out.print("Connection: close\r\n");

	    out.print("\r\n");

	    FileInputStream fis = new FileInputStream(file2);

	    byte[] buf = new byte[(int)file2.length()];

	    int len;

	    while ((len = fis.read(buf)) != -1) {

		out.print(new String(buf));

	    }

	    out.flush();

	    s.close();

	    ss.close();

	} catch (Exception e) { 

	    e.printStackTrace();

	}

    }

    ResponseCacheHandlerTest() throws Exception {

	/* start the server */

	ss = new ServerSocket(0);

	(new Thread(this)).start();

	/* establish http connection to server */

	url1 = new URL("http://localhost/file1.cache");

	HttpURLConnection http = (HttpURLConnection)url1.openConnection();

	InputStream is = null;

	System.out.println("headers1:"+http.getHeaderFields());

	try {

	    is = http.getInputStream();

	} catch (IOException ioex) {

	    throw new RuntimeException(ioex.getMessage());

	}

	BufferedReader r = new BufferedReader(new InputStreamReader(is));

	String x;

	File fileout = new File("file1");

	PrintStream out = new PrintStream( 

				 new BufferedOutputStream(

				    new FileOutputStream(fileout)));

	while ((x=r.readLine()) != null) {

	    out.print(x+"\n");

	}

	out.flush();

	out.close();

	// if !(file1.out == file1.cache)

	// throw new RuntimeException("testing ResponseCacheHandler.get() failed");

	http.disconnect();

	// testing ResponseCacheHandler.put()

	url2 = new URL("http://localhost:" + 

		       Integer.toString(ss.getLocalPort())+"/file2.1");

	http = (HttpURLConnection)url2.openConnection();

	System.out.println("headers2:"+http.getHeaderFields());

	try {

	    is = http.getInputStream();

	} catch (IOException ioex) {

	    throw new RuntimeException(ioex.getMessage());

	}

	DataInputStream dis = new DataInputStream(is);

	fileout = new File("file2.2");

	byte[] buf = new byte[(int)new File("file2.1").length()];

	DataOutputStream dos =  new DataOutputStream(new FileOutputStream(fileout));

	int len;

	while ((len=dis.read(buf)) != -1) {

	    dos.write(buf, 0, len);

	}

	dos.flush();

	dos.close();

	// if !(file2.1 == file2.2 == file2.cache)

	// throw new RuntimeException("testing ResponseCacheHandler.put() failed");

    }

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

	ResponseCacheHandler.setDefault(new MyResponseCacheHandler());

	new ResponseCacheHandlerTest();

    }

    static class MyResponseCacheHandler extends ResponseCacheHandler {

	public Response 

	get(URI uri, Map requestHeaders)

	    throws IOException {

	    if (uri.equals(ParseUtil.toURI(url1))) {

		return new MyResponse("file1.cache");

	    } if (uri.equals(ParseUtil.toURI(url2)) && !flag) {

		flag = true;

		return null;

	    } else {

		return new MyResponse("file2.cache");

	    }

	}

	public boolean put(URI uri, Response response)  throws IOException {

	    // save cache to file2.cache

	    // 1. serialize headers into file2.cache

	    // 2. write data to file2.cache

	    File file = new File("file2.cache");

	    FileOutputStream fos = new FileOutputStream(file);

	    ObjectOutputStream oos = new ObjectOutputStream(fos);

	    oos.writeObject(response.getHeaders());

	    ReadableByteChannel rbc = response.getBody();

	    int len = (int)new File("file2.1").length();

	    ByteBuffer buf = ByteBuffer.allocate(len);

	    byte[] b = new byte[len];

	    while ((len = rbc.read(buf)) != -1) {

		buf.flip();

		buf.get(b, 0, len);

		buf.compact();

		oos.write(b, 0, len);

	    }

	    oos.close();

	    return true;

	}

    }

    static class MyResponse extends Response {

	FileInputStream fis;

	Map headers;

	public MyResponse(String filename) {

	    try {

		fis = new FileInputStream(new File(filename));

		headers = (Map)new ObjectInputStream(fis).readObject();

	    } catch (Exception ex) {

		throw new RuntimeException(ex.getMessage());

	    }

	}

	public ReadableByteChannel getBody() throws IOException {

	    return fis.getChannel();

	}

	public Map getHeaders() throws IOException {

	    return headers;

	}

    }

}