/* @test @(#)file2.1	1.1 03/08/09
 * @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;
	}
    }
}