Exemplos de CeleryNodeMonitor.__enter__ em Python

Exemplo n.º 1

Arquivo: celery_node_monitor_test.py Projeto: vup1120/oq-engine

 def test_all_nodes_were_down(self):
     ping = self.inspect().ping
     ping.return_value = {}
     mon = CeleryNodeMonitor(no_distribute=False, interval=1)
     with self.assertRaises(SystemExit), mock.patch('sys.stderr') as stderr:
         mon.__enter__()
     self.assertEqual(ping.call_count, 1)  # called only once
     self.assertTrue(stderr.write.called)  # an error message was printed

Exemplo n.º 2

 def test_all_nodes_were_down(self):
     ping = self.inspect().ping
     ping.return_value = {}
     mon = CeleryNodeMonitor(no_distribute=False, interval=1)
     with self.assertRaises(SystemExit), mock.patch('sys.stderr') as stderr:
         mon.__enter__()
     self.assertEqual(ping.call_count, 1)  # called only once
     self.assertTrue(stderr.write.called)  # an error message was printed