diff --git a/Makefile b/Makefile index 990ddbd..e0161a7 100644 --- a/Makefile +++ b/Makefile @@ -7,16 +7,6 @@ SHELL = bash OUTPUT_DIR = artifacts/ KEYWORDS_REPO = git@github.com:nspcc-dev/neofs-keywords.git VENVS = $(shell ls -1d venv/*/ | sort -u | xargs basename -a) -ROOT_DIR := $(shell dirname $(realpath $(firstword $(MAKEFILE_LIST)))) -DEV_IMAGE_PY ?= registry.spb.yadro.com/tools/pytest-neofs-x86_64:7 -SETUP_DIR ?= $(CURDIR)/.setup -DEV_ENV_DEPLOY_DIR ?= /opt/dev-env - -DOCKER_NETWORK = --network host -ifeq ($(shell uname -s),Darwin) - DOCKER_NETWORK = --network bridge -p 389:389 -p 636:636 -endif - .PHONY: all all: venvs @@ -42,7 +32,6 @@ clean: pytest-local: @echo "⇒ Run Pytest" - export PYTHONPATH=$(ROOT_DIR)/neofs-keywords/lib:$(ROOT_DIR)/neofs-keywords/robot:$(ROOT_DIR)/robot/resources/lib:$(ROOT_DIR)/robot/resources/lib/python_keywords:$(ROOT_DIR)/robot/variables && \ python -m pytest pytest_tests/testsuites/ help: diff --git a/pytest_tests/helpers/sbercloud_helper.py b/pytest_tests/helpers/sbercloud_helper.py index a5fcf45..3f41fac 100644 --- a/pytest_tests/helpers/sbercloud_helper.py +++ b/pytest_tests/helpers/sbercloud_helper.py @@ -1,41 +1,52 @@ import json +import os from dataclasses import dataclass +from typing import Optional import requests -from yaml import FullLoader -from yaml import load as yaml_load +import yaml @dataclass -class SberCloudCtx: - sber_login: str = None - sber_password: str = None - sber_domain: str = None - sber_project_id: str = None - sber_iam_url: str = None - sber_ecss: list = None +class SberCloudConfig: + login: Optional[str] = None + password: Optional[str] = None + domain: Optional[str] = None + project_id: Optional[str] = None + iam_url: Optional[str] = None @staticmethod - def from_dict(sbercloud_dict: dict) -> 'SberCloudCtx': - return SberCloudCtx(**sbercloud_dict) + def from_dict(config_dict: dict) -> 'SberCloudConfig': + return SberCloudConfig(**config_dict) @staticmethod - def from_yaml(config: str) -> 'SberCloudCtx': - with open(config) as yaml_file: - config_from_yaml = yaml_load(yaml_file, Loader=FullLoader) - return SberCloudCtx.from_dict(config_from_yaml) + def from_yaml(config_path: str) -> 'SberCloudConfig': + with open(config_path) as file: + config_dict = yaml.load(file, Loader=yaml.FullLoader) + return SberCloudConfig.from_dict(config_dict["sbercloud"]) + + @staticmethod + def from_env() -> 'SberCloudConfig': + config_dict = { + "domain": os.getenv("SBERCLOUD_DOMAIN"), + "login": os.getenv("SBERCLOUD_LOGIN"), + "password": os.getenv("SBERCLOUD_PASSWORD"), + "project_id": os.getenv("SBERCLOUD_PROJECT_ID"), + "iam_url": os.getenv("SBERCLOUD_IAM_URL"), + } + return SberCloudConfig.from_dict(config_dict) class SberCloud: - def __init__(self, config: str): - self.sbercloud_config = SberCloudCtx().from_yaml(config) + def __init__(self, config: SberCloudConfig) -> None: + self.config = config self.ecs_url = None self.project_id = None self.token = None - self.update_token() - self.ecss = self.get_ecss() + self._initialize() + self.ecs_nodes = self.get_ecs_nodes() - def update_token(self): + def _initialize(self) -> None: data = { 'auth': { 'identity': { @@ -43,68 +54,85 @@ class SberCloud: 'password': { 'user': { 'domain': { - 'name': self.sbercloud_config.sber_domain + 'name': self.config.domain }, - 'name': self.sbercloud_config.sber_login, - 'password': self.sbercloud_config.sber_password + 'name': self.config.login, + 'password': self.config.password } } }, 'scope': { 'project': { - 'id': self.sbercloud_config.sber_project_id + 'id': self.config.project_id } } } } - response = requests.post(f'{self.sbercloud_config.sber_iam_url}/v3/auth/tokens', data=json.dumps(data), - headers={'Content-Type': 'application/json'}) - self.ecs_url = [catalog['endpoints'][0]['url'] - for catalog in response.json()['token']['catalog'] if catalog['type'] == 'ecs'][0] + response = requests.post( + f'{self.config.iam_url}/v3/auth/tokens', + data=json.dumps(data), + headers={'Content-Type': 'application/json'} + ) + self.ecs_url = [ + catalog['endpoints'][0]['url'] + for catalog in response.json()['token']['catalog'] if catalog['type'] == 'ecs' + ][0] self.project_id = self.ecs_url.split('/')[-1] self.token = response.headers['X-Subject-Token'] - def find_esc_by_ip(self, ip: str, update: bool = False) -> str: - if not self.ecss or update: - self.ecss = self.get_ecss() - ecss = [ecs for ecs in self.ecss if ip in [ - ecs_ip['addr'] for ecs_ip in [ecs_ip for ecs_ips in ecs['addresses'].values() for ecs_ip in ecs_ips]]] - assert len(ecss) == 1 - return ecss[0]['id'] + def find_ecs_node_by_ip(self, ip: str, no_cache: bool = False) -> str: + if not self.ecs_nodes or no_cache: + self.ecs_nodes = self.get_ecs_nodes() + nodes_by_ip = [ + node for node in self.ecs_nodes + if ip in [ + node_ip['addr'] + for node_ips in node['addresses'].values() + for node_ip in node_ips + ] + ] + assert len(nodes_by_ip) == 1 + return nodes_by_ip[0]['id'] - def get_ecss(self) -> [dict]: + def get_ecs_nodes(self) -> list[dict]: response = requests.get(f'{self.ecs_url}/cloudservers/detail', headers={'X-Auth-Token': self.token}).json() return response['servers'] - def start_node(self, node_id: str = None, node_ip: str = None): + def start_node(self, node_id: Optional[str] = None, node_ip: Optional[str] = None) -> None: data = { 'os-start': { 'servers': [ { - 'id': node_id or self.find_esc_by_ip(node_ip) + 'id': node_id or self.find_ecs_node_by_ip(node_ip) } ] } } - response = requests.post(f'{self.ecs_url}/cloudservers/action', - data=json.dumps(data), - headers={'Content-Type': 'application/json', 'X-Auth-Token': self.token}) - assert response.status_code < 300, f'Status:{response.status_code}. Server not started: {response.json()}' + response = requests.post( + f'{self.ecs_url}/cloudservers/action', + data=json.dumps(data), + headers={'Content-Type': 'application/json', 'X-Auth-Token': self.token} + ) + assert response.status_code < 300, \ + f'Status:{response.status_code}. Server not started: {response.json()}' - def stop_node(self, node_id: str = None, node_ip: str = None, hard: bool = False): + def stop_node(self, node_id: Optional[str] = None, node_ip: Optional[str] = None, + hard: bool = False) -> None: data = { 'os-stop': { 'type': 'HARD' if hard else 'SOFT', 'servers': [ { - 'id': node_id or self.find_esc_by_ip(node_ip) + 'id': node_id or self.find_ecs_node_by_ip(node_ip) } - ] } } - response = requests.post(f'{self.ecs_url}/cloudservers/action', - data=json.dumps(data), - headers={'Content-Type': 'application/json', 'X-Auth-Token': self.token}) - assert response.status_code < 300, f'Status:{response.status_code}. Server not stopped: {response.json()}' + response = requests.post( + f'{self.ecs_url}/cloudservers/action', + data=json.dumps(data), + headers={'Content-Type': 'application/json', 'X-Auth-Token': self.token} + ) + assert response.status_code < 300, \ + f'Status:{response.status_code}. Server not stopped: {response.json()}' diff --git a/pytest_tests/testsuites/failovers/test_failover_storage.py b/pytest_tests/testsuites/failovers/test_failover_storage.py index 50458e8..18e11bd 100644 --- a/pytest_tests/testsuites/failovers/test_failover_storage.py +++ b/pytest_tests/testsuites/failovers/test_failover_storage.py @@ -1,14 +1,14 @@ import logging -import os import allure import pytest -from common import STORAGE_NODE_SSH_PRIVATE_KEY_PATH, STORAGE_NODE_SSH_USER, STORAGE_NODE_SSH_PASSWORD +from common import (STORAGE_NODE_SSH_PRIVATE_KEY_PATH, STORAGE_NODE_SSH_USER, + STORAGE_NODE_SSH_PASSWORD) from python_keywords.container import create_container from python_keywords.neofs_verbs import get_object, put_object from python_keywords.utility_keywords import generate_file, get_file_hash -from sbercloud_helper import SberCloud +from sbercloud_helper import SberCloud, SberCloudConfig from ssh_helper import HostClient, HostIsNotAvailable from wellknown_acl import PUBLIC_ACL from .failover_utils import wait_all_storage_node_returned, wait_object_replication_on_nodes @@ -21,7 +21,8 @@ stopped_hosts = [] def sbercloud_client(): with allure.step('Connect to SberCloud'): try: - yield SberCloud(f'{os.getcwd()}/configuration/sbercloud.yaml') + config = SberCloudConfig.from_env() + yield SberCloud(config) except Exception as err: pytest.fail(f'SberCloud infrastructure not available. Error\n{err}') @@ -42,7 +43,7 @@ def panic_reboot_host(ip: str = None): ssh.exec('sudo sh -c "echo b > /proc/sysrq-trigger"', timeout=1) -def return_all_storage_nodes(sbercloud_client: SberCloud): +def return_all_storage_nodes(sbercloud_client: SberCloud) -> None: for host in list(stopped_hosts): with allure.step(f'Start storage node {host}'): sbercloud_client.start_node(node_ip=host.split(':')[-2])