1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28 package org.apache.http.nio.protocol;
29
30 import org.apache.http.HttpEntity;
31 import org.apache.http.HttpResponse;
32 import org.apache.http.HttpVersion;
33 import org.apache.http.entity.StringEntity;
34 import org.apache.http.message.BasicHttpResponse;
35 import org.apache.http.protocol.HTTP;
36 import org.junit.After;
37 import org.junit.Assert;
38 import org.junit.Before;
39 import org.junit.Test;
40
41 public class TestErrorResponseProducer {
42
43 private ErrorResponseProducer erp;
44 private HttpResponse response;
45 private HttpEntity entity;
46
47 @Before
48 public void setUp() throws Exception {
49 response = new BasicHttpResponse(HttpVersion.HTTP_1_1, 200, "OK");
50 entity = new StringEntity("stuff");
51 }
52
53 @After
54 public void tearDown() throws Exception {
55 }
56
57 @Test
58 public void testGenerateResponseKeepAlive() {
59 erp = new ErrorResponseProducer(response, entity, true);
60 final HttpResponse res = erp.generateResponse();
61
62 Assert.assertEquals(HTTP.CONN_KEEP_ALIVE, res.getFirstHeader(HTTP.CONN_DIRECTIVE).getValue());
63 Assert.assertEquals(entity, res.getEntity());
64 Assert.assertEquals(200, res.getStatusLine().getStatusCode());
65 }
66
67 @Test
68 public void testGenerateResponseClose() {
69 erp = new ErrorResponseProducer(response, entity, false);
70 final HttpResponse res = erp.generateResponse();
71
72 Assert.assertEquals(HTTP.CONN_CLOSE, res.getFirstHeader(HTTP.CONN_DIRECTIVE).getValue());
73 Assert.assertEquals(entity, res.getEntity());
74 Assert.assertEquals(200, res.getStatusLine().getStatusCode());
75 }
76
77 }