tests/oom-platform-cert-service/truststoremerger/libraries/TrustMergerManager.py

   1 import docker
   2 import os
   3 import shutil
   4 from EnvsReader import EnvsReader
   5 from docker.types import Mount
   6
   7 ARCHIVES_PATH = os.getenv("WORKSPACE") + "/archives/"
   8
   9
  10 class TrustMergerManager:
  11
  12   def __init__(self, mount_path, truststores_path):
  13     self.mount_path = mount_path
  14     self.truststores_path = truststores_path
  15
  16   def run_merger_container(self, merger_image, merger_name, path_to_env):
  17     self.remove_mount_dir()
  18     shutil.copytree(self.truststores_path, self.mount_path)
  19     client = docker.from_env()
  20     environment = EnvsReader().read_env_list_from_file(path_to_env)
  21     container = client.containers.run(
  22         image=merger_image,
  23         name=merger_name,
  24         environment=environment,
  25         user='root',  # Run container as root to avoid permission issues with volume mount access
  26         mounts=[Mount(target='/var/certs', source=self.mount_path, type='bind')],
  27         detach=True
  28     )
  29     exitcode = container.wait()
  30     return exitcode
  31
  32   def create_mount_dir(self):
  33     if not os.path.exists(self.mount_path):
  34       os.makedirs(self.mount_path)
  35
  36   def remove_mount_dir(self):
  37     if os.path.exists(self.mount_path):
  38       shutil.rmtree(self.mount_path)
  39
  40   def remove_merger_container_and_save_logs(self, container_name, log_file_name):
  41     client = docker.from_env()
  42     container = client.containers.get(container_name)
  43     text_file = open(ARCHIVES_PATH + "merger_container_" + log_file_name + ".log", "w")
  44     text_file.write(container.logs())
  45     text_file.close()
  46     container.remove()
  47     self.remove_mount_dir()