--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/java/net/ResponseCache/file2.1	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,184 @@
+/* @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;
+	}
+    }
+}