def test_set_read_cb_when_data_to_flush_but_pending_read_event(self):
sock = EventSocket()
sock._read_buf = bytearray('somedata')
sock._parent_read_cb = None
sock._pending_read_cb_event = 'pending_event'
sock._set_read_cb( 'parent_read_cb' )
assert_equals( 'parent_read_cb', sock._parent_read_cb )
assert_equals( 'pending_event', sock._pending_read_cb_event )
def test_parent_read_timer_cb_when_closed(self):
sock = EventSocket()
sock._pending_read_cb_event = 'foo'
sock._closed = True
sock._parent_read_cb = mock()
sock._parent_read_timer_cb()
assert_equals( None, sock._error_msg )
assert_equals( 'foo', sock._pending_read_cb_event )
def test_parent_read_timer_cb(self):
sock = EventSocket()
sock._pending_read_cb_event = 'foo'
sock._parent_read_cb = mock()
expect( sock._parent_read_cb ).args( sock )
sock._parent_read_timer_cb()
assert_equals( 'error processing socket input buffer', sock._error_msg )
assert_equals( None, sock._pending_read_cb_event )
def test_set_read_cb_when_should_flush(self):
sock = EventSocket()
sock._read_buf = bytearray('somedata')
sock._parent_read_cb = None
sock._pending_read_cb_event = None
expect(eventsocket.event.timeout).args(0, sock._protected_cb, sock._parent_read_timer_cb).returns('timeout_event')
sock._set_read_cb( 'parent_read_cb' )
assert_equals( 'parent_read_cb', sock._parent_read_cb )
assert_equals( 'timeout_event', sock._pending_read_cb_event )
def test_read_cb_when_parent_cb_and_is_a_pending_event_and_already_buffered_data(self):
sock = EventSocket()
sock._read_buf = bytearray('foo')
sock._sock = mock()
sock._peername = 'peername'
sock._parent_read_cb = 'p_read_cb'
sock._pending_read_cb_event = 'pending_read'
mock( sock, 'getsockopt' )
expect( sock.getsockopt ).args( socket.SOL_SOCKET, socket.SO_RCVBUF ).returns( 42 )
expect( sock._sock.recv ).args( 42 ).returns( 'sumdata' )
expect( sock._flag_activity )
assert_true( sock._read_cb() )
assert_equals( bytearray('foosumdata'), sock._read_buf )
def test_accept_cb_when_no_logger_and_no_parent_cb(self):
sock = EventSocket()
sock._sock = mock()
sock._parent_read_cb = 'p_read_cb'
sock._parent_error_cb = 'p_error_cb'
sock._parent_close_cb = 'p_close_cb'
sock._debug = False
sock._logger = None
sock._max_read_buffer = 42
expect(sock._sock.accept).returns( ('connection', 'address') )
expect(EventSocket.__init__).args( read_cb='p_read_cb', error_cb='p_error_cb',
close_cb='p_close_cb', sock='connection', debug=False,
logger=None, max_read_buffer=42 )
assert_true( sock._accept_cb() )
assert_equals( 'error accepting new socket', sock._error_msg )
def test_read_cb_when_debugging_and_parent_cb_and_no_pending_event(self):
sock = EventSocket()
sock._sock = mock()
sock._logger = mock()
sock._peername = 'peername'
sock._debug = True
sock._parent_read_cb = 'p_read_cb'
mock( sock, 'getsockopt' )
expect( sock.getsockopt ).args( socket.SOL_SOCKET, socket.SO_RCVBUF ).returns( 42 )
expect( sock._sock.recv ).args( 42 ).returns( 'sumdata' )
expect( sock._logger.debug ).args( 'read 7 bytes from peername' )
expect( sock._flag_activity )
expect( eventsocket.event.timeout ).args( 0, sock._protected_cb, sock._parent_read_timer_cb ).returns('pending_read')
assert_true( sock._read_cb() )
assert_equals( bytearray('sumdata'), sock._read_buf )
assert_equals( 'pending_read', sock._pending_read_cb_event )
def test_accept_cb_when_logger_and_parent_cb(self):
sock = EventSocket()
sock._sock = mock()
sock._parent_accept_cb = 'p_accept_cb'
sock._parent_read_cb = 'p_read_cb'
sock._parent_error_cb = 'p_error_cb'
sock._parent_close_cb = 'p_close_cb'
sock._debug = True
sock._logger = mock()
sock._max_read_buffer = 42
expect(sock._sock.accept).returns( ('connection', 'address') )
expect(sock._logger.debug).args( "accepted connection from address" )
expect(EventSocket.__init__).args( read_cb='p_read_cb', error_cb='p_error_cb',
close_cb='p_close_cb', sock='connection', debug=True,
logger=sock._logger, max_read_buffer=42 )
expect(sock._protected_cb).args( 'p_accept_cb', is_a(EventSocket) )
assert_true( sock._accept_cb() )
