summaryrefslogtreecommitdiffstats
path: root/simple/simple-http/src/test/java/org/simpleframework/http/core/AccumulatorTest.java
blob: 70db4a30386a9f062e63e6769caa7af7304a2e25 (plain)
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package org.simpleframework.http.core;

import java.io.IOException;

import junit.framework.TestCase;

public class AccumulatorTest extends TestCase {
   
   public void testAccumulator() throws IOException {
      MockChannel channel = new MockChannel(null);
      MockObserver monitor = new MockObserver();
      MockRequest request = new MockRequest();
      MockResponse response = new MockResponse();
      Conversation support = new Conversation(request, response);
      ResponseBuffer buffer = new ResponseBuffer(monitor, response, support, channel);
      
      byte[] content = { 'T', 'E', 'S', 'T' }; 
      
      // Start a HTTP/1.1 conversation
      request.setMajor(1);
      request.setMinor(1);
      
      // Write to a zero capacity buffer
      buffer.expand(0);
      buffer.write(content, 0, content.length);
      
      assertEquals(response.getValue("Connection"), "keep-alive");
      assertEquals(response.getValue("Transfer-Encoding"), "chunked");
      assertEquals(response.getValue("Content-Length"), null);
      assertEquals(response.getContentLength(), -1);
      assertTrue(response.isCommitted());
      
      channel = new MockChannel(null);
      monitor = new MockObserver();
      request = new MockRequest();
      response = new MockResponse();
      support = new Conversation(request, response);
      buffer = new ResponseBuffer(monitor, response, support, channel);
      
      // Start a HTTP/1.0 conversation
      request.setMajor(1);
      request.setMinor(0);
      
      // Write to a zero capacity buffer
      buffer.expand(0);
      buffer.write(content, 0, content.length);
      
      assertEquals(response.getValue("Connection"), "close");
      assertEquals(response.getValue("Transfer-Encoding"), null);
      assertEquals(response.getValue("Content-Length"), null);
      assertEquals(response.getContentLength(), -1);
      assertTrue(response.isCommitted());
      
      channel = new MockChannel(null);
      monitor = new MockObserver();
      request = new MockRequest();
      response = new MockResponse();
      support = new Conversation(request, response);
      buffer = new ResponseBuffer(monitor, response, support, channel);
      
      // Start a HTTP/1.1 conversation
      request.setMajor(1);
      request.setMinor(1);
      
      // Write to a large capacity buffer
      buffer.expand(1024);
      buffer.write(content, 0, content.length);
      
      assertEquals(response.getValue("Connection"), null);
      assertEquals(response.getValue("Transfer-Encoding"), null);
      assertEquals(response.getValue("Content-Length"), null);
      assertEquals(response.getContentLength(), -1);
      assertFalse(response.isCommitted());
      assertFalse(monitor.isReady());
      assertFalse(monitor.isClose());
      assertFalse(monitor.isError());
      
      // Flush the buffer
      buffer.close();
      
      assertEquals(response.getValue("Connection"), "keep-alive");
      assertEquals(response.getValue("Transfer-Encoding"), null);
      assertEquals(response.getValue("Content-Length"), "4");
      assertEquals(response.getContentLength(), 4);
      assertTrue(response.isCommitted());
      assertTrue(monitor.isReady());
      assertFalse(monitor.isClose());
      assertFalse(monitor.isError());

      boolean catchOverflow = false;
      
      try {
         buffer.write(content, 0, content.length);
      } catch(Exception e) {
         catchOverflow = true;
      }
      assertTrue(catchOverflow);
   }
}