jdk/test/sun/net/www/http/ChunkedInputStream/ChunkedEncodingTest.java
changeset 2 90ce3da70b43
child 1339 5fb372f1d51f
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/sun/net/www/http/ChunkedInputStream/ChunkedEncodingTest.java	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,116 @@
+/*
+ * Copyright 2004-2007 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.com if you need additional information or
+ * have any questions.
+ */
+
+/**
+ * @test
+ * @bug 4333920
+ * @library ../../../../../sun/net/www/httptest/
+ * @build HttpCallback HttpServer ClosedChannelList HttpTransaction
+ * @run main ChunkedEncodingTest
+ * @summary ChunkedEncodingTest unit test
+ */
+
+import java.io.*;
+import java.net.*;
+import java.security.*;
+
+public class ChunkedEncodingTest implements HttpCallback {
+    private static String FNPrefix;
+    private String[] respBody = new String[52];
+    private byte[][] bufs = new byte[52][8*1024];
+    private static MessageDigest md5;
+    private static byte[] file1Mac, file2Mac;
+    public void request (HttpTransaction req) {
+        try {
+            FileInputStream fis = new FileInputStream(FNPrefix+"test.txt");
+            DigestInputStream dis = null;
+            md5.reset();
+            dis = new DigestInputStream(fis, md5);
+            for (int i = 0; i < 52; i++) {
+                int n = dis.read(bufs[i]);
+                respBody[i] = new String(bufs[i], 0, n);
+            }
+            file1Mac = dis.getMessageDigest().digest();
+            dis.close();
+            req.setResponseEntityBody(respBody);
+            req.sendResponse(200, "OK");
+            req.orderlyClose();
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+
+    static void read (InputStream is) throws IOException {
+        int c;
+        System.out.println ("reading");
+
+        DigestInputStream dis = null;
+        md5.reset();
+        dis = new DigestInputStream(is, md5);
+        while ((c=dis.read()) != -1);
+        file2Mac = dis.getMessageDigest().digest();
+        dis.close();
+        System.out.println ("finished reading");
+    }
+
+    static void client (String u) throws Exception {
+        URL url = new URL (u);
+        System.out.println ("client opening connection to: " + u);
+        URLConnection urlc = url.openConnection ();
+        InputStream is = urlc.getInputStream ();
+        read (is);
+        is.close();
+    }
+
+    static HttpServer server;
+
+    public static void test () throws Exception {
+        try {
+
+            FNPrefix = System.getProperty("test.src", ".")+"/";
+            md5 = MessageDigest.getInstance("MD5");
+            server = new HttpServer (new ChunkedEncodingTest(), 1, 10, 0);
+            System.out.println ("Server: listening on port: " + server.getLocalPort());
+            client ("http://localhost:"+server.getLocalPort()+"/d1/foo.html");
+        } catch (Exception e) {
+            if (server != null) {
+                server.terminate();
+            }
+            throw e;
+        }
+        if (!MessageDigest.isEqual(file1Mac, file2Mac)) {
+            except ("The file sent by server is different from the original file");
+        }
+
+        server.terminate();
+    }
+
+    public static void main(String[] args) throws Exception {
+        test();
+    }
+
+    public static void except (String s) {
+        server.terminate();
+        throw new RuntimeException (s);
+    }
+}