frostfs-testcases/pytest_tests/testsuites/services/test_binaries.py

import logging
from http import HTTPStatus
from re import match

import allure
import pytest
import requests

from common import BIN_VERSIONS_FILE
from env_properties import read_env_properties, save_env_properties
from service_helper import get_storage_service_helper

logger = logging.getLogger('NeoLogger')


@allure.title('Check binaries versions')
@pytest.mark.check_binaries
def test_binaries_versions(request):
    """
    Compare binaries versions from external source (url) and deployed on servers.
    """
    if not BIN_VERSIONS_FILE:
        pytest.skip('File with binaries and versions was not provided')

    binaries_to_check = download_versions_info(BIN_VERSIONS_FILE)
    with allure.step('Get binaries versions from servers'):
        helper = get_storage_service_helper()
        got_versions = helper.get_binaries_version(binaries=list(binaries_to_check.keys()))

    env_properties = read_env_properties(request.config)

    # compare versions from servers and file
    failed_versions = {}
    additional_env_properties = {}
    for binary, version in binaries_to_check.items():
        actual_version = got_versions.get(binary)
        if actual_version != version:
            failed_versions[binary] = f'Expected version {version}, found version {actual_version}'

        # If some binary was not listed in the env properties file, let's add it
        # so that we have full information about versions in allure report
        if env_properties and binary not in env_properties:
            additional_env_properties[binary] = actual_version

    if env_properties and additional_env_properties:
        save_env_properties(request.config, additional_env_properties)

    # create clear beautiful error with aggregation info
    if failed_versions:
        msg = '\n'.join({f'{binary}: {error}' for binary, error in failed_versions.items()})
        raise AssertionError(f'Found binaries with unexpected versions:\n{msg}')


@allure.step('Download versions info from {url}')
def download_versions_info(url: str) -> dict:
    binaries_to_version = {}

    response = requests.get(url)

    assert response.status_code == HTTPStatus.OK, \
        f'Got {response.status_code} code. Content {response.json()}'

    content = response.text
    assert content, f'Expected file with content, got {response}'

    for line in content.split('\n'):
        m = match('(.*)=(.*)', line)
        if not m:
            logger.warning(f'Could not get binary/version from {line}')
            continue
        bin_name, bin_version = m.group(1), m.group(2)
        binaries_to_version[bin_name] = bin_version

    return binaries_to_version