from cluster_migration_core.core.framework_step import FrameworkStep
from cluster_migration_core.cluster_management.cluster import Cluster, ClusterNotStartedInTimeException


class StartSourceCluster(FrameworkStep):
    """
    This step starts the source cluster and ensures it is healthy
    """

    def _run(self):
        # Get the state we need
        docker_client = self.state.docker_client
        shared_volume = self.state.shared_volume
        source_cluster_config = self.state.test_config.clusters_def.source

        # Begin the step body
        self.logger.info("Creating source cluster...")
        cluster = Cluster("source-cluster", source_cluster_config, docker_client, shared_volume=shared_volume)
        cluster.start()

        try:
            max_wait_sec = 30
            self.logger.info(f"Waiting up to {max_wait_sec} sec for cluster to be active...")
            cluster.wait_for_cluster_to_start_up(max_wait_time_sec=30)
        except ClusterNotStartedInTimeException as exception:
            self.fail("Cluster did not start within time limit", exception)
        self.logger.info(f"Cluster {cluster.name} is active")

        # Update our state
        self.state.source_cluster = cluster