def test_process_with_body_frame_too_big(self):
obj = channel.ContentFrameDispatcher()
method_frame = frame.Method(1, spec.Basic.Deliver())
obj.process(method_frame)
header_frame = frame.Header(1, 6, spec.BasicProperties)
obj.process(header_frame)
body_frame = frame.Body(1, 'abcd1234')
self.assertRaises(exceptions.BodyTooLongError, obj.process, body_frame)
def test_process_with_body_frame_six_bytes(self):
obj = channel.ContentFrameDispatcher(False)
method_frame = frame.Method(1, spec.Basic.Deliver())
obj.process(method_frame)
header_frame = frame.Header(1, 10, spec.BasicProperties)
obj.process(header_frame)
body_frame = frame.Body(1, 'abc123')
obj.process(body_frame)
self.assertEqual(obj._seen_so_far, 6)
def test_process_with_full_message(self):
obj = channel.ContentFrameDispatcher(False)
method_frame = frame.Method(1, spec.Basic.Deliver())
obj.process(method_frame)
header_frame = frame.Header(1, 6, spec.BasicProperties)
obj.process(header_frame)
body_frame = frame.Body(1, 'abc123')
response = obj.process(body_frame)
self.assertEqual(response, (method_frame, header_frame, 'abc123'))
def test_process_with_body_frame_partial(self):
obj = channel.ContentFrameDispatcher(False)
value = frame.Header(1, 100, spec.BasicProperties)
obj.process(value)
value = frame.Method(1, spec.Basic.Deliver())
obj.process(value)
value = frame.Body(1, 'abc123')
obj.process(value)
self.assertEqual(obj._body_fragments, [value.fragment])
def test_ascii_body_instance(self):
obj = channel.ContentFrameDispatcher(False)
method_frame = frame.Method(1, spec.Basic.Deliver())
obj.process(method_frame)
header_frame = frame.Header(1, 11, spec.BasicProperties)
obj.process(header_frame)
body_frame = frame.Body(1, 'foo-bar-baz')
method_frame, header_frame, body_value = obj.process(body_frame)
self.assertIsInstance(body_value, str)
def test_utf8_body_instance(self):
obj = channel.ContentFrameDispatcher(False)
method_frame = frame.Method(1, spec.Basic.Deliver())
obj.process(method_frame)
header_frame = frame.Header(1, 14, spec.BasicProperties)
obj.process(header_frame)
body_frame = frame.Body(1, 'utf8_value=\xe2\x9c\x93')
method_frame, header_frame, body_value = obj.process(body_frame)
self.assertIsInstance(body_value, unicode)
def test_reset_method_frame(self):
obj = channel.ContentFrameDispatcher(False)
method_frame = frame.Method(1, spec.Basic.Deliver())
obj.process(method_frame)
header_frame = frame.Header(1, 10, spec.BasicProperties)
obj.process(header_frame)
body_frame = frame.Body(1, 'abc123')
obj.process(body_frame)
obj._reset()
self.assertEqual(obj._method_frame, None)
def test_binary_non_unicode_value(self):
expectation =('a', 0.8)
obj = channel.ContentFrameDispatcher(False)
method_frame = frame.Method(1, spec.Basic.Deliver())
obj.process(method_frame)
header_frame = frame.Header(1, 20, spec.BasicProperties)
obj.process(header_frame)
body_frame = frame.Body(1, marshal.dumps(expectation))
method_frame, header_frame, body_value = obj.process(body_frame)
self.assertEqual(marshal.loads(body_value), expectation)
def test_utf8_body_value(self):
expectation = u'utf8_value=✓'
obj = channel.ContentFrameDispatcher(False)
method_frame = frame.Method(1, spec.Basic.Deliver())
obj.process(method_frame)
header_frame = frame.Header(1, 14, spec.BasicProperties)
obj.process(header_frame)
body_frame = frame.Body(1, 'utf8_value=\xe2\x9c\x93')
method_frame, header_frame, body_value = obj.process(body_frame)
self.assertEqual(body_value, expectation)
def test_ascii_body_value(self):
expectation = 'foo-bar-baz'
self.obj = channel.ContentFrameDispatcher()
method_frame = frame.Method(1, spec.Basic.Deliver())
self.obj.process(method_frame)
header_frame = frame.Header(1, 11, spec.BasicProperties)
self.obj.process(header_frame)
body_frame = frame.Body(1, 'foo-bar-baz')
method_frame, header_frame, body_value = self.obj.process(body_frame)
self.assertEqual(body_value, expectation)
self.assertIsInstance(body_value, str)
def test_process_with_unexpected_frame_type(self):
obj = channel.ContentFrameDispatcher(False)
value = frame.Method(1, spec.Basic.Qos())
self.assertRaises(exceptions.UnexpectedFrameError,
obj.process, value)
def test_process_with_content_header(self):
value = frame.Header(1, 100, spec.BasicProperties)
obj = channel.ContentFrameDispatcher(False)
obj.process(value)
self.assertEqual(obj._header_frame, value)
def test_init_body_fragments(self):
obj = channel.ContentFrameDispatcher(False)
self.assertEqual(obj._body_fragments, list())
def test_init_seen_so_far(self):
obj = channel.ContentFrameDispatcher(False)
self.assertEqual(obj._seen_so_far, 0)
def test_init_header_frame(self):
obj = channel.ContentFrameDispatcher(False)
self.assertEqual(obj._header_frame, None)
def test_process_with_basic_deliver(self):
value = frame.Method(1, spec.Basic.Deliver())
obj = channel.ContentFrameDispatcher(False)
obj.process(value)
self.assertEqual(obj._method_frame, value)
def setUp(self):
self.obj = channel.ContentFrameDispatcher()
def test_init_method_frame(self):
obj = channel.ContentFrameDispatcher()
self.assertEqual(obj._method_frame, None)
