fix(api): set Referer and Origin headers to make polling work

Author: palazzem

src/elmo/api/client.py

@@ -46,6 +46,13 @@ def __init__(self, base_url=None, domain=None, session_id=None):
         self._router = Router(base_url)
         self._domain = domain
         self._session = Session()
+        self._session.headers.update(
+            {
+                "Referer": "https://webservice.elmospa.com/",
+                "Origin": "https://webservice.elmospa.com",
+            }
+        )
+
         self._session_id = session_id
         self._panel = None
         self._lock = Lock()

tests/test_client.py

@@ -2657,3 +2657,14 @@ def test_client_query_last_id_value_error_empty(server, mocker):
     # Test
     inputs = client.query(query.INPUTS)
     assert inputs["last_id"] == 0
+
+
+def test_client_constructor_sets_required_headers():
+    """Should set required Referer and Origin headers during initialization otherwise
+    polling will fail.
+    Regression test for: https://github.com/palazzem/econnect-python/issues/158
+    """
+    # Test with default base URL
+    client_default = ElmoClient()
+    assert client_default._session.headers["Referer"] == "https://webservice.elmospa.com/"
+    assert client_default._session.headers["Origin"] == "https://webservice.elmospa.com"