added lookup_url as a parameter for Connection and AsyncConnection.

This should make test cases easier.
hahn-th · May 22, 2018 · 906a21a · 906a21a
1 parent cf0ae94
commit 906a21a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/homematicip/async/connection.py b/homematicip/async/connection.py
@@ -39,12 +39,11 @@ def ws_connected(self):
         """Websocket is connected."""
         return self.socket_connection is not None
 
-    async def init(self, accesspoint_id, lookup=True, **kwargs):
+    async def init(self, accesspoint_id, lookup=True, lookup_url="https://lookup.homematic.com:48335/getHost", **kwargs):
         self.set_token_and_characteristics(accesspoint_id)
 
         if lookup:
-            result = await self.api_call("https://lookup.homematic.com:48335/getHost",
-                                         json.dumps(self.clientCharacteristics), full_url=True)
+            result = await self.api_call(lookup_url, json.dumps(self.clientCharacteristics), full_url=True)
 
             self._urlREST = result["urlREST"]
             self._urlWebSocket = result["urlWebSocket"]

diff --git a/homematicip/connection.py b/homematicip/connection.py
@@ -11,15 +11,13 @@
 
 
 class Connection(BaseConnection):
-    def init(self, accesspoint_id, lookup=True, **kwargs):
+    def init(self, accesspoint_id, lookup=True, lookup_url="https://lookup.homematic.com:48335/getHost",**kwargs):
         self.set_token_and_characteristics(accesspoint_id)
 
         if lookup:
             while True:
                 try:
-                    result = requests.post(
-                        "https://lookup.homematic.com:48335/getHost",
-                        json=self.clientCharacteristics, timeout=3)
+                    result = requests.post(lookup_url, json=self.clientCharacteristics, timeout=3)
                     js = json.loads(result.text)
                     self._urlREST = js["urlREST"]
                     self._urlWebSocket = js["urlWebSocket"]