Change base transport's publish_status() to expect a component argument

praekeltfoundation · Oct 15, 2015 · 80f8314 · 80f8314
1 parent 707ac66
commit 80f8314
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/vumi/transports/base.py b/vumi/transports/base.py
@@ -168,11 +168,11 @@ def publish_delivery_report(self, user_message_id, delivery_status, **kw):
                                   delivery_status=delivery_status,
                                   event_type='delivery_report', **kw)
 
-    def publish_status(self, status, **kw):
+    def publish_status(self, component, status, **kw):
         """
         Helper method for publishing a status message.
         """
-        msg = TransportStatus(status=status, **kw)
+        msg = TransportStatus(component=component, status=status, **kw)
         return self.connectors[self.status_connector_name].publish_status(msg)
 
     def _send_failure_eb(self, f, message):

diff --git a/vumi/transports/tests/test_base.py b/vumi/transports/tests/test_base.py
@@ -120,7 +120,10 @@ def test_publish_status(self):
             'transport_name': 'foo'
         })
 
-        msg = yield transport.publish_status('major', reasons=['many lemons'])
+        msg = yield transport.publish_status(
+            'a', 'major', reasons=['many lemons'])
+
+        self.assertEqual(msg['component'], 'a')
         self.assertEqual(msg['status'], 'major')
         self.assertEqual(msg['reasons'], ['many lemons'])