--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/test/jdk/java/net/CookieHandler/CookieManagerTest.java Tue Sep 12 19:03:39 2017 +0200
@@ -0,0 +1,345 @@
+/*
+ * Copyright (c) 2005, 2013, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+
+/*
+ * @test
+ * @summary Unit test for java.net.CookieManager
+ * @bug 6244040 7150552 7051862
+ * @modules jdk.httpserver
+ * @run main/othervm -ea CookieManagerTest
+ * @author Edward Wang
+ */
+
+import com.sun.net.httpserver.*;
+import java.io.IOException;
+import java.net.*;
+import static java.net.Proxy.NO_PROXY;
+
+public class CookieManagerTest {
+
+ static CookieTransactionHandler httpTrans;
+ static HttpServer server;
+
+ static final String hostAddress = getAddr();
+
+ /** Returns an IP literal suitable for use by the test. */
+ static String getAddr() {
+ try {
+ InetAddress lh = InetAddress.getLocalHost();
+ System.out.println("Trying: " + lh);
+ if (lh.isReachable(5_000)) {
+ System.out.println("Using: " + lh);
+ return lh.getHostAddress();
+ }
+ } catch (IOException x) {
+ System.out.println("Debug: caught:" + x);
+ }
+ System.out.println("Using: \"127.0.0.1\"");
+ return "127.0.0.1";
+ }
+
+ public static void main(String[] args) throws Exception {
+ startHttpServer();
+ makeHttpCall();
+
+ if (httpTrans.badRequest) {
+ throw new RuntimeException("Test failed : bad cookie header");
+ }
+ checkCookiePolicy();
+ }
+
+ public static void startHttpServer() throws IOException {
+ httpTrans = new CookieTransactionHandler();
+ server = HttpServer.create(new InetSocketAddress(0), 0);
+ server.createContext("/", httpTrans);
+ server.start();
+ }
+
+ /*
+ * Checks if CookiePolicy.ACCEPT_ORIGINAL_SERVER#shouldAccept()
+ * returns false for null arguments
+ */
+ private static void checkCookiePolicy() throws Exception {
+ CookiePolicy cp = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
+ boolean retVal;
+ retVal = cp.shouldAccept(null, null);
+ checkValue(retVal);
+ retVal = cp.shouldAccept(null, new HttpCookie("CookieName", "CookieVal"));
+ checkValue(retVal);
+ retVal = cp.shouldAccept((new URL("http", "localhost", 2345, "/")).toURI(),
+ null);
+ checkValue(retVal);
+ }
+
+ private static void checkValue(boolean val) {
+ if (val)
+ throw new RuntimeException("Return value is not false!");
+ }
+
+ public static void makeHttpCall() throws IOException {
+ try {
+ int port = server.getAddress().getPort();
+ System.out.println("http server listenining on: " + port);
+
+ // install CookieManager to use
+ CookieHandler.setDefault(new CookieManager());
+
+ for (int i = 0; i < CookieTransactionHandler.testCount; i++) {
+ System.out.println("====== CookieManager test " + (i+1)
+ + " ======");
+ ((CookieManager)CookieHandler.getDefault())
+ .setCookiePolicy(CookieTransactionHandler.testPolicies[i]);
+ ((CookieManager)CookieHandler.getDefault())
+ .getCookieStore().removeAll();
+ URL url = new URL("http" ,
+ hostAddress,
+ server.getAddress().getPort(),
+ CookieTransactionHandler.testCases[i][0]
+ .serverPath);
+ System.out.println("Requesting " + url);
+ HttpURLConnection uc = (HttpURLConnection)url.openConnection(NO_PROXY);
+ uc.getResponseCode();
+ uc.disconnect();
+ }
+ } finally {
+ server.stop(0);
+ }
+ }
+}
+
+class CookieTransactionHandler implements HttpHandler {
+
+ private int testcaseDone = 0;
+ private int testDone = 0;
+
+ public static boolean badRequest = false;
+ // the main test control logic will also loop exactly this number
+ // to send http request
+ public static final int testCount = 6;
+
+ @Override
+ public void handle(HttpExchange exchange) throws IOException {
+ if (testDone < testCases[testcaseDone].length) {
+ // still have other tests to run,
+ // check the Cookie header and then redirect it
+ if (testDone > 0) checkRequest(exchange.getRequestHeaders());
+ exchange.getResponseHeaders().add("Location",
+ testCases[testcaseDone][testDone].serverPath);
+ exchange.getResponseHeaders()
+ .add(testCases[testcaseDone][testDone].headerToken,
+ testCases[testcaseDone][testDone].cookieToSend);
+ exchange.sendResponseHeaders(302, -1);
+ testDone++;
+ } else {
+ // the last test of this test case
+ if (testDone > 0) checkRequest(exchange.getRequestHeaders());
+ testcaseDone++;
+ testDone = 0;
+ exchange.sendResponseHeaders(200, -1);
+ }
+ exchange.close();
+ }
+
+ private void checkRequest(Headers hdrs) {
+
+ assert testDone > 0;
+ String cookieHeader = hdrs.getFirst("Cookie");
+ if (cookieHeader != null &&
+ cookieHeader
+ .equalsIgnoreCase(testCases[testcaseDone][testDone-1]
+ .cookieToRecv))
+ {
+ System.out.printf("%15s %s\n", "PASSED:", cookieHeader);
+ } else {
+ System.out.printf("%15s %s\n", "FAILED:", cookieHeader);
+ System.out.printf("%15s %s\n\n", "should be:",
+ testCases[testcaseDone][testDone-1].cookieToRecv);
+ badRequest = true;
+ }
+ }
+
+ // test cases
+ public static class CookieTestCase {
+ public String headerToken;
+ public String cookieToSend;
+ public String cookieToRecv;
+ public String serverPath;
+
+ public CookieTestCase(String h, String cts, String ctr, String sp) {
+ headerToken = h;
+ cookieToSend = cts;
+ cookieToRecv = ctr;
+ serverPath = sp;
+ }
+ };
+
+ /*
+ * these two must match each other,
+ * i.e. testCases.length == testPolicies.length
+ */
+
+ // the test cases to run; each test case may contain multiple roundtrips
+ public static CookieTestCase[][] testCases = null;
+ // indicates what CookiePolicy to use with each test cases
+ public static CookiePolicy[] testPolicies = null;
+
+ CookieTransactionHandler() {
+ testCases = new CookieTestCase[testCount][];
+ testPolicies = new CookiePolicy[testCount];
+
+ String localHostAddr = CookieManagerTest.hostAddress;
+
+ int count = 0;
+
+ // an http session with Netscape cookies exchanged
+ testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
+ testCases[count++] = new CookieTestCase[]{
+ new CookieTestCase("Set-Cookie",
+ "CUSTOMER=WILE:BOB; " +
+ "path=/; expires=Sat, 09-Nov-2030 23:12:40 GMT;" + "domain=." +
+ localHostAddr,
+ "CUSTOMER=WILE:BOB",
+ "/"
+ ),
+ new CookieTestCase("Set-Cookie",
+ "PART_NUMBER=ROCKET_LAUNCHER_0001; path=/;" + "domain=." + localHostAddr,
+ "CUSTOMER=WILE:BOB; PART_NUMBER=ROCKET_LAUNCHER_0001",
+ "/"
+ ),
+ new CookieTestCase("Set-Cookie",
+ "SHIPPING=FEDEX; path=/foo;" + "domain=." + localHostAddr,
+ "CUSTOMER=WILE:BOB; PART_NUMBER=ROCKET_LAUNCHER_0001",
+ "/"
+ ),
+ new CookieTestCase("Set-Cookie",
+ "SHIPPING=FEDEX; path=/foo;" + "domain=." + localHostAddr,
+ "CUSTOMER=WILE:BOB; PART_NUMBER=ROCKET_LAUNCHER_0001; SHIPPING=FEDEX",
+ "/foo"
+ )
+ };
+
+ // check whether or not path rule is applied
+ testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
+ testCases[count++] = new CookieTestCase[]{
+ new CookieTestCase("Set-Cookie",
+ "PART_NUMBER=ROCKET_LAUNCHER_0001; path=/;" + "domain=." + localHostAddr,
+ "PART_NUMBER=ROCKET_LAUNCHER_0001",
+ "/"
+ ),
+ new CookieTestCase("Set-Cookie",
+ "PART_NUMBER=RIDING_ROCKET_0023; path=/ammo;" + "domain=." + localHostAddr,
+ "PART_NUMBER=RIDING_ROCKET_0023; PART_NUMBER=ROCKET_LAUNCHER_0001",
+ "/ammo"
+ )
+ };
+
+ // an http session with rfc2965 cookies exchanged
+ testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
+ testCases[count++] = new CookieTestCase[]{
+ new CookieTestCase("Set-Cookie2",
+ "Customer=\"WILE_E_COYOTE\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr,
+ "$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"",
+ "/acme/login"
+ ),
+ new CookieTestCase("Set-Cookie2",
+ "Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\";Path=\"/acme\";" + "domain=." + localHostAddr,
+ "$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";" + "$Domain=\"." +
+ localHostAddr + "\"" + "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";"
+ + "$Domain=\"." + localHostAddr + "\"",
+ "/acme/pickitem"
+ ),
+ new CookieTestCase("Set-Cookie2",
+ "Shipping=\"FedEx\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr,
+ "$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";" + "$Domain=\"." + localHostAddr +
+ "\"" + "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";" + "$Domain=\"."
+ + localHostAddr + "\"" + "; Shipping=\"FedEx\";$Path=\"/acme\";" +
+ "$Domain=\"." + localHostAddr + "\"",
+ "/acme/shipping"
+ )
+ };
+
+ // check whether or not the path rule is applied
+ testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
+ testCases[count++] = new CookieTestCase[]{
+ new CookieTestCase("Set-Cookie2",
+ "Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr,
+ "$Version=\"1\"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"",
+ "/acme/ammo"
+ ),
+ new CookieTestCase("Set-Cookie2",
+ "Part_Number=\"Riding_Rocket_0023\"; Version=\"1\"; Path=\"/acme/ammo\";" + "domain=."
+ + localHostAddr,
+ "$Version=\"1\"; Part_Number=\"Riding_Rocket_0023\";$Path=\"/acme/ammo\";$Domain=\"."
+ + localHostAddr + "\"" + "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";"
+ + "$Domain=\"." + localHostAddr + "\"",
+ "/acme/ammo"
+ ),
+ new CookieTestCase("",
+ "",
+ "$Version=\"1\"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";" + "$Domain=\"." + localHostAddr + "\"",
+ "/acme/parts"
+ )
+ };
+
+ // new cookie should overwrite old cookie
+ testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
+ testCases[count++] = new CookieTestCase[]{
+ new CookieTestCase("Set-Cookie2",
+ "Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr,
+ "$Version=\"1\"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"",
+ "/acme"
+ ),
+ new CookieTestCase("Set-Cookie2",
+ "Part_Number=\"Rocket_Launcher_2000\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr,
+ "$Version=\"1\"; Part_Number=\"Rocket_Launcher_2000\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"",
+ "/acme"
+ )
+ };
+
+ // cookies without domain attributes
+ // RFC 2965 states that domain should default to host
+ testPolicies[count] = CookiePolicy.ACCEPT_ALL;
+ testCases[count++] = new CookieTestCase[]{
+ new CookieTestCase("Set-Cookie2",
+ "Customer=\"WILE_E_COYOTE\"; Version=\"1\"; Path=\"/acme\"",
+ "$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"",
+ "/acme/login"
+ ),
+ new CookieTestCase("Set-Cookie2",
+ "Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\";Path=\"/acme\"",
+ "$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"" +
+ "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"",
+ "/acme/pickitem"
+ ),
+ new CookieTestCase("Set-Cookie2",
+ "Shipping=\"FedEx\"; Version=\"1\"; Path=\"/acme\"",
+ "$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"" +
+ "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"" +
+ "; Shipping=\"FedEx\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"",
+ "/acme/shipping"
+ )
+ };
+
+ assert count == testCount;
+ }
+}