[java] Add more Netty tests

jackojun · Jul 7, 2020 · b7ef910 · b7ef910
1 parent 2dd9bb9
commit b7ef910
Showing 1 changed file with 73 additions and 2 deletions.
diff --git a/java/server/test/org/openqa/selenium/netty/server/NettyServerTest.java b/java/server/test/org/openqa/selenium/netty/server/NettyServerTest.java
@@ -18,7 +18,11 @@
 package org.openqa.selenium.netty.server;
 
 import com.google.common.collect.ImmutableMap;
+
+import org.junit.Ignore;
 import org.junit.Test;
+import org.openqa.selenium.grid.config.CompoundConfig;
+import org.openqa.selenium.grid.config.Config;
 import org.openqa.selenium.grid.config.MapConfig;
 import org.openqa.selenium.grid.server.BaseServerOptions;
 import org.openqa.selenium.grid.server.Server;
@@ -27,10 +31,14 @@
 import org.openqa.selenium.remote.http.HttpRequest;
 import org.openqa.selenium.remote.http.HttpResponse;
 
+import java.net.URL;
 import java.util.concurrent.atomic.AtomicInteger;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
 import static org.openqa.selenium.remote.http.Contents.utf8String;
+import static org.openqa.selenium.remote.http.HttpMethod.DELETE;
 import static org.openqa.selenium.remote.http.HttpMethod.GET;
 
 public class NettyServerTest {
@@ -59,8 +67,8 @@ public void ensureMultipleCallsWorkAsExpected() {
       }
     ).start();
 
-      // TODO: avoid using netty for this
-      HttpClient client = HttpClient.Factory.createDefault().createClient(server.getUrl());
+    // TODO: avoid using netty for this
+    HttpClient client = HttpClient.Factory.createDefault().createClient(server.getUrl());
 
     HttpResponse res = client.execute(new HttpRequest(GET, "/does-not-matter"));
     outputHeaders(res);
@@ -71,6 +79,69 @@ public void ensureMultipleCallsWorkAsExpected() {
     assertThat(count.get()).isEqualTo(2);
   }
 
+  @Test
+  public void shouldDisableAllowOrigin() {
+    // TODO: Server setup
+    Server<?> server = new NettyServer(
+        new BaseServerOptions(
+            new MapConfig(
+                ImmutableMap.of("server", ImmutableMap.of("port", PortProber.findFreePort())))),
+        req -> {
+          return new HttpResponse().setContent(utf8String("Count is "));
+        }
+    ).start();
+
+    // TODO: Client setup
+    URL url = server.getUrl();
+    HttpClient client = HttpClient.Factory.createDefault().createClient(url);
+    HttpRequest request = new HttpRequest(DELETE, "/session");
+    String exampleUrl = "http://www.example.com";
+    request.setHeader("Origin", exampleUrl);
+    request.setHeader("Accept", "*/*");
+    HttpResponse response = client.execute(request);
+
+    // TODO: Assertion
+    assertEquals("Access-Control-Allow-Credentials should be null", null,
+                 response.getHeader("Access-Control-Allow-Credentials"));
+
+    assertEquals("Access-Control-Allow-Origin should be null",
+                 null,
+                 response.getHeader("Access-Control-Allow-Origin"));
+  }
+
+  @Test
+  @Ignore
+  public void shouldAllowCORS() {
+    // TODO: Server setup
+    Config cfg = new CompoundConfig(
+        new MapConfig(ImmutableMap.of("server", ImmutableMap.of("allow-cors", "true"))));
+    BaseServerOptions options = new BaseServerOptions(cfg);
+    assertTrue("Allow CORS should be enabled", options.getAllowCORS());
+
+    // TODO: Server setup
+    Server<?> server = new NettyServer(
+        options,
+        req -> new HttpResponse()
+    ).start();
+
+    // TODO: Client setup
+    URL url = server.getUrl();
+    HttpClient client = HttpClient.Factory.createDefault().createClient(url);
+    HttpRequest request = new HttpRequest(DELETE, "/session");
+    String exampleUrl = "http://www.example.com";
+    request.setHeader("Origin", exampleUrl);
+    request.setHeader("Accept", "*/*");
+    HttpResponse response = client.execute(request);
+
+    // TODO: Assertion
+    assertEquals("Access-Control-Allow-Credentials should be true", "true",
+                 response.getHeader("Access-Control-Allow-Credentials"));
+
+    assertEquals("Access-Control-Allow-Origin should be equal to origin in request header",
+                 exampleUrl,
+                 response.getHeader("Access-Control-Allow-Origin"));
+  }
+
   private void outputHeaders(HttpResponse res) {
     res.getHeaderNames().forEach(name ->
       res.getHeaders(name).forEach(value -> System.out.printf("%s -> %s\n", name, value)));