aboutsummaryrefslogtreecommitdiffstats
path: root/tests/dcaegen2-collectors-hv-ves/testcases/libraries/VesHvContainersUtilsLibrary.py
blob: 989a796ce5f9d545bddc74fca2a38d2b8adbcb3d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
from time import time

from robot.api import logger
import os.path
import docker
from io import BytesIO
from os.path import basename
from tarfile import TarFile, TarInfo

LOCALHOST = "localhost"


class VesHvContainersUtilsLibrary:

    def get_consul_api_access_url(self, method, image_name, port):
        return self.create_url(
            method,
            self.get_instance_address(image_name, port)
        )

    def get_xnf_sim_api_access_url(self, method, host):
        if is_running_inside_docker():
            return self.create_url(method, host)
        else:
            logger.info("File `/.dockerenv` not found. Assuming local environment and using localhost.")
            port_from_container_name = str(host)[-4:]
            return self.create_url(method, LOCALHOST + ":" + port_from_container_name)

    def get_dcae_app_api_access_url(self, method, image_name, port):
        return self.create_url(
            method,
            self.get_instance_address(image_name, port)
        )

    def get_instance_address(self, image_name, port):
        if is_running_inside_docker():
            return image_name + ":" + port
        else:
            logger.info("File `/.dockerenv` not found. Assuming local environment and using localhost.")
            return LOCALHOST + ":" + port

    def create_url(self, method, host_address):
        return method + host_address

def is_running_inside_docker():
    return os.path.isfile("/.dockerenv")

def copy_to_container(container_id, filepaths, path='/etc/ves-hv'):
    with create_archive(filepaths) as archive:
        docker.APIClient('unix:///var/run/docker.sock') \
            .put_archive(container=container_id, path=(path), data=archive)


def create_archive(filepaths):
    tarstream = BytesIO()
    tarfile = TarFile(fileobj=tarstream, mode='w')
    for filepath in filepaths:
        file = open(filepath, 'r')
        file_data = file.read()

        tarinfo = TarInfo(name=basename(file.name))
        tarinfo.size = len(file_data)
        tarinfo.mtime = time()

        tarfile.addfile(tarinfo, BytesIO(file_data))

    tarfile.close()
    tarstream.seek(0)
    return tarstream