def test_fetch_empty(self):
channel = Mock()
reader = Mock(channel=channel)
# test
r = Receiver(reader)
r.inbox = Mock()
r.inbox.empty.return_value = False
r.inbox.get.side_effect = Empty
r._wait = Mock(side_effect=r.inbox.put)
self.assertRaises(r.fetch)
def test_fetch_empty(self):
channel = Mock()
reader = Mock(channel=channel)
# test
r = Receiver(reader)
r.inbox = Mock()
r.inbox.empty.return_value = False
r.inbox.get.side_effect = Empty
r._wait = Mock(side_effect=r.inbox.put)
self.assertRaises(r.fetch)
def test_fetch(self):
timeout = 10
channel = Mock()
reader = Mock(channel=channel)
received = 33
# test
r = Receiver(reader)
r.inbox = Mock()
r.inbox.empty.return_value = True
r.inbox.get.return_value = received
r._wait = Mock(side_effect=r.inbox.put)
message = r.fetch(timeout)
# validation
fd = channel.connection.sock.fileno.return_value
r._wait.assert_called_once_with(fd, channel, timeout)
self.assertEqual(message, received)
def test_fetch(self):
timeout = 10
channel = Mock()
reader = Mock(channel=channel)
received = 33
# test
r = Receiver(reader)
r.inbox = Mock()
r.inbox.empty.return_value = True
r.inbox.get.return_value = received
r._wait = Mock(side_effect=r.inbox.put)
message = r.fetch(timeout)
# validation
fd = channel.connection.sock.fileno.return_value
r._wait.assert_called_once_with(fd, channel, timeout)
self.assertEqual(message, received)
