def setUp(self):
self.engine = Engine()
self.poller = MagicMock()
self.poller.add = MagicMock()
self.engine._poller = self.poller
self.channel = MagicMock()
self.channel.fileno = "foo"
self.channel._events = "bar"
def test_channel_is_modified_on_poller(self):
engine = Engine()
channel = MagicMock()
channel.fileno = "foo"
channel._events = "bar"
engine._poller.modify = MagicMock()
engine.modify_channel(channel)
engine._poller.modify.assert_called_once_with(channel.fileno,
channel._events)
def setUp(self):
self.times_called = []
self.engine = Engine()
def setUp(self):
self.engine = Engine()
def test_cancelling_timer_calls_engine_remove_timer(self):
engine = Engine()
engine._remove_timer = MagicMock()
timer = _Timer(engine, None, None)
timer.cancel()
engine._remove_timer.assert_called_once_with(timer)
def test_engine_local_instances(self):
engine1 = Engine()
engine2 = Engine()
self.assertFalse(engine1 is engine2)
def test_channel_gets_removed_from_engine(self):
engine = Engine()
engine.remove_channel = MagicMock()
channel = _Channel(socket=socket.socket(), engine=engine)
channel.close()
engine.remove_channel.assert_called_once_with(channel)
def test_channel_gets_added_to_engine(self):
engine = Engine()
engine.add_channel = MagicMock()
channel = _Channel(socket=socket.socket(), engine=engine)
engine.add_channel.assert_called_once_with(channel)
channel.close()
def test_channel_constructor_engine_arg(self):
engine = Engine()
channel = _Channel(engine=engine)
self.assertTrue(channel.engine is engine)
