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_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_set_read_cb_when_no_reason_to_schedule_flush(self):
   sock = EventSocket()
   sock._set_read_cb( 'readit' )
   assert_equals( 'readit', sock._parent_read_cb )