LNX-329 - Add reporting url checking

Author: homiedopie

stackify/rum.py

@@ -22,26 +22,40 @@ def insert_rum_script():
         return None
 
     transaction_id = get_transaction_id()
-
     if not transaction_id:
         return None
 
+    reporting_url = get_reporting_url()
+    if not reporting_url:
+        return None
+
+    application_name = config.application
+    if not application_name:
+        return None
+
+    environment = config.environment
+    if not environment:
+        return None
+
     settings = {
         "ID": transaction_id
     }
 
-    application_name = config.application
     if application_name:
         application_name_b64 = base64.b64encode(application_name.encode("utf-8")).decode("utf-8")
         if (application_name_b64):
             settings["Name"] = application_name_b64
 
-    environment = config.environment
     if environment:
         environment_b64 = base64.b64encode(environment.encode("utf-8")).decode("utf-8")
         if (environment_b64):
             settings["Env"] = environment_b64
 
+    if reporting_url:
+        reporting_url_b64 = base64.b64encode(reporting_url.encode("utf-8")).decode("utf-8")
+        if (reporting_url_b64):
+            settings["Trans"] = reporting_url_b64
+
     if not settings:
         return None
 
@@ -54,3 +68,6 @@ def insert_rum_script():
 
 def get_transaction_id():
     return ''
+
+def get_reporting_url():
+    return ''

tests/test_rum.py

@@ -71,9 +71,11 @@ def test_default_insert_rum_script_from_apm_without_transaction(self):
         assert not rum_data
         self.restore_apm_installed()
 
+    @mock.patch('stackify.rum.get_reporting_url')
     @mock.patch('stackify.rum.get_transaction_id')
-    def test_default_insert_rum_script(self, func):
+    def test_default_insert_rum_script(self, func, func_reporting_url):
         func.return_value = '123'
+        func_reporting_url.return_value = 'test reporting url'
         self.update_apm_installed(False)
         self.update_common_config(
             rum_key='asd',
@@ -84,7 +86,8 @@ def test_default_insert_rum_script(self, func):
         rum_settings = {
             "ID": '123',
             "Name": 'YXBw',
-            "Env": 'ZW52'
+            "Env": 'ZW52',
+            "Trans": 'dGVzdCByZXBvcnRpbmcgdXJs'
         }
 
         result = stackify.rum.insert_rum_script()
@@ -131,9 +134,11 @@ def test_default_insert_rum_script_no_details(self):
         self.reset_common_config()
         self.restore_apm_installed()
 
+    @mock.patch('stackify.rum.get_reporting_url')
     @mock.patch('stackify.rum.get_transaction_id')
-    def test_default_insert_rum_script_from_api(self, func):
+    def test_default_insert_rum_script_from_api(self, func, func_reporting_url):
         func.return_value = '123'
+        func_reporting_url.return_value = 'test reporting url'
         self.update_apm_installed(False)
         self.create_config(
             rum_key='asd1',
@@ -143,7 +148,8 @@ def test_default_insert_rum_script_from_api(self, func):
         rum_settings = {
             "ID": '123',
             "Name": 'YXBwMQ==',
-            "Env": 'ZW52MQ=='
+            "Env": 'ZW52MQ==',
+            "Trans": 'dGVzdCByZXBvcnRpbmcgdXJs'
         }
         result = stackify.rum.insert_rum_script()
         self.reset_common_config()