def test_start_no_replies(self):
c = Mock()
c.app.connection_for_read = _amqp_connection()
mingle = Mingle(c)
I = c.app.control.inspect.return_value = Mock()
I.hello.return_value = {}
mingle.start(c)
def test_start_no_replies(self):
c = Mock()
c.app.connection_for_read = _amqp_connection()
mingle = Mingle(c)
I = c.app.control.inspect.return_value = Mock()
I.hello.return_value = {}
mingle.start(c)
def test_start(self):
try:
c = Mock()
c.app.connection_for_read = _amqp_connection()
mingle = Mingle(c)
self.assertTrue(mingle.enabled)
Aig = LimitedSet()
Big = LimitedSet()
Aig.add('Aig-1')
Aig.add('Aig-2')
Big.add('Big-1')
I = c.app.control.inspect.return_value = Mock()
I.hello.return_value = {
'*****@*****.**': {
'clock': 312,
'revoked': Aig._data,
},
'*****@*****.**': {
'clock': 29,
'revoked': Big._data,
},
'*****@*****.**': {
'error': 'unknown method',
},
}
mingle.start(c)
I.hello.assert_called_with(c.hostname, worker_state.revoked._data)
c.app.clock.adjust.assert_has_calls([
call(312),
call(29),
],
any_order=True)
self.assertIn('Aig-1', worker_state.revoked)
self.assertIn('Aig-2', worker_state.revoked)
self.assertIn('Big-1', worker_state.revoked)
finally:
worker_state.revoked.clear()
def test_start(self):
try:
c = Mock()
c.app.connection_for_read = _amqp_connection()
mingle = Mingle(c)
self.assertTrue(mingle.enabled)
Aig = LimitedSet()
Big = LimitedSet()
Aig.add('Aig-1')
Aig.add('Aig-2')
Big.add('Big-1')
I = c.app.control.inspect.return_value = Mock()
I.hello.return_value = {
'*****@*****.**': {
'clock': 312,
'revoked': Aig._data,
},
'*****@*****.**': {
'clock': 29,
'revoked': Big._data,
},
'*****@*****.**': {
'error': 'unknown method',
},
}
mingle.start(c)
I.hello.assert_called_with(c.hostname, worker_state.revoked._data)
c.app.clock.adjust.assert_has_calls([
call(312), call(29),
], any_order=True)
self.assertIn('Aig-1', worker_state.revoked)
self.assertIn('Aig-2', worker_state.revoked)
self.assertIn('Big-1', worker_state.revoked)
finally:
worker_state.revoked.clear()
