def test_header(self):
res = grole.Response(None, 123, 'foo', {'foo': 'bar'}, 'bar')
writer = FakeWriter()
a_wait(res._write(writer))
for line in writer.data.split(b'\r\n'):
if line.startswith(b'bar'):
self.assertEqual(line, b'bar 123 foo')
elif line.startswith(b'Content-Type'):
self.assertEqual(line, b'Content-Type: text/plain')
elif line.startswith(b'Content-Length'):
self.assertEqual(line, b'Content-Length: 0')
elif line.startswith(b'foo'):
self.assertEqual(line, b'foo: bar')
elif line.startswith(b'Server'):
self.assertEqual(line, b'Server: grole/' + grole.__version__.encode())
else:
if line != b'':
self.fail('Extra data: ' + line.decode())
def test_data(self):
writer = FakeWriter()
a_wait(self.res._write(writer))
self.assertEqual(writer.data, b'{"foo": "bar"}')
def test_data(self):
res = grole.ResponseBody(b'foo', content_type='bar')
writer = FakeWriter()
a_wait(res._write(writer))
self.assertEqual(writer.data, b'foo')
def test_data(self):
writer = FakeWriter()
a_wait(self.res._write(writer))
self.assertEqual(writer.data, b'4\r\nfoo\n\r\n0\r\n\r\n')