def test_periodic(self):
c = self.Consumer()
c.app.connection_for_read = _amqp_connection()
g = Gossip(c)
g.on_node_lost = Mock()
state = g.state = Mock()
worker = Mock()
state.workers = {'foo': worker}
worker.alive = True
worker.hostname = 'foo'
g.periodic()
worker.alive = False
g.periodic()
g.on_node_lost.assert_called_with(worker)
with pytest.raises(KeyError):
state.workers['foo']
def Consumer(self, hostname='*****@*****.**', pid=4312):
c = Mock()
c.app.connection = _amqp_connection()
c.hostname = hostname
c.pid = pid
return c
