frostfs-testcases/pytest_tests/testsuites/container/test_container.py

import json

import allure
import pytest
from epoch import tick_epoch
from python_keywords.container import (
    create_container,
    delete_container,
    get_container,
    list_containers,
    wait_for_container_creation,
    wait_for_container_deletion,
)
from utility import placement_policy_from_container
from wellknown_acl import PRIVATE_ACL_F


@pytest.mark.parametrize("name", ["", "test-container"], ids=["No name", "Set particular name"])
@pytest.mark.sanity
@pytest.mark.container
def test_container_creation(prepare_wallet_and_deposit, name):
    scenario_title = f"with name {name}" if name else "without name"
    allure.dynamic.title(f"User can create container {scenario_title}")

    wallet = prepare_wallet_and_deposit
    with open(wallet) as file:
        json_wallet = json.load(file)

    placement_rule = "REP 2 IN X CBF 1 SELECT 2 FROM * AS X"
    cid = create_container(wallet, rule=placement_rule, name=name)

    containers = list_containers(wallet)
    assert cid in containers, f"Expected container {cid} in containers: {containers}"

    container_info: str = get_container(wallet, cid, json_mode=False)
    container_info = container_info.casefold()  # To ignore case when comparing with expected values

    info_to_check = {
        f"basic ACL: {PRIVATE_ACL_F} (private)",
        f"owner ID: {json_wallet.get('accounts')[0].get('address')}",
        f"container ID: {cid}",
    }
    if name:
        info_to_check.add(f"Name={name}")

    with allure.step("Check container has correct information"):
        expected_policy = placement_rule.casefold()
        actual_policy = placement_policy_from_container(container_info)
        assert (
            actual_policy == expected_policy
        ), f"Expected policy\n{expected_policy} but got policy\n{actual_policy}"

        for info in info_to_check:
            expected_info = info.casefold()
            assert (
                expected_info in container_info
            ), f"Expected {expected_info} in container info:\n{container_info}"

    with allure.step("Delete container and check it was deleted"):
        delete_container(wallet, cid)
        tick_epoch()
        wait_for_container_deletion(wallet, cid)


@allure.title("Parallel container creation and deletion")
@pytest.mark.sanity
@pytest.mark.container
def test_container_creation_deletion_parallel(prepare_wallet_and_deposit):
    containers_count = 3
    wallet = prepare_wallet_and_deposit
    placement_rule = "REP 2 IN X CBF 1 SELECT 2 FROM * AS X"

    cids: list[str] = []
    with allure.step(f"Create {containers_count} containers"):
        for _ in range(containers_count):
            cids.append(
                create_container(
                    wallet, rule=placement_rule, await_mode=False, wait_for_creation=False
                )
            )

    with allure.step(f"Wait for containers occur in container list"):
        for cid in cids:
            wait_for_container_creation(wallet, cid, sleep_interval=containers_count)

    with allure.step("Delete containers and check they were deleted"):
        for cid in cids:
            delete_container(wallet, cid)
        tick_epoch()
        wait_for_container_deletion(wallet, cid)