def test_init(self): expect(ProtocolClass.__init__).args('foo', a='b') klass = ExchangeClass.__new__(ExchangeClass) klass.__init__('foo', a='b') assert_equals({ 11: klass._recv_declare_ok, 21: klass._recv_delete_ok, }, klass.dispatch_map) assert_equals(deque(), klass._declare_cb) assert_equals(deque(), klass._delete_cb)
def test_init(self): expect(ProtocolClass.__init__).args('foo', a='b') klass = ExchangeClass.__new__(ExchangeClass) klass.__init__('foo', a='b') assert_equals( { 11: klass._recv_declare_ok, 21: klass._recv_delete_ok, }, klass.dispatch_map) assert_equals(deque(), klass._declare_cb) assert_equals(deque(), klass._delete_cb)
def setUp(self): super(ExchangeClassTest, self).setUp() ch = mock() ch.channel_id = 42 ch.logger = mock() self.klass = ExchangeClass(ch)
class ExchangeClassTest(Chai): def setUp(self): super(ExchangeClassTest, self).setUp() ch = mock() ch.channel_id = 42 ch.logger = mock() self.klass = ExchangeClass(ch) def test_init(self): expect(ProtocolClass.__init__).args('foo', a='b') klass = ExchangeClass.__new__(ExchangeClass) klass.__init__('foo', a='b') assert_equals( { 11: klass._recv_declare_ok, 21: klass._recv_delete_ok, }, klass.dispatch_map) assert_equals(deque(), klass._declare_cb) assert_equals(deque(), klass._delete_cb) def test_cleanup(self): self.klass._cleanup() assert_equals(None, self.klass._declare_cb) assert_equals(None, self.klass._delete_cb) assert_equals(None, self.klass._channel) assert_equals(None, self.klass.dispatch_map) def test_declare_default_args(self): w = mock() expect(self.klass.allow_nowait).returns(True) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args(self.klass.default_ticket).returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_shortstr).args('topic').returns(w) expect(w.write_bits).args(False, False, False, False, True).returns(w) expect(w.write_table).args({}) expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 10, w).returns('frame') expect(self.klass.send_frame).args('frame') stub(self.klass.channel.add_synchronous_cb) self.klass.declare('exchange', 'topic') assert_equals(deque(), self.klass._declare_cb) def test_declare_with_args(self): w = mock() stub(self.klass.allow_nowait) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args('t').returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_shortstr).args('topic').returns(w) expect(w.write_bits).args('p', 'd', False, False, False).returns(w) expect(w.write_table).args('table') expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 10, w).returns('frame') expect(self.klass.send_frame).args('frame') expect(self.klass.channel.add_synchronous_cb).args(self.klass._recv_declare_ok) self.klass.declare('exchange', 'topic', passive='p', durable='d', nowait=False, arguments='table', ticket='t') assert_equals(deque([None]), self.klass._declare_cb) def test_declare_with_cb(self): w = mock() expect(self.klass.allow_nowait).returns(True) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args('t').returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_shortstr).args('topic').returns(w) expect(w.write_bits).args('p', 'd', False, False, False).returns(w) expect(w.write_table).args('table') expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 10, w).returns('frame') expect(self.klass.send_frame).args('frame') expect(self.klass.channel.add_synchronous_cb).args(self.klass._recv_declare_ok) self.klass.declare('exchange', 'topic', passive='p', durable='d', nowait=True, arguments='table', ticket='t', cb='foo') assert_equals(deque(['foo']), self.klass._declare_cb) def test_recv_declare_ok_no_cb(self): self.klass._declare_cb = deque([None]) self.klass._recv_declare_ok('frame') assert_equals(deque(), self.klass._declare_cb) def test_recv_declare_ok_with_cb(self): cb = mock() self.klass._declare_cb = deque([cb]) expect(cb) self.klass._recv_declare_ok('frame') assert_equals(deque(), self.klass._declare_cb) def test_delete_default_args(self): w = mock() expect(self.klass.allow_nowait).returns(True) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args(self.klass.default_ticket).returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_bits).args(False, True) expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 20, w).returns('frame') expect(self.klass.send_frame).args('frame') stub(self.klass.channel.add_synchronous_cb) self.klass.delete('exchange') assert_equals(deque(), self.klass._delete_cb) def test_delete_with_args(self): w = mock() stub(self.klass.allow_nowait) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args('t').returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_bits).args('maybe', False) expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 20, w).returns('frame') expect(self.klass.send_frame).args('frame') expect(self.klass.channel.add_synchronous_cb).args(self.klass._recv_delete_ok) self.klass.delete('exchange', if_unused='maybe', nowait=False, ticket='t') assert_equals(deque([None]), self.klass._delete_cb) def test_delete_with_cb(self): w = mock() expect(self.klass.allow_nowait).returns(True) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args('t').returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_bits).args('maybe', False) expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 20, w).returns('frame') expect(self.klass.send_frame).args('frame') expect(self.klass.channel.add_synchronous_cb).args(self.klass._recv_delete_ok) self.klass.delete('exchange', if_unused='maybe', nowait=True, ticket='t', cb='foo') assert_equals(deque(['foo']), self.klass._delete_cb) def test_recv_delete_ok_no_cb(self): self.klass._delete_cb = deque([None]) self.klass._recv_delete_ok('frame') assert_equals(deque(), self.klass._delete_cb) def test_recv_delete_ok_with_cb(self): cb = mock() self.klass._delete_cb = deque([cb]) expect(cb) self.klass._recv_delete_ok('frame') assert_equals(deque(), self.klass._delete_cb)
class ExchangeClassTest(Chai): def setUp(self): super(ExchangeClassTest, self).setUp() ch = mock() ch.channel_id = 42 ch.logger = mock() self.klass = ExchangeClass(ch) def test_init(self): expect(ProtocolClass.__init__).args('foo', a='b') klass = ExchangeClass.__new__(ExchangeClass) klass.__init__('foo', a='b') assert_equals({ 11: klass._recv_declare_ok, 21: klass._recv_delete_ok, }, klass.dispatch_map) assert_equals(deque(), klass._declare_cb) assert_equals(deque(), klass._delete_cb) def test_cleanup(self): self.klass._cleanup() assert_equals(None, self.klass._declare_cb) assert_equals(None, self.klass._delete_cb) assert_equals(None, self.klass._channel) assert_equals(None, self.klass.dispatch_map) def test_declare_default_args(self): w = mock() expect(self.klass.allow_nowait).returns(True) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args(self.klass.default_ticket).returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_shortstr).args('topic').returns(w) expect(w.write_bits).args(False, False, False, False, True).returns(w) expect(w.write_table).args({}) expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 10, w).returns('frame') expect(self.klass.send_frame).args('frame') stub(self.klass.channel.add_synchronous_cb) self.klass.declare('exchange', 'topic') assert_equals(deque(), self.klass._declare_cb) def test_declare_with_args(self): w = mock() stub(self.klass.allow_nowait) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args('t').returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_shortstr).args('topic').returns(w) expect(w.write_bits).args('p', 'd', False, False, False).returns(w) expect(w.write_table).args('table') expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 10, w).returns('frame') expect(self.klass.send_frame).args('frame') expect(self.klass.channel.add_synchronous_cb).args( self.klass._recv_declare_ok) self.klass.declare('exchange', 'topic', passive='p', durable='d', nowait=False, arguments='table', ticket='t') assert_equals(deque([None]), self.klass._declare_cb) def test_declare_with_cb(self): w = mock() expect(self.klass.allow_nowait).returns(True) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args('t').returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_shortstr).args('topic').returns(w) expect(w.write_bits).args('p', 'd', False, False, False).returns(w) expect(w.write_table).args('table') expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 10, w).returns('frame') expect(self.klass.send_frame).args('frame') expect(self.klass.channel.add_synchronous_cb).args( self.klass._recv_declare_ok) self.klass.declare('exchange', 'topic', passive='p', durable='d', nowait=True, arguments='table', ticket='t', cb='foo') assert_equals(deque(['foo']), self.klass._declare_cb) def test_recv_declare_ok_no_cb(self): self.klass._declare_cb = deque([None]) self.klass._recv_declare_ok('frame') assert_equals(deque(), self.klass._declare_cb) def test_recv_declare_ok_with_cb(self): cb = mock() self.klass._declare_cb = deque([cb]) expect(cb) self.klass._recv_declare_ok('frame') assert_equals(deque(), self.klass._declare_cb) def test_delete_default_args(self): w = mock() expect(self.klass.allow_nowait).returns(True) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args(self.klass.default_ticket).returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_bits).args(False, True) expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 20, w).returns('frame') expect(self.klass.send_frame).args('frame') stub(self.klass.channel.add_synchronous_cb) self.klass.delete('exchange') assert_equals(deque(), self.klass._delete_cb) def test_delete_with_args(self): w = mock() stub(self.klass.allow_nowait) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args('t').returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_bits).args('maybe', False) expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 20, w).returns('frame') expect(self.klass.send_frame).args('frame') expect(self.klass.channel.add_synchronous_cb).args( self.klass._recv_delete_ok) self.klass.delete('exchange', if_unused='maybe', nowait=False, ticket='t') assert_equals(deque([None]), self.klass._delete_cb) def test_delete_with_cb(self): w = mock() expect(self.klass.allow_nowait).returns(True) expect(mock(exchange_class, 'Writer')).returns(w) expect(w.write_short).args('t').returns(w) expect(w.write_shortstr).args('exchange').returns(w) expect(w.write_bits).args('maybe', False) expect(mock(exchange_class, 'MethodFrame')).args(42, 40, 20, w).returns('frame') expect(self.klass.send_frame).args('frame') expect(self.klass.channel.add_synchronous_cb).args( self.klass._recv_delete_ok) self.klass.delete('exchange', if_unused='maybe', nowait=True, ticket='t', cb='foo') assert_equals(deque(['foo']), self.klass._delete_cb) def test_recv_delete_ok_no_cb(self): self.klass._delete_cb = deque([None]) self.klass._recv_delete_ok('frame') assert_equals(deque(), self.klass._delete_cb) def test_recv_delete_ok_with_cb(self): cb = mock() self.klass._delete_cb = deque([cb]) expect(cb) self.klass._recv_delete_ok('frame') assert_equals(deque(), self.klass._delete_cb)