def test_node_pod_resources_returns_all_node_pods(node_pods: int):
node = api_to_agent_node(APINodeFactory.build())
for _ in range(node_pods):
node.add_pod(api_to_agent_pod(APIPodFactory.build()))
resources = dict(node.pod_resources())
pod_resources = section.PodResources(**resources)
assert sum(len(pods) for _, pods in pod_resources) == node_pods
def test_pod_resources_from_api_pods(pods_count: int):
pods = [
APIPodFactory.build(
metadata=PodMetaDataFactory.build(name=str(i)),
status=PodStatusFactory.build(phase=api.Phase.RUNNING, ),
) for i in range(pods_count)
]
pod_resources = agent._pod_resources_from_api_pods(pods)
assert pod_resources == section.PodResources(
running=[str(i) for i in range(pods_count)], )
def pod_resources(self) -> section.PodResources:
resources: DefaultDict[str, List[str]] = defaultdict(list)
for pod in self._pods:
resources[pod.phase].append(pod.name())
return section.PodResources(**resources)
def test_node_pod_resources_one_pod_per_phase(node: agent.Node):
resources = dict(node.pod_resources())
pod_resources = section.PodResources(**resources)
for _phase, pods in pod_resources:
assert len(pods) == 1
def test_daemon_set_pod_resources_one_pod_per_phase(daemon_set):
resources = dict(daemon_set.pod_resources())
pod_resources = section.PodResources(**resources)
for _phase, pods in pod_resources:
assert len(pods) == 1
def test_daemon_set_pod_resources_returns_all_pods(daemon_set,
daemon_set_pods):
resources = dict(daemon_set.pod_resources())
pod_resources = section.PodResources(**resources)
assert sum(len(pods) for _, pods in pod_resources) == daemon_set_pods
def test_node_pod_resources_returns_all_node_pods(node, node_pods):
resources = dict(node.pod_resources())
pod_resources = section.PodResources(**resources)
assert sum(len(pods) for _, pods in pod_resources) == node_pods
def test_statefulset_pod_resources_one_pod_per_phase(statefulset: agent_kube.StatefulSet):
resources = dict(statefulset.pod_resources())
pod_resources = section.PodResources(**resources)
for _phase, pods in pod_resources:
assert len(pods) == 1
def test_statefulset_pod_resources_returns_all_pods(
statefulset: agent_kube.StatefulSet, statefulset_pods: int
):
resources = dict(statefulset.pod_resources())
pod_resources = section.PodResources(**resources)
assert sum(len(pods) for _, pods in pod_resources) == statefulset_pods
