def test_flag_activity(self): sock = EventSocket() sock._inactive_event = mock() sock._inactive_timeout = 42 expect( sock._inactive_event.delete ) expect( eventsocket.event.timeout ).args( 42, sock._protected_cb, sock._inactive_cb ).returns( 'doitagain' ) sock._flag_activity() assert_equals( 'doitagain', sock._inactive_event )