import dotmap

from src import suspicions
from src.suspicions import cluster
from . import base


class ClusterCase(base.BaseCase):
    def __init__(self, cluster_id: str, project_id: str, tenant_id: str, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.cluster_id = cluster_id
        self.project_id = project_id
        self.tenant_id = tenant_id

    def build_ctx(self) -> dotmap.DotMap:
        return dotmap.DotMap({
            "cluster_id": self.cluster_id,
            "project_id": self.project_id,
            "tenant_id": self.tenant_id,
        })


class TheProjectsFirstInfraClusterCreationDurationTooLong(ClusterCase):
    def get_suspicions(self) -> list[type(suspicions.SuspicionABC)]:
        return [cluster.ClusterNotReady]


class InfraClusterScalingDurationMoreThan1h(ClusterCase):
    def get_suspicions(self) -> list[type(suspicions.SuspicionABC)]:
        return [cluster.ClusterNotReady]