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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
from VesHvContainersUtilsLibrary import copy_to_container
import HttpRequests
import os
import docker
from robot.api import logger
from time import sleep
XNF_SIMULATOR_NAME = "xNF Simulator"
SIMULATOR_IMAGE_NAME = "onap/org.onap.dcaegen2.collectors.hv-ves.hv-collector-xnf-simulator"
SIMULATOR_IMAGE_FULL_NAME = os.getenv("DOCKER_REGISTRY") + "/" + SIMULATOR_IMAGE_NAME + ":latest"
WORKSPACE_ENV = os.getenv("WORKSPACE")
certificates_dir_path = WORKSPACE_ENV + "/test/csit/plans/dcaegen2-collectors-hv-ves/testsuites/ssl/"
collector_certs_lookup_dir = "/etc/ves-hv/"
ONE_SECOND_IN_NANOS = 10 ** 9
class XnfSimulatorLibrary:
def start_xnf_simulators(self, list_of_ports, valid_certs=True):
logger.info("Creating " + str(len(list_of_ports)) + " xNF Simulator containers")
dockerClient = docker.from_env()
cert_name_prefix = "" if valid_certs else "invalid_"
self.pullImageIfAbsent(dockerClient)
logger.info("Using image: " + SIMULATOR_IMAGE_FULL_NAME)
simulators_addresses = self.create_simulators(dockerClient, list_of_ports, cert_name_prefix)
self.assert_containers_startup_was_successful(dockerClient)
dockerClient.close()
return simulators_addresses
def pullImageIfAbsent(self, dockerClient):
try:
dockerClient.images.get(SIMULATOR_IMAGE_FULL_NAME)
except:
logger.console("Image " + SIMULATOR_IMAGE_FULL_NAME + " will be pulled from repository. "
"This can take a while.")
dockerClient.images.pull(SIMULATOR_IMAGE_FULL_NAME)
def create_simulators(self, dockerClient, list_of_ports, cert_name_prefix):
simulators_addresses = []
for port in list_of_ports:
container = self.run_simulator(dockerClient, port,
collector_certs_lookup_dir + cert_name_prefix + "client.crt",
collector_certs_lookup_dir + cert_name_prefix + "client.key",
collector_certs_lookup_dir + cert_name_prefix + "trust.crt"
)
logger.info("Started container: " + container.name + " " + container.id)
simulators_addresses.append(container.name + ":" + port)
return simulators_addresses
def run_simulator(self, dockerClient, port, client_crt_path, client_key_path, client_trust_store):
xNF_startup_command = ["--listen-port", port,
"--ves-host", "ves-hv-collector",
"--ves-port", "6061",
"--cert-file", client_crt_path,
"--private-key-file", client_key_path,
"--trust-cert-file", client_trust_store]
xNF_healthcheck_command = {
"interval": 5 * ONE_SECOND_IN_NANOS,
"timeout": 3 * ONE_SECOND_IN_NANOS,
"retries": 1,
"test": ["CMD", "curl", "--request", "GET",
"--fail", "--silent", "--show-error",
"localhost:" + port + "/healthcheck"]
}
logger.info("Startup command: " + str(xNF_startup_command))
logger.info("Healthcheck command: " + str(xNF_healthcheck_command))
return dockerClient.containers.run(SIMULATOR_IMAGE_FULL_NAME,
command=xNF_startup_command,
healthcheck=xNF_healthcheck_command,
detach=True,
network="ves-hv-default",
ports={port + "/tcp": port},
volumes=self.container_volumes(),
name="ves-hv-collector-xnf-simulator" + port)
def container_volumes(self):
return {certificates_dir_path: {"bind": collector_certs_lookup_dir, "mode": 'rw'}}
def assert_containers_startup_was_successful(self, dockerClient):
checks_amount = 6
check_interval_in_seconds = 5
for _ in range(checks_amount):
sleep(check_interval_in_seconds)
all_containers_healthy = True
for container in self.get_simulators_list(dockerClient):
all_containers_healthy = all_containers_healthy and self.is_container_healthy(container)
if (all_containers_healthy):
return
raise ContainerException("One of xNF simulators containers did not pass the healthcheck.")
def is_container_healthy(self, container):
container_health = container.attrs['State']['Health']['Status']
return container_health == 'healthy' and container.status == 'running'
def stop_and_remove_all_xnf_simulators(self, suite_name):
dockerClient = docker.from_env()
for container in self.get_simulators_list(dockerClient):
logger.info("Stopping and removing container: " + container.id)
log_filename = WORKSPACE_ENV + "/archives/containers_logs/" + \
suite_name.split(".")[-1] + "_" + container.name + ".log"
file = open(log_filename, "w+")
file.write(container.logs())
file.close()
container.stop()
container.remove()
dockerClient.close()
def get_simulators_list(self, dockerClient):
return dockerClient.containers.list(filters={"ancestor": SIMULATOR_IMAGE_FULL_NAME}, all=True)
def send_messages(self, simulator_url, message_filepath):
logger.info("Reading message to simulator from: " + message_filepath)
file = open(message_filepath, "rb")
data = file.read()
file.close()
logger.info("POST at: " + simulator_url)
resp = HttpRequests.session_without_env().post(simulator_url, data=data, timeout=5)
HttpRequests.checkStatusCode(resp.status_code, XNF_SIMULATOR_NAME)
class ContainerException(Exception):
def __init__(self, message):
super(ContainerException, self).__init__(message)
|
