added tests for new add_job_by_name

sqjobs/connectors/dummy.py

@@ -38,10 +38,10 @@ def set_retry_time(self, queue_name, message_id, delay):
         self.retried_jobs.setdefault(queue_name, []).append((message_id, delay))
         self.num_retried_jobs += 1
 
-    def serialize_job(self, job_class, job_id, args, kwargs):
+    def serialize_job(self, job_name, job_id, args, kwargs):
         return {
             'id': job_id,
-            'name': job_class._task_name(),
+            'name': job_name,
             'args': args,
             'kwargs': kwargs
         }

sqjobs/tests/broker_test.py

@@ -36,6 +36,18 @@ def test_add_job_to_broker(self):
         messages = broker.connector.jobs['sqjobs']
         assert len(messages) == 1
 
+    def test_add_job_by_name_to_broker(self):
+        broker = StandardBroker(self.connector)
+        broker.add_job_by_name('Adder', 'sqjobs', 2, 3)
+
+        queues = list(broker.connector.jobs.keys())
+        assert len(queues) == 1
+        assert queues[0] == 'sqjobs'
+        assert broker.connector.num_jobs == 1
+
+        messages = broker.connector.jobs['sqjobs']
+        assert len(messages) == 1
+
     def test_right_payload_args_when_job_is_added(self):
         broker = StandardBroker(self.connector)
         broker.add_job(Adder, 2, 3)
@@ -45,6 +57,15 @@ def test_right_payload_args_when_job_is_added(self):
 
         assert message == {'args': (2, 3), 'kwargs': {}, 'name': 'adder'}
 
+    def test_right_payload_args_when_job_is_added_by_name(self):
+        broker = StandardBroker(self.connector)
+        broker.add_job_by_name('adder', 'sqjobs', 2, 3)
+
+        message = broker.connector.jobs['sqjobs'][0]
+        del message['id']  # Check that exists, but we don't care about the value
+
+        assert message == {'args': (2, 3), 'kwargs': {}, 'name': 'adder'}
+
     def test_right_payload_kwargs_when_job_is_added(self):
         broker = StandardBroker(self.connector)
         broker.add_job(Adder, num2=2, num1=3)
@@ -54,6 +75,15 @@ def test_right_payload_kwargs_when_job_is_added(self):
 
         assert message == {'args': (), 'kwargs': {'num1': 3, 'num2': 2}, 'name': 'adder'}
 
+    def test_right_payload_kwargs_when_job_is_added_by_name(self):
+        broker = StandardBroker(self.connector)
+        broker.add_job_by_name('adder', 'sqjobs', num2=2, num1=3)
+
+        message = broker.connector.jobs['sqjobs'][0]
+        del message['id']  # Check that exists, but we don't care about the value
+
+        assert message == {'args': (), 'kwargs': {'num1': 3, 'num2': 2}, 'name': 'adder'}
+
     def test_right_payload_both_when_job_is_added(self):
         broker = StandardBroker(self.connector)
         broker.add_job(Adder, 2, num2=3)
@@ -63,6 +93,15 @@ def test_right_payload_both_when_job_is_added(self):
 
         assert message == {'args': (2,), 'kwargs': {'num2': 3}, 'name': 'adder'}
 
+    def test_right_payload_both_when_job_is_added_by_name(self):
+        broker = StandardBroker(self.connector)
+        broker.add_job_by_name('adder', 'sqjobs', 2, num2=3)
+
+        message = broker.connector.jobs['sqjobs'][0]
+        del message['id']  # Check that exists, but we don't care about the value
+
+        assert message == {'args': (2,), 'kwargs': {'num2': 3}, 'name': 'adder'}
+
     def test_multiple_jobs_are_stored_correctly_by_the_broker(self):
         broker = StandardBroker(self.connector)
         job_ids, jobs = [], []

sqjobs/tests/connector_test.py

@@ -149,7 +149,7 @@ def test_serialize_job_returns_valid_json(self):
         }
 
         serializer = sqs_connector.serialize_job(
-            job_class=Adder,
+            job_name='adder',
             job_id=15,
             args=[3, 4],
             kwargs={'first_param': 1, 'second_param': 'two'}