import allure
import epoch
import pytest
from cluster import Cluster
from frostfs_testlib.shell import Shell


# To skip adding every mandatory singleton dependency to EACH test function
class ClusterTestBase:
    shell: Shell
    cluster: Cluster

    @pytest.fixture(scope="session", autouse=True)
    def fill_mandatory_dependencies(self, cluster: Cluster, client_shell: Shell):
        ClusterTestBase.shell = client_shell
        ClusterTestBase.cluster = cluster
        yield

    @allure.title("Tick {epochs_to_tick} epochs")
    def tick_epochs(self, epochs_to_tick: int):
        for _ in range(epochs_to_tick):
            self.tick_epoch()

    def tick_epoch(self):
        epoch.tick_epoch(self.shell, self.cluster)

    def wait_for_epochs_align(self):
        epoch.wait_for_epochs_align(self.shell, self.cluster)

    def get_epoch(self):
        return epoch.get_epoch(self.shell, self.cluster)

    def ensure_fresh_epoch(self):
        return epoch.ensure_fresh_epoch(self.shell, self.cluster)