mirror of https://github.com/peter4431/rq.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
875 B
Python
30 lines
875 B
Python
13 years ago
|
from tests import RQTestCase
|
||
|
|
from tests.fixtures import decorated_job
|
||
|
|
|
||
|
|
from rq.decorators import job
|
||
|
|
from rq.job import Job
|
||
|
|
|
||
|
|
|
||
|
|
class TestDecorator(RQTestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super(TestDecorator, self).setUp()
|
||
|
|
|
||
|
|
def test_decorator_preserves_functionality(self):
|
||
|
|
"""
|
||
|
|
Ensure that a decorated function's functionality is still preserved
|
||
|
|
"""
|
||
|
|
self.assertEqual(decorated_job(1, 2), 3)
|
||
|
|
|
||
|
|
def test_decorator_adds_delay_attr(self):
|
||
|
|
"""
|
||
|
|
Ensure that decorator adds a delay attribute to function that returns
|
||
|
|
a Job instance when called.
|
||
|
|
"""
|
||
|
|
self.assertTrue(hasattr(decorated_job, 'delay'))
|
||
|
|
result = decorated_job.delay(1, 2)
|
||
|
|
self.assertTrue(isinstance(result, Job))
|
||
|
|
# Ensure that job returns the right result when performed
|
||
|
|
self.assertEqual(result.perform(), 3)
|
||
|
|
|