コード例 #1
0
  @Test
  public void test204() throws Exception {
    ByteBuffer header = BufferUtils.allocate(8096);
    ByteBuffer content = BufferUtils.toBuffer("0123456789");

    HttpGenerator gen = new HttpGenerator();

    MetaData.Response info =
        new MetaData.Response(HttpVersion.HTTP_1_1, 204, "Foo", new HttpFields(), 10);
    info.getFields().add("Content-Type", "test/data");
    info.getFields().add("Last-Modified", DateGenerator.__01Jan1970);

    HttpGenerator.Result result = gen.generateResponse(info, header, null, content, true);

    assertEquals(gen.isNoContent(), true);
    assertEquals(HttpGenerator.Result.FLUSH, result);
    assertEquals(HttpGenerator.State.COMPLETING, gen.getState());
    String responseheaders = BufferUtils.toString(header);
    BufferUtils.clear(header);

    result = gen.generateResponse(null, null, null, content, false);
    assertEquals(HttpGenerator.Result.DONE, result);
    assertEquals(HttpGenerator.State.END, gen.getState());

    assertThat(responseheaders, containsString("HTTP/1.1 204 Foo"));
    assertThat(responseheaders, containsString("Last-Modified: Thu, 01 Jan 1970 00:00:00 GMT"));
    assertThat(responseheaders, not(containsString("Content-Length: 10")));

    // Note: the HttpConnection.process() method is responsible for actually
    // excluding the content from the response based on
    // generator.isNoContent()==true
  }
コード例 #2
0
  @Test
  public void testResponseWithChunkedContent() throws Exception {
    ByteBuffer header = BufferUtils.allocate(4096);
    ByteBuffer chunk = BufferUtils.allocate(HttpGenerator.CHUNK_SIZE);
    ByteBuffer content0 = BufferUtils.toBuffer("Hello World! ");
    ByteBuffer content1 = BufferUtils.toBuffer("The quick brown fox jumped over the lazy dog. ");
    HttpGenerator gen = new HttpGenerator();

    HttpGenerator.Result result = gen.generateResponse(null, null, null, content0, false);
    assertEquals(HttpGenerator.Result.NEED_INFO, result);
    assertEquals(HttpGenerator.State.START, gen.getState());

    MetaData.Response info =
        new MetaData.Response(HttpVersion.HTTP_1_1, 200, null, new HttpFields(), -1);
    info.getFields().add("Last-Modified", DateGenerator.__01Jan1970);
    result = gen.generateResponse(info, null, null, content0, false);
    assertEquals(HttpGenerator.Result.NEED_HEADER, result);
    assertEquals(HttpGenerator.State.START, gen.getState());

    result = gen.generateResponse(info, header, null, content0, false);
    assertEquals(HttpGenerator.Result.FLUSH, result);
    assertEquals(HttpGenerator.State.COMMITTED, gen.getState());

    String out = BufferUtils.toString(header);
    BufferUtils.clear(header);
    out += BufferUtils.toString(content0);
    BufferUtils.clear(content0);

    result = gen.generateResponse(null, null, chunk, content1, false);
    assertEquals(HttpGenerator.Result.FLUSH, result);
    assertEquals(HttpGenerator.State.COMMITTED, gen.getState());
    out += BufferUtils.toString(chunk);
    BufferUtils.clear(chunk);
    out += BufferUtils.toString(content1);
    BufferUtils.clear(content1);

    result = gen.generateResponse(null, null, chunk, null, true);
    assertEquals(HttpGenerator.Result.CONTINUE, result);
    assertEquals(HttpGenerator.State.COMPLETING, gen.getState());

    result = gen.generateResponse(null, null, chunk, null, true);
    assertEquals(HttpGenerator.Result.FLUSH, result);
    assertEquals(HttpGenerator.State.COMPLETING, gen.getState());
    out += BufferUtils.toString(chunk);
    BufferUtils.clear(chunk);

    result = gen.generateResponse(null, null, chunk, null, true);
    assertEquals(HttpGenerator.Result.DONE, result);
    assertEquals(HttpGenerator.State.END, gen.getState());

    assertThat(out, containsString("HTTP/1.1 200 OK"));
    assertThat(out, containsString("Last-Modified: Thu, 01 Jan 1970 00:00:00 GMT"));
    assertThat(out, not(containsString("Content-Length")));
    assertThat(out, containsString("Transfer-Encoding: chunked"));
    assertThat(out, containsString("\r\n\r\nD\r\n"));
    assertThat(out, containsString("\r\nHello World! \r\n"));
    assertThat(out, containsString("\r\n2E\r\n"));
    assertThat(out, containsString("\r\nThe quick brown fox jumped over the lazy dog. \r\n"));
    assertThat(out, containsString("\r\n0\r\n"));
  }
コード例 #3
0
  @Test
  public void testResponseUpgrade() throws Exception {
    ByteBuffer header = BufferUtils.allocate(8096);

    HttpGenerator gen = new HttpGenerator();

    HttpGenerator.Result result = gen.generateResponse(null, null, null, null, true);
    assertEquals(HttpGenerator.Result.NEED_INFO, result);
    assertEquals(HttpGenerator.State.START, gen.getState());

    MetaData.Response info =
        new MetaData.Response(HttpVersion.HTTP_1_1, 101, null, new HttpFields(), -1);
    info.getFields().add("Upgrade", "WebSocket");
    info.getFields().add("Connection", "Upgrade");
    info.getFields().add("Sec-WebSocket-Accept", "123456789==");

    result = gen.generateResponse(info, header, null, null, true);
    assertEquals(HttpGenerator.Result.FLUSH, result);
    assertEquals(HttpGenerator.State.COMPLETING, gen.getState());
    String head = BufferUtils.toString(header);
    BufferUtils.clear(header);

    result = gen.generateResponse(info, null, null, null, false);
    assertEquals(HttpGenerator.Result.DONE, result);
    assertEquals(HttpGenerator.State.END, gen.getState());

    assertEquals(0, gen.getContentPrepared());

    assertThat(head, startsWith("HTTP/1.1 101 Switching Protocols"));
    assertThat(head, containsString("Upgrade: WebSocket\r\n"));
    assertThat(head, containsString("Connection: Upgrade\r\n"));
  }
コード例 #4
0
  @Test
  public void testResponseNoContent() throws Exception {
    ByteBuffer header = BufferUtils.allocate(8096);

    HttpGenerator gen = new HttpGenerator();

    HttpGenerator.Result result = gen.generateResponse(null, null, null, null, true);
    assertEquals(HttpGenerator.Result.NEED_INFO, result);
    assertEquals(HttpGenerator.State.START, gen.getState());

    MetaData.Response info =
        new MetaData.Response(HttpVersion.HTTP_1_1, 200, null, new HttpFields(), -1);
    info.getFields().add("Last-Modified", DateGenerator.__01Jan1970);

    result = gen.generateResponse(info, null, null, null, true);
    assertEquals(HttpGenerator.Result.NEED_HEADER, result);

    result = gen.generateResponse(info, header, null, null, true);
    assertEquals(HttpGenerator.Result.FLUSH, result);
    assertEquals(HttpGenerator.State.COMPLETING, gen.getState());
    String head = BufferUtils.toString(header);
    BufferUtils.clear(header);

    result = gen.generateResponse(null, null, null, null, false);
    assertEquals(HttpGenerator.Result.DONE, result);
    assertEquals(HttpGenerator.State.END, gen.getState());

    assertEquals(0, gen.getContentPrepared());
    assertThat(head, containsString("HTTP/1.1 200 OK"));
    assertThat(head, containsString("Last-Modified: Thu, 01 Jan 1970 00:00:00 GMT"));
    assertThat(head, containsString("Content-Length: 0"));
  }
コード例 #5
0
  @Test
  public void test101SwitchingProtocolsResponse() throws Exception {
    MetaData.Response switchingProtocolsResponse =
        new MetaData.Response(HttpVersion.HTTP_1_1, 101, new HttpFields());
    switchingProtocolsResponse.getFields().put(HttpHeader.CONNECTION, HttpHeaderValue.UPGRADE);
    switchingProtocolsResponse.getFields().put(HttpHeader.UPGRADE, "h2c");

    ByteBuffer header = BufferUtils.allocate(4096);
    HttpGenerator gen = new HttpGenerator();
    HttpGenerator.Result result =
        gen.generateResponse(switchingProtocolsResponse, header, null, null, true);
    System.out.println(result + "|" + gen.getState());
    assertEquals(HttpGenerator.Result.FLUSH, result);
    assertEquals(HttpGenerator.State.COMPLETING, gen.getState());

    result = gen.generateResponse(null, null, null, null, true);
    System.out.println(result + "|" + gen.getState());
    assertEquals(HttpGenerator.Result.DONE, result);
    assertEquals(HttpGenerator.State.END, gen.getState());
    System.out.println(BufferUtils.toString(header));
  }
コード例 #6
0
  @Test
  public void testComplexChars() throws Exception {
    ByteBuffer header = BufferUtils.allocate(8096);
    ByteBuffer content = BufferUtils.toBuffer("0123456789");

    HttpGenerator gen = new HttpGenerator();

    HttpGenerator.Result result = gen.generateResponse(null, null, null, content, true);
    assertEquals(HttpGenerator.Result.NEED_INFO, result);
    assertEquals(HttpGenerator.State.START, gen.getState());

    MetaData.Response info =
        new MetaData.Response(HttpVersion.HTTP_1_1, 200, null, new HttpFields(), 10);
    info.getFields().add("Content-Type", "test/data;\r\nextra=value");
    info.getFields().add("Last-Modified", DateGenerator.__01Jan1970);

    result = gen.generateResponse(info, null, null, content, true);
    assertEquals(HttpGenerator.Result.NEED_HEADER, result);

    result = gen.generateResponse(info, header, null, content, true);
    assertEquals(HttpGenerator.Result.FLUSH, result);
    assertEquals(HttpGenerator.State.COMPLETING, gen.getState());
    String response = BufferUtils.toString(header);
    BufferUtils.clear(header);
    response += BufferUtils.toString(content);
    BufferUtils.clear(content);

    result = gen.generateResponse(null, null, null, content, false);
    assertEquals(HttpGenerator.Result.DONE, result);
    assertEquals(HttpGenerator.State.END, gen.getState());

    assertEquals(10, gen.getContentPrepared());

    assertThat(response, containsString("HTTP/1.1 200 OK"));
    assertThat(response, containsString("Last-Modified: Thu, 01 Jan 1970 00:00:00 GMT"));
    assertThat(response, containsString("Content-Type: test/data;  extra=value"));
    assertThat(response, containsString("Content-Length: 10"));
    assertThat(response, containsString("\r\n0123456789"));
  }