/*
* JBoss, Home of Professional Open Source.
* Copyright 2014 Red Hat, Inc., and individual contributors
* as indicated by the @author tags.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.undertow.client.http;
import io.undertow.testutils.category.UnitTest;
import io.undertow.util.HttpString;
import io.undertow.util.Protocols;
import io.undertow.util.StatusCodes;
import org.junit.Assert;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import java.nio.ByteBuffer;
/**
* Tests that the parser can resume when it is given partial input
*
* @author Stuart Douglas
*/
@Category(UnitTest.class)
public class ResponseParserResumeTestCase {
public static final String DATA = "HTTP/1.1 200 OK\r\nHost: www.somehost.net\r\nOtherHeader: some\r\n value\r\nHostee:another\r\nAccept-garbage: a\r\n\r\ntttt";
@Test
public void testMethodSplit() {
byte[] in = DATA.getBytes();
for (int i = 0; i < in.length - 4; ++i) {
try {
testResume(i, in);
} catch (Throwable e) {
throw new RuntimeException("Test failed at split " + i, e);
}
}
}
@Test
public void testOneCharacterAtATime() {
byte[] in = DATA.getBytes();
final ResponseParseState context = new ResponseParseState();
HttpResponseBuilder result = new HttpResponseBuilder();
ByteBuffer buffer = ByteBuffer.wrap(in);
buffer.limit(1);
while (context.state != ResponseParseState.PARSE_COMPLETE) {
HttpResponseParser.INSTANCE.handle(buffer, context, result);
buffer.limit(buffer.limit() + 1);
}
runAssertions(result, context);
}
private void testResume(final int split, byte[] in) {
final ResponseParseState context = new ResponseParseState();
HttpResponseBuilder result = new HttpResponseBuilder();
ByteBuffer buffer = ByteBuffer.wrap(in);
buffer.limit(split);
HttpResponseParser.INSTANCE.handle(buffer, context, result);
Assert.assertEquals(0, buffer.remaining());
buffer.limit(buffer.capacity());
HttpResponseParser.INSTANCE.handle(buffer,context, result);
runAssertions(result, context);
Assert.assertEquals(4, buffer.remaining());
}
private void runAssertions(final HttpResponseBuilder result, final ResponseParseState context) {
Assert.assertEquals(StatusCodes.OK, result.getStatusCode());
Assert.assertEquals("OK", result.getReasonPhrase());
Assert.assertSame(Protocols.HTTP_1_1, result.getProtocol());
Assert.assertEquals("www.somehost.net", result.getResponseHeaders().getFirst(new HttpString("Host")));
Assert.assertEquals("some value", result.getResponseHeaders().getFirst(new HttpString("OtherHeader")));
Assert.assertEquals("another", result.getResponseHeaders().getFirst(new HttpString("Hostee")));
Assert.assertEquals("a", result.getResponseHeaders().getFirst(new HttpString("Accept-garbage")));
Assert.assertEquals(4, result.getResponseHeaders().getHeaderNames().size());
Assert.assertEquals(ResponseParseState.PARSE_COMPLETE, context.state);
}
}