org.openqa.selenium.remote.http.HttpRequest Java Exaples

Source File: LocalNode.java From selenium with Apache License 2.0

6 votes

@Override
public HttpResponse executeWebDriverCommand(HttpRequest req) {
  // True enough to be good enough
  SessionId id = getSessionId(req.getUri()).map(SessionId::new)
    .orElseThrow(() -> new NoSuchSessionException("Cannot find session: " + req));

  SessionSlot slot = currentSessions.getIfPresent(id);
  if (slot == null) {
    throw new NoSuchSessionException("Cannot find session with id: " + id);
  }

  HttpResponse toReturn = slot.execute(req);
  if (req.getMethod() == DELETE && req.getUri().equals("/session/" + id)) {
    stop(id);
  }
  return toReturn;
}

Source File: AddToSessionMap.java From selenium with Apache License 2.0

6 votes

@Override
public HttpResponse execute(HttpRequest req) {
  try (Span span = newSpanAsChildOf(tracer, req, "sessions.add_session")) {
    HTTP_REQUEST.accept(span, req);

    Session session = json.toType(string(req), Session.class);
    Objects.requireNonNull(session, "Session to add must be set");

    SESSION_ID.accept(span, session.getId());
    CAPABILITIES.accept(span, session.getCapabilities());
    span.setAttribute("session.uri", session.getUri().toString());

    sessions.add(session);

    return new HttpResponse().setContent(asJson(ImmutableMap.of("value", true)));
  }
}

Source File: ProxyCdpIntoGrid.java From selenium with Apache License 2.0

6 votes

@Override
public Optional<Consumer<Message>> apply(String uri, Consumer<Message> downstream) {
  Objects.requireNonNull(uri);
  Objects.requireNonNull(downstream);

  Optional<SessionId> sessionId = HttpSessionId.getSessionId(uri).map(SessionId::new);
  if (!sessionId.isPresent()) {
    return Optional.empty();
  }

  try {
    Session session = sessions.get(sessionId.get());

    HttpClient client = clientFactory.createClient(ClientConfig.defaultConfig().baseUri(session.getUri()));
    WebSocket upstream = client.openSocket(new HttpRequest(GET, uri), new ForwardingListener(downstream));

    return Optional.of(upstream::send);

  } catch (NoSuchSessionException e) {
    LOG.info("Attempt to connect to non-existant session: " + uri);
    return Optional.empty();
  }
}

Source File: ListImages.java From selenium with Apache License 2.0

6 votes

public Set<Image> apply(Reference reference) {
  Require.nonNull("Reference to search for", reference);

  String familiarName = reference.getFamiliarName();
  Map<String, Object> filters = ImmutableMap.of("reference", ImmutableMap.of(familiarName, true));

  // https://docs.docker.com/engine/api/v1.40/#operation/ImageList
  HttpRequest req = new HttpRequest(GET, "/v1.40/images/json")
    .addHeader("Content-Length", "0")
    .addHeader("Content-Type", JSON_UTF_8)
    .addQueryParameter("filters", JSON.toJson(filters));

  HttpResponse response = DockerMessages.throwIfNecessary(
    client.execute(req),
  "Unable to list images for %s", reference);

  Set<ImageSummary> images =
    JSON.toType(string(response), SET_OF_IMAGE_SUMMARIES);

  return images.stream()
    .map(org.openqa.selenium.docker.Image::new)
    .collect(toImmutableSet());
}

Source File: ResourceHandler.java From selenium with Apache License 2.0

6 votes

private HttpResponse readDirectory(HttpRequest req, Resource resource) {
  if (!req.getUri().endsWith("/")) {
    String dest = UrlPath.relativeToContext(req, req.getUri() + "/");
    return new HttpResponse()
      .setStatus(HTTP_MOVED_TEMP)
      .addHeader("Location", dest);
  }

  String links = resource.list().stream()
    .map(res -> String.format("<li><a href=\"%s\">%s</a>", res.name(), res.name()))
    .sorted()
    .collect(Collectors.joining("\n", "<ul>\n", "</ul>\n"));

  String html = String.format(
    "<html><title>Listing of %s</title><body><h1>%s</h1>%s",
    resource.name(),
    resource.name(),
    links);

  return new HttpResponse()
    .addHeader("Content-Type", HTML_UTF_8.toString())
    .setContent(utf8String(html));
}

Source File: NoSessionHandler.java From selenium with Apache License 2.0

6 votes

@Override
public HttpResponse execute(HttpRequest req) throws UncheckedIOException {
  // We're not using ImmutableMap for the outer map because it disallows null values.
  Map<String, Object> responseMap = new HashMap<>();
  responseMap.put("sessionId", sessionId.toString());
  responseMap.put("status", NO_SUCH_SESSION);
  responseMap.put("value", ImmutableMap.of(
      "error", "invalid session id",
      "message", String.format("No active session with ID %s", sessionId),
      "stacktrace", ""));
  responseMap = Collections.unmodifiableMap(responseMap);

  byte[] payload = json.toJson(responseMap).getBytes(UTF_8);

  return new HttpResponse().setStatus(HTTP_NOT_FOUND)
    .setHeader("Content-Type", JSON_UTF_8.toString())
    .setContent(bytes(payload));
}

Source File: HttpHandlerServletTest.java From selenium with Apache License 2.0

6 votes

@Test
public void shouldReturnValueFromHandlerIfUrlMatches() throws IOException {
  String cheerfulGreeting = "Hello, world!";

  HttpHandlerServlet servlet = new HttpHandlerServlet(
      Route.matching(req -> true)
          .to(() -> req -> new HttpResponse().setContent(utf8String(cheerfulGreeting))));

  HttpServletRequest request = requestConverter.apply(new HttpRequest(GET, "/hello-world"));
  FakeHttpServletResponse response = new FakeHttpServletResponse();

  servlet.service(request, response);

  assertThat(response.getStatus()).isEqualTo(HTTP_OK);
  assertThat(response.getBody()).isEqualTo(cheerfulGreeting);
}

Source File: EndToEndTest.java From selenium with Apache License 2.0

6 votes

@Test
public void shouldAllowPassthroughForJWPMode() {
  HttpRequest request = new HttpRequest(POST, "/session");
  request.setContent(asJson(
      ImmutableMap.of(
          "desiredCapabilities", ImmutableMap.of(
              "browserName", "cheese"))));

  HttpClient client = clientFactory.createClient(server.getUrl());
  HttpResponse response = client.execute(request);

  assertEquals(200, response.getStatus());

  Map<String, Object> topLevel = json.toType(string(response), MAP_TYPE);

  // There should be a numeric status field
  assertEquals(topLevel.toString(), 0L, topLevel.get("status"));
  // The session id
  assertTrue(string(request), topLevel.containsKey("sessionId"));

  // And the value should be the capabilities.
  Map<?, ?> value = (Map<?, ?>) topLevel.get("value");
  assertEquals(string(request), "cheese", value.get("browserName"));
}

Source File: WebSocketServingTest.java From selenium with Apache License 2.0

6 votes

@Test
public void shouldStillBeAbleToServeHttpTraffic() {
  server = new NettyServer(
    defaultOptions(),
    req -> new HttpResponse().setContent(utf8String("Brie!")),
    (uri, sink) -> {
      if ("/foo".equals(uri)) {
        return Optional.of(msg -> sink.accept(new TextMessage("Peas!")));
      }
      return Optional.empty();
    }).start();

  HttpClient client = HttpClient.Factory.createDefault().createClient(server.getUrl());
  HttpResponse res = client.execute(new HttpRequest(GET, "/cheese"));

  assertThat(Contents.string(res)).isEqualTo("Brie!");
}

Source File: PageHandler.java From selenium with Apache License 2.0

6 votes

@Override
public HttpResponse execute(HttpRequest req) throws UncheckedIOException {

  int lastIndex = req.getUri().lastIndexOf('/');
  String pageNumber =
      (lastIndex == -1 ? "Unknown" : req.getUri().substring(lastIndex + 1));
  String body = String.format("<html><head><title>Page%s</title></head>" +
                             "<body>Page number <span id=\"pageNumber\">%s</span>" +
                             "<p><a href=\"../xhtmlTest.html\" target=\"_top\">top</a>" +
                             "</body></html>",
                             pageNumber, pageNumber);

  return new HttpResponse()
    .setHeader("Content-Type", "text/html")
    .setContent(utf8String(body));
}

Source File: GetAllSessions.java From selenium with Apache License 2.0

6 votes

@Override
public HttpResponse execute(HttpRequest req) throws UncheckedIOException {
  List<Map<String, Object>> value = new ArrayList<>();

  allSessions.getAllSessions().forEach(s -> value.add(
      ImmutableMap.of("id", s.getId().toString(), "capabilities", s.getCapabilities())));

  Map<String, Object> payloadObj = ImmutableMap.of(
      "status", SUCCESS,
      "value", value);

  // Write out a minimal W3C status response.
  byte[] payload = json.toJson(payloadObj).getBytes(UTF_8);

  return new HttpResponse().setStatus(HTTP_OK)
    .setHeader("Content-Type", JSON_UTF_8.toString())
    .setContent(bytes(payload));
}

Source File: JsonHttpCommandCodecTest.java From selenium with Apache License 2.0

6 votes

@Test
public void ignoresNullSessionIdInSessionBody() {
  Map<String, Object> map = new HashMap<>();
  map.put("sessionId", null);
  map.put("fruit", "apple");
  map.put("color", "red");
  map.put("size", "large");
  String data = new Json().toJson(map);
  HttpRequest request = new HttpRequest(POST, "/fruit/apple/size/large");
  request.setContent(utf8String(data));
  codec.defineCommand("pick", POST, "/fruit/:fruit/size/:size");

  Command decoded = codec.decode(request);
  assertThat(decoded.getSessionId()).isNull();
  assertThat(decoded.getParameters()).isEqualTo((Map<String, String>) ImmutableMap.of(
      "fruit", "apple", "size", "large", "color", "red"));
}

Source File: HttpClientTestBase.java From selenium with Apache License 2.0

6 votes

private HttpResponse executeWithinServer(HttpRequest request, HttpServlet servlet)
    throws Exception {
  Server server = new Server(PortProber.findFreePort());
  ServletContextHandler handler = new ServletContextHandler();
  handler.setContextPath("");
  ServletHolder holder = new ServletHolder(servlet);
  handler.addServlet(holder, "/*");

  server.setHandler(handler);

  server.start();
  try {
    HttpClient client = createFactory().createClient(fromUri(server.getURI()));
    return client.execute(request);
  } finally {
    server.stop();
  }
}

Source File: HttpClientTestBase.java From selenium with Apache License 2.0

6 votes

private HttpResponse getQueryParameterResponse(HttpRequest request) throws Exception {
  return executeWithinServer(
      request,
      new HttpServlet() {
        @Override
        protected void doGet(HttpServletRequest req, HttpServletResponse resp)
            throws IOException {
          try (Writer writer = resp.getWriter()) {
            JsonOutput json = new Json().newOutput(writer);
            json.beginObject();
            req.getParameterMap()
                .forEach((key, value) -> {
                  json.name(key);
                  json.beginArray();
                  Stream.of(value).forEach(json::write);
                  json.endArray();
                });
            json.endObject();
          }
        }
      });
}

Source File: JettyServerTest.java From selenium with Apache License 2.0

6 votes

@Test
public void exceptionsThrownByHandlersAreConvertedToAProperPayload() {
  Server<?> server = new JettyServer(
    emptyOptions,
    req -> {
      throw new UnableToSetCookieException("Yowza");
    });

  server.start();
  URL url = server.getUrl();
  HttpClient client = HttpClient.Factory.createDefault().createClient(url);
  HttpResponse response = client.execute(new HttpRequest(GET, "/status"));

  assertThat(response.getStatus()).isEqualTo(HTTP_INTERNAL_ERROR);

  Throwable thrown = null;
  try {
    thrown = ErrorCodec.createDefault().decode(new Json().toType(string(response), MAP_TYPE));
  } catch (IllegalArgumentException ignored) {
    fail("Apparently the command succeeded" + string(response));
  }

  assertThat(thrown).isInstanceOf(UnableToSetCookieException.class);
  assertThat(thrown.getMessage()).startsWith("Yowza");
}

Source File: EndToEndTest.java From selenium with Apache License 2.0

6 votes

@Test
public void shouldAllowPassthroughForW3CMode() {
  HttpRequest request = new HttpRequest(POST, "/session");
  request.setContent(asJson(
      ImmutableMap.of(
          "capabilities", ImmutableMap.of(
              "alwaysMatch", ImmutableMap.of("browserName", "cheese")))));

  HttpClient client = clientFactory.createClient(server.getUrl());
  HttpResponse response = client.execute(request);

  assertEquals(200, response.getStatus());

  Map<String, Object> topLevel = json.toType(string(response), MAP_TYPE);

  // There should not be a numeric status field
  assertFalse(string(request), topLevel.containsKey("status"));

  // And the value should have all the good stuff in it: the session id and the capabilities
  Map<?, ?> value = (Map<?, ?>) topLevel.get("value");
  assertThat(value.get("sessionId")).isInstanceOf(String.class);

  Map<?, ?> caps = (Map<?, ?>) value.get("capabilities");
  assertEquals("cheese", caps.get("browserName"));
}

Source File: JsonHttpCommandCodecTest.java From selenium with Apache License 2.0

5 votes

@Test
public void decodingUsesUtf8IfNoEncodingSpecified() {
  codec.defineCommand("num", POST, "/one");

  byte[] data = "{\"char\":\"水\"}".getBytes(UTF_8);
  HttpRequest request = new HttpRequest(POST, "/one");
  request.setHeader(CONTENT_TYPE, JSON_UTF_8.withoutParameters().toString());
  request.setHeader(CONTENT_LENGTH, String.valueOf(data.length));
  request.setContent(bytes(data));

  Command command = codec.decode(request);
  assertThat((String) command.getParameters().get("char")).isEqualTo("水");
}

Source File: JsonHttpCommandCodecTest.java From selenium with Apache License 2.0

5 votes

@Test
public void treatsEmptyPathAsRoot_recognizedCommand() {
  codec.defineCommand("num", POST, "/");

  byte[] data = "{\"char\":\"水\"}".getBytes(UTF_8);
  HttpRequest request = new HttpRequest(POST, "");
  request.setHeader(CONTENT_TYPE, JSON_UTF_8.withoutParameters().toString());
  request.setHeader(CONTENT_LENGTH, String.valueOf(data.length));
  request.setContent(bytes(data));

  Command command = codec.decode(request);
  assertThat(command.getName()).isEqualTo("num");
}

Source File: JsonHttpCommandCodecTest.java From selenium with Apache License 2.0

5 votes

@Test
public void canExtractSessionIdFromPathParameters() {
  HttpRequest request = new HttpRequest(GET, "/foo/bar/baz");
  codec.defineCommand("foo", GET, "/foo/:sessionId/baz");

  Command decoded = codec.decode(request);
  assertThat(decoded.getSessionId()).isEqualTo(new SessionId("bar"));
}

Source File: JsonHttpCommandCodecTest.java From selenium with Apache License 2.0

5 votes

@Test
public void canDecodeCommandWithNoParameters() {
  HttpRequest request = new HttpRequest(GET, "/foo/bar/baz");
  codec.defineCommand("foo", GET, "/foo/bar/baz");

  Command decoded = codec.decode(request);
  assertThat(decoded.getName()).isEqualTo("foo");
  assertThat(decoded.getSessionId()).isNull();
  assertThat(decoded.getParameters()).isEmpty();
}

Source File: RemoteNode.java From selenium with Apache License 2.0

5 votes

@Override
public NodeStatus getStatus() {
  HttpRequest req = new HttpRequest(GET, "/status");
  HttpTracing.inject(tracer, tracer.getCurrentContext(), req);

  HttpResponse res = client.execute(req);

  try (Reader reader = reader(res);
       JsonInput in = JSON.newInput(reader)) {
    in.beginObject();

    // Skip everything until we find "value"
    while (in.hasNext()) {
      if ("value".equals(in.nextName())) {
        in.beginObject();

        while (in.hasNext()) {
          if ("node".equals(in.nextName())) {
            return in.read(NodeStatus.class);
          } else {
            in.skipValue();
          }
        }

        in.endObject();
      } else {
        in.skipValue();
      }
    }
  } catch (IOException e) {
    throw new UncheckedIOException(e);
  }

  throw new IllegalStateException("Unable to read status");
}

Source File: JsonHttpCommandCodecTest.java From selenium with Apache License 2.0

5 votes

@Test
public void preventsCachingGetRequests() {
  codec.defineCommand("foo", GET, "/foo");
  HttpRequest request = codec.encode(new Command(null, "foo"));
  assertThat(request.getMethod()).isEqualTo(GET);
  assertThat(request.getHeader(CACHE_CONTROL)).isEqualTo("no-cache");
}

Source File: JsonHttpCommandCodecTest.java From selenium with Apache License 2.0

5 votes

@Test
public void encodingAPostWithUrlParameters() {
  codec.defineCommand("foo", POST, "/foo/:bar/baz");
  Command command = new Command(null, "foo", ImmutableMap.of("bar", "apples123"));

  String encoding = "{\n  \"bar\": \"apples123\"\n}";

  HttpRequest request = codec.encode(command);
  assertThat(request.getMethod()).isEqualTo(POST);
  assertThat(request.getHeader(CONTENT_TYPE)).isEqualTo(JSON_UTF_8.toString());
  assertThat(request.getHeader(CONTENT_LENGTH)).isEqualTo(String.valueOf(encoding.length()));
  assertThat(request.getUri()).isEqualTo("/foo/apples123/baz");
  assertThat(string(request)).isEqualTo(encoding);
}

Source File: NodeTest.java From selenium with Apache License 2.0

5 votes

@Test
public void canReturnStatus() {
  node.newSession(createSessionRequest(caps))
    .map(CreateSessionResponse::getSession)
    .orElseThrow(() -> new AssertionError("Cannot create session"));

  HttpRequest req = new HttpRequest(GET, "/status");
  HttpResponse res = node.execute(req);
  assertThat(res.getStatus()).isEqualTo(200);
  Map<String, Object> status = new Json().toType(string(res), MAP_TYPE);
  assertThat(status).containsOnlyKeys("value");
  assertThat(status).extracting("value").asInstanceOf(MAP)
      .containsEntry("ready", true)
      .containsEntry("message", "Ready")
      .containsKey("node");
  assertThat(status).extracting("value.node").asInstanceOf(MAP)
      .containsKey("id")
      .containsEntry("uri", "http://localhost:1234")
      .containsEntry("maxSessions", (long) 2)
      .containsKey("stereotypes")
      .containsKey("sessions");
  assertThat(status).extracting("value.node.stereotypes").asInstanceOf(LIST)
      .hasSize(1)
      .element(0).asInstanceOf(MAP)
      .containsEntry("capabilities", Collections.singletonMap("browserName", "cheese"))
      .containsEntry("count", (long) 3);
  assertThat(status).extracting("value.node.sessions").asInstanceOf(LIST)
      .hasSize(1)
      .element(0).asInstanceOf(MAP)
      .containsEntry("currentCapabilities", Collections.singletonMap("browserName", "cheese"))
      .containsEntry("stereotype", Collections.singletonMap("browserName", "cheese"))
      .containsKey("sessionId");
}

Source File: ErrorHandler.java From selenium with Apache License 2.0

5 votes

@Override
public HttpResponse execute(HttpRequest req) throws UncheckedIOException {
  return new HttpResponse()
    .setHeader("Cache-Control", "none")
    .setHeader("Content-Type", JSON_UTF_8.toString())
    .setStatus(errors.getHttpStatusCode(throwable))
    .setContent(asJson(errors.encode(throwable)));
}

Source File: RemoteDistributor.java From selenium with Apache License 2.0

5 votes

@Override
public RemoteDistributor add(Node node) {
  HttpRequest request = new HttpRequest(POST, "/se/grid/distributor/node");
  HttpTracing.inject(tracer, tracer.getCurrentContext(), request);
  request.setContent(asJson(node.getStatus()));

  HttpResponse response = client.execute(request);

  Values.get(response, Void.class);

  LOG.info(String.format("Added node %s.", node.getId()));

  return this;
}

Source File: RemoveFromSession.java From selenium with Apache License 2.0

5 votes

@Override
public HttpResponse execute(HttpRequest req) {
  try (Span span = newSpanAsChildOf(tracer, req, "sessions.remove_session")) {
    HTTP_REQUEST.accept(span, req);
    SESSION_ID.accept(span, id);

    sessions.remove(id);
    return new HttpResponse();
  }
}

Source File: ExceptionHandler.java From selenium with Apache License 2.0

5 votes

@Override
public HttpResponse execute(HttpRequest req) throws UncheckedIOException {
  int code = ERRORS.toStatusCode(exception);
  String status = ERRORS.toState(code);

  Map<String, Object> toSerialise = new HashMap<>();

  // W3C
  Map<String, Object> value = new HashMap<>();
  value.put("message", exception.getMessage());
  value.put("stacktrace", Throwables.getStackTraceAsString(exception));
  value.put("error", status);

  // JSON Wire Protocol
  toSerialise.put("status", code);
    value.put(
        "stackTrace",
        Stream.of(exception.getStackTrace())
            .map(ste -> {
              HashMap<String, Object> line = new HashMap<>();
              line.put("fileName", ste.getFileName());
              line.put("lineNumber", ste.getLineNumber());
              line.put("className", ste.getClassName());
              line.put("methodName", ste.getMethodName());
              return line;
            })
            .collect(ImmutableList.toImmutableList()));

  toSerialise.put("value", value);

  byte[] bytes = toJson.toJson(toSerialise).getBytes(UTF_8);

  return new HttpResponse()
    .setStatus(HTTP_INTERNAL_ERROR)
    .setHeader("Content-Type", JSON_UTF_8.toString())
    .setContent(bytes(bytes));
}

Source File: ProtocolConverter.java From selenium with Apache License 2.0

5 votes

private CommandCodec<HttpRequest> getCommandCodec(Dialect dialect) {
  switch (dialect) {
    case OSS:
      return new JsonHttpCommandCodec();

    case W3C:
      return new W3CHttpCommandCodec();

    default:
      throw new IllegalStateException("Unknown dialect: " + dialect);
  }
}

Source File: ReverseProxyHandlerTest.java From selenium with Apache License 2.0

5 votes

@Test
public void shouldForwardRequestsToEndPoint() {
  HttpHandler handler = new ReverseProxyHandler(tracer, factory.createClient(server.url));
  HttpRequest req = new HttpRequest(HttpMethod.GET, "/ok");
  req.addHeader("X-Cheese", "Cake");
  handler.execute(req);

  // HTTP headers are case insensitive. This is how the HttpUrlConnection likes to encode things
  assertEquals("Cake", server.lastRequest.getHeader("x-cheese"));
}

org.openqa.selenium.remote.http.HttpRequest Java Examples