Ejemplo n.º 1
0
    def test_mark_as_done(self):
        tb = DatabaseBackend(self.uri, app=self.app)

        tid = uuid()

        self.assertEqual(tb.get_state(tid), states.PENDING)
        self.assertIsNone(tb.get_result(tid))

        tb.mark_as_done(tid, 42)
        self.assertEqual(tb.get_state(tid), states.SUCCESS)
        self.assertEqual(tb.get_result(tid), 42)
Ejemplo n.º 2
0
    def test_mark_as_done(self):
        tb = DatabaseBackend(self.uri, app=self.app)

        tid = uuid()

        assert tb.get_state(tid) == states.PENDING
        assert tb.get_result(tid) is None

        tb.mark_as_done(tid, 42)
        assert tb.get_state(tid) == states.SUCCESS
        assert tb.get_result(tid) == 42
Ejemplo n.º 3
0
    def test_mark_as_done(self):
        tb = DatabaseBackend(self.uri, app=self.app)

        tid = uuid()

        assert tb.get_state(tid) == states.PENDING
        assert tb.get_result(tid) is None

        tb.mark_as_done(tid, 42)
        assert tb.get_state(tid) == states.SUCCESS
        assert tb.get_result(tid) == 42
Ejemplo n.º 4
0
    def test_mark_as_done(self):
        tb = DatabaseBackend(self.uri, app=self.app)

        tid = uuid()

        self.assertEqual(tb.get_state(tid), states.PENDING)
        self.assertIsNone(tb.get_result(tid))

        tb.mark_as_done(tid, 42)
        self.assertEqual(tb.get_state(tid), states.SUCCESS)
        self.assertEqual(tb.get_result(tid), 42)
Ejemplo n.º 5
0
 def test_mark_as_retry(self):
     tb = DatabaseBackend(self.uri, app=self.app)
     tid = uuid()
     try:
         raise KeyError('foo')
     except KeyError as exception:
         import traceback
         trace = '\n'.join(traceback.format_stack())
         tb.mark_as_retry(tid, exception, traceback=trace)
         assert tb.get_state(tid) == states.RETRY
         assert isinstance(tb.get_result(tid), KeyError)
         assert tb.get_traceback(tid) == trace
Ejemplo n.º 6
0
 def test_mark_as_retry(self):
     tb = DatabaseBackend(self.uri, app=self.app)
     tid = uuid()
     try:
         raise KeyError('foo')
     except KeyError as exception:
         import traceback
         trace = '\n'.join(traceback.format_stack())
         tb.mark_as_retry(tid, exception, traceback=trace)
         assert tb.get_state(tid) == states.RETRY
         assert isinstance(tb.get_result(tid), KeyError)
         assert tb.get_traceback(tid) == trace
Ejemplo n.º 7
0
    def test_mark_as_failure(self):
        tb = DatabaseBackend(self.uri, app=self.app)

        tid3 = uuid()
        try:
            raise KeyError('foo')
        except KeyError as exception:
            import traceback
            trace = '\n'.join(traceback.format_stack())
            tb.mark_as_failure(tid3, exception, traceback=trace)
            self.assertEqual(tb.get_state(tid3), states.FAILURE)
            self.assertIsInstance(tb.get_result(tid3), KeyError)
            self.assertEqual(tb.get_traceback(tid3), trace)
Ejemplo n.º 8
0
 def test_missing_task_id_is_PENDING(self):
     tb = DatabaseBackend(self.uri, app=self.app)
     assert tb.get_state('xxx-does-not-exist') == states.PENDING
Ejemplo n.º 9
0
 def test_mark_as_revoked(self):
     tb = DatabaseBackend(self.uri, app=self.app)
     tid = uuid()
     tb.mark_as_revoked(tid)
     assert tb.get_state(tid) == states.REVOKED
Ejemplo n.º 10
0
 def test_mark_as_started(self):
     tb = DatabaseBackend(self.uri, app=self.app)
     tid = uuid()
     tb.mark_as_started(tid)
     assert tb.get_state(tid) == states.STARTED
Ejemplo n.º 11
0
 def test_missing_task_id_is_PENDING(self):
     tb = DatabaseBackend(self.uri, app=self.app)
     assert tb.get_state('xxx-does-not-exist') == states.PENDING
Ejemplo n.º 12
0
 def test_mark_as_revoked(self):
     tb = DatabaseBackend(self.uri, app=self.app)
     tid = uuid()
     tb.mark_as_revoked(tid)
     assert tb.get_state(tid) == states.REVOKED
Ejemplo n.º 13
0
 def test_mark_as_started(self):
     tb = DatabaseBackend(self.uri, app=self.app)
     tid = uuid()
     tb.mark_as_started(tid)
     assert tb.get_state(tid) == states.STARTED