Fix HttpRpcTransportHelper so that it properly implements IHelper.

praekeltfoundation · Feb 6, 2014 · 2ac2813 · 2ac2813
1 parent 02c6bd8
commit 2ac2813
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/vumi/transports/httprpc/tests/helpers.py b/vumi/transports/httprpc/tests/helpers.py
@@ -42,9 +42,15 @@ def __init__(self, transport_class, use_riak=False, request_defaults=None,
 
         generate_proxies(self, self._transport_helper)
 
+    def setup(self, *args, **kw):
+        return self._transport_helper.setup(*args, **kw)
+
+    def cleanup(self):
+        return self._transport_helper.cleanup()
+
     @inlineCallbacks
     def get_transport(self, config, cls=None, start=True):
-        transport = yield super(HttpRpcTransportHelper, self).get_transport(
+        transport = yield self._transport_helper.get_transport(
             config, cls=cls, start=True)
         self.transport_url = transport.get_transport_url(config['web_path'])
         returnValue(transport)