class TestEngineAddChannel(unittest.TestCase): 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_added_to_engine(self): self.engine.add_channel(self.channel) self.assertTrue(self.channel.fileno in self.engine._channels) self.assertEqual(self.engine._channels[self.channel.fileno], self.channel) def test_channel_is_added_to_poller(self): self.engine.add_channel(self.channel) self.poller.add.assert_called_once_with(self.channel.fileno, self.channel._events)
class TestEngineAddChannel(unittest.TestCase): 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_added_to_engine(self): self.engine.add_channel(self.channel) self.assertTrue(self.channel.fileno in self.engine._channels) self.assertEqual(self.engine._channels[self.channel.fileno], self.channel) def test_channel_is_added_to_poller(self): self.engine.add_channel(self.channel) self.poller.add.assert_called_once_with(self.channel.fileno, self.channel._events)
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()