Use Port from Environment

[osm/tests.git] / robot-systest / resources / epa_03-crud_operations_on_sdnc_data.py

diff --git a/robot-systest/resources/epa_03-crud_operations_on_sdnc_data.py b/robot-systest/resources/epa_03-crud_operations_on_sdnc_data.py
index d195cde..827baf6 100644 (file)
--- a/robot-systest/resources/epa_03-crud_operations_on_sdnc_data.py
+++ b/robot-systest/resources/epa_03-crud_operations_on_sdnc_data.py
@@ -11,23 +11,23 @@
 #   limitations under the License.
 
 import os
-import yaml
 from pathlib import Path
 
+import yaml
+
+from common_helpers import get_prometheus_info
+
+
 # Prometheus host and port
-if os.environ.get("PROMETHEUS_HOSTNAME", False):
-    prometheus_host = os.environ.get("PROMETHEUS_HOSTNAME")
-    prometheus_port = "9090"
-else:
-    prometheus_host = os.environ.get("OSM_HOSTNAME")
-    prometheus_port = "9091"
+(prometheus_host, prometheus_port) = get_prometheus_info()
+
 
 # SDNC Configuration
 sdnc_name = "epa_03_crud_operations_on_sdnc_test"
 sdnc_user = os.environ.get("SDNC_USER")
 if not sdnc_user:
     raise Exception("'SDNC_USER' environment variable not found")
-sdnc_password  = os.environ.get("SDNC_PASSWORD")
+sdnc_password = os.environ.get("SDNC_PASSWORD")
 if not sdnc_password:
     raise Exception("'SDNC_PASSWORD' environment variable not found")
 sdnc_url = os.environ.get("SDNC_URL")