def assertExpires(self, dec, expired, tasks=10):
     worker = Worker(hostname='fuzzie')
     worker.on_online(timestamp=time())
     for total in xrange(tasks):
         task = create_task(worker)
         task.on_received(timestamp=time() - dec)
         task.on_succeeded(timestamp=time() - dec, result=42)
         self.assertTrue(task.name)
         self.assertTrue(self.cam.handle_task((task.uuid, task)))
     self.assertEqual(self.cam.on_cleanup(), expired)
 def assertExpires(self, dec, expired, tasks=10):
     worker = Worker(hostname="fuzzie")
     worker.on_online(timestamp=time())
     for total in xrange(tasks):
         task = create_task(worker)
         task.on_received(timestamp=time() - dec)
         task.on_succeeded(timestamp=time() - dec, result=42)
         self.assertTrue(task.name)
         self.assertTrue(self.cam.handle_task((task.uuid, task)))
     self.assertEqual(self.cam.on_cleanup(), expired)
 def test_handle_worker(self):
     worker = Worker(hostname="fuzzie")
     worker.on_online(timestamp=time())
     m = self.cam.handle_worker((worker.hostname, worker))
     self.assertTrue(m)
     self.assertTrue(m.hostname)
     self.assertTrue(m.last_heartbeat)
     self.assertTrue(m.is_alive())
     self.assertEqual(unicode(m), unicode(m.hostname))
     self.assertTrue(repr(m))
 def test_handle_worker(self):
     worker = Worker(hostname='fuzzie')
     worker.on_online(timestamp=time())
     self.cam._last_worker_write.clear()
     m = self.cam.handle_worker((worker.hostname, worker))
     self.assertTrue(m)
     self.assertTrue(m.hostname)
     self.assertTrue(m.last_heartbeat)
     self.assertTrue(m.is_alive())
     self.assertEqual(unicode(m), unicode(m.hostname))
     self.assertTrue(repr(m))
 def test_handle_worker(self):
     worker = Worker(hostname='fuzzie')
     worker.on_online(time(), time())
     self.cam._last_worker_write.clear()
     m = self.cam.handle_worker((worker.hostname, worker))
     self.assertTrue(m)
     self.assertTrue(m.hostname)
     self.assertTrue(m.last_heartbeat)
     self.assertTrue(m.is_alive())
     self.assertEqual(unicode(m), unicode(m.hostname))
     self.assertTrue(repr(m))
    def test_handle_task_received(self):
        worker = Worker(hostname='fuzzie')
        worker.on_online(timestamp=time())
        self.cam.handle_worker((worker.hostname, worker))

        task = create_task(worker)
        task.on_received(timestamp=time())
        self.assertEqual(task.state, 'RECEIVED')
        mt = self.cam.handle_task((task.uuid, task))
        self.assertEqual(mt.name, task.name)
        self.assertTrue(unicode(mt))
        self.assertTrue(repr(mt))
        mt.eta = celery.now()
        self.assertIn('eta', unicode(mt))
        self.assertIn(mt, models.TaskState.objects.active())
    def test_handle_task_received(self):
        worker = Worker(hostname="fuzzie")
        worker.on_online(timestamp=time())
        self.cam.handle_worker((worker.hostname, worker))

        task = create_task(worker)
        task.on_received(timestamp=time())
        self.assertEqual(task.state, "RECEIVED")
        mt = self.cam.handle_task((task.uuid, task))
        self.assertEqual(mt.name, task.name)
        self.assertTrue(unicode(mt))
        self.assertTrue(repr(mt))
        mt.eta = now()
        self.assertIn("eta", unicode(mt))
        self.assertIn(mt, models.TaskState.objects.active())
    def test_handle_task(self):
        worker1 = Worker(hostname='fuzzie')
        worker1.on_online(timestamp=time())
        mw = self.cam.handle_worker((worker1.hostname, worker1))
        task1 = create_task(worker1)
        task1.on_received(timestamp=time())
        mt = self.cam.handle_task((task1.uuid, task1))
        self.assertEqual(mt.worker, mw)

        worker2 = Worker(hostname=None)
        task2 = create_task(worker2)
        task2.on_received(timestamp=time())
        mt = self.cam.handle_task((task2.uuid, task2))
        self.assertIsNone(mt.worker)

        task1.on_succeeded(timestamp=time(), result=42)
        mt = self.cam.handle_task((task1.uuid, task1))
        self.assertEqual(mt.name, task1.name)
        self.assertEqual(mt.result, 42)

        task3 = create_task(worker1, name=None)
        task3.on_revoked(timestamp=time())
        mt = self.cam.handle_task((task3.uuid, task3))
        self.assertIsNone(mt)
    def test_handle_task(self):
        worker1 = Worker(hostname="fuzzie")
        worker1.on_online(timestamp=time())
        mw = self.cam.handle_worker((worker1.hostname, worker1))
        task1 = create_task(worker1)
        task1.on_received(timestamp=time())
        mt = self.cam.handle_task((task1.uuid, task1))
        self.assertEqual(mt.worker, mw)

        worker2 = Worker(hostname=None)
        task2 = create_task(worker2)
        task2.on_received(timestamp=time())
        mt = self.cam.handle_task((task2.uuid, task2))
        self.assertIsNone(mt.worker)

        task1.on_succeeded(timestamp=time(), result=42)
        mt = self.cam.handle_task((task1.uuid, task1))
        self.assertEqual(mt.name, task1.name)
        self.assertEqual(mt.result, 42)

        task3 = create_task(worker1, name=None)
        task3.on_revoked(timestamp=time())
        mt = self.cam.handle_task((task3.uuid, task3))
        self.assertIsNone(mt)