FIX onramp_simulator

CompassionCH · Mar 11, 2024 · ced1909 · ced1909
1 parent 79f6f14
commit ced1909
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/message_center_compassion/controllers/onramp.py b/message_center_compassion/controllers/onramp.py
@@ -22,7 +22,7 @@
 # Put any authorized sender here. Its address must be part of the headers
 # in order to handle a request.
 AUTHORIZED_SENDERS = [
-    "CHTest",
+    "OnrampSimulator",
     "CISalesforce",
     "CISFDC",
     "CINetsuite",

diff --git a/message_center_compassion/tests/onramp_base_test.py b/message_center_compassion/tests/onramp_base_test.py
@@ -30,7 +30,7 @@ def setUp(self):
             "Authorization": "Bearer fake_token",
             "x-cim-MessageType": "http://schemas.ci.org/ci/services/"
             "communications/2015/09/SBCStructured",
-            "x-cim-FromAddress": "CHTest",
+            "x-cim-FromAddress": "OnrampSimulator",
             "x-cim-ToAddress": "CH",
         }
         self.opener.headers.update(headers)

diff --git a/onramp_simulator/tools/onramp_connector.py b/onramp_simulator/tools/onramp_connector.py
@@ -20,11 +20,12 @@ def test_message(self, test_message):
         """Sends a message to any onramp.
         :param test_message (onramp.simulator record): the message to send
         """
+        country_code = self.env["res.company"]._get_main_company().country_id.code
         headers = {
             "Content-type": "application/json",
             "x-cim-MessageType": test_message.message_type_url,
-            "x-cim-FromAddress": "CHTest",
-            "x-cim-ToAddress": "SC",
+            "x-cim-FromAddress": "OnrampSimulator",
+            "x-cim-ToAddress": country_code,
         }
         url = test_message.server_url
         body = test_message.body_json