TLS sdc-be-init: truststore & keystore handling

[sdc.git] / catalog-be / src / main / resources / scripts / sdcBePy / consumers / run.py

diff --git a/catalog-be/src/main/resources/scripts/sdcBePy/consumers/run.py b/catalog-be/src/main/resources/scripts/sdcBePy/consumers/run.py
index 59a0b61..6f5cbe7 100644 (file)
--- a/catalog-be/src/main/resources/scripts/sdcBePy/consumers/run.py
+++ b/catalog-be/src/main/resources/scripts/sdcBePy/consumers/run.py
@@ -1,14 +1,15 @@
 import time
 
-from sdcBePy.common.healthCheck import check_backend, RETRY_ATTEMPTS, get_args
+from sdcBePy import properties
+from sdcBePy.common.healthCheck import check_backend, get_args
 from sdcBePy.common.sdcBeProxy import SdcBeProxy
 from sdcBePy.consumers.models.consumerCandidateList import get_consumers
 from sdcBePy.users.run import colors
 
 
-def be_consumers_init(be_ip, be_port, protocol, consumer_candidate_list):
-    sdc_be_proxy = SdcBeProxy(be_ip, be_port, protocol)
-    if check_backend(sdc_be_proxy, RETRY_ATTEMPTS):
+def be_consumers_init(be_ip, be_port, header, protocol, tls_cert, tls_key, tls_key_pw, ca_cert, consumer_candidate_list):
+    sdc_be_proxy = SdcBeProxy(be_ip, be_port, header, protocol, tls_cert, tls_key, tls_key_pw, ca_cert)
+    if check_backend(sdc_be_proxy, properties.retry_attempts):
         for consumer in consumer_candidate_list:
             if sdc_be_proxy.check_user(consumer.consumer_name) != 200:
                 result = sdc_be_proxy.create_consumer(*consumer.get_parameters())
@@ -27,8 +28,8 @@ def be_consumers_init(be_ip, be_port, protocol, consumer_candidate_list):
 
 
 def main():
-    be_ip, be_port, protocol = get_args()
-    be_consumers_init(be_ip, be_port, protocol, get_consumers())
+    be_ip, be_port, header, protocol, tls_key, tls_cert, tls_key_pw, ca_cert = get_args()
+    be_consumers_init(be_ip, be_port, header, protocol, tls_cert, tls_key, tls_key_pw, ca_cert, get_consumers())
 
 
 if __name__ == '__main__':