Fix set expire in headers & proxydialog close

Eyepea · Apr 30, 2018 · 717f0ff · 717f0ff
1 parent 6c4caa3
commit 717f0ff
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/aiosip/dialog.py b/aiosip/dialog.py
@@ -492,3 +492,6 @@ def proxy(self, message):
         else:
             message.headers['Via'].insert(0, self.proxy_peer.generate_via_headers())
             self.proxy_peer.send_message(message)
+
+    async def close(self, *args, **kwargs):
+        self._close()
diff --git a/aiosip/peers.py b/aiosip/peers.py
@@ -105,19 +105,20 @@ async def request(self, method, from_details, to_details, contact_details=None,
             raise
 
     async def subscribe(self, expires=3600, **kwargs):
-        headers = kwargs.get('headers', CIMultiDict())
 
         if expires:
+            headers = kwargs.get('headers', CIMultiDict())
             headers['Expires'] = expires
+            kwargs['headers'] = headers
 
         return await self.request('SUBSCRIBE', **kwargs)
 
     async def register(self, expires=3600, **kwargs):
 
-        headers = kwargs.get('headers', CIMultiDict())
-
         if expires:
+            headers = kwargs.get('headers', CIMultiDict())
             headers['Expires'] = expires
+            kwargs['headers'] = headers
 
         return await self.request('REGISTER', **kwargs)