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()
