Close queue connection.

diff --git a/pithos/backends/lib/rabbitmq/queue.py b/pithos/backends/lib/rabbitmq/queue.py
index c2ab22a..f53ad97 100644 (file)
--- a/pithos/backends/lib/rabbitmq/queue.py
+++ b/pithos/backends/lib/rabbitmq/queue.py
@@ -31,7 +31,7 @@
 # interpreted as representing official policies, either expressed
 # or implied, of GRNET S.A.
 
-from pithos.lib.queue import exchange_connect, exchange_send, Receipt
+from pithos.lib.queue import exchange_connect, exchange_send, exchange_close, Receipt
 
 
 class Queue(object):
@@ -48,3 +48,6 @@ class Queue(object):
     def send(self, user, resource, value, details):
         body = Receipt(self.client_id, user, resource, value, details).format()
         exchange_send(self.conn, self.message_key, body)
+    
+    def close(self):
+        exchange_close(self.conn)

diff --git a/pithos/backends/modular.py b/pithos/backends/modular.py
index 961d258..2963d53 100644 (file)
--- a/pithos/backends/modular.py
+++ b/pithos/backends/modular.py
@@ -134,11 +134,15 @@ class ModularBackend(BaseBackend):
             class NoQueue:
                 def send(self, *args):
                     pass
+                
+                def close(self):
+                    pass
             
             self.queue = NoQueue()
     
     def close(self):
         self.wrapper.close()
+        self.queue.close()
     
     @backend_method
     def list_accounts(self, user, marker=None, limit=10000):