add probes and support newer Python versions

Signed-off-by: Sylvain Hellegouarch <sh@defuze.org>

.github/workflows/build-and-test.yaml

@@ -27,7 +27,7 @@ jobs:
  runs-on: ubuntu-22.04
  strategy:
  matrix:
- python-version: [3.7, 3.8, 3.9, "3.10"]
+ python-version: [3.7, 3.8, 3.9, "3.10", "3.11", "3.12"]
  steps:
  - uses: actions/checkout@v3
  - name: Set up Python ${{ matrix.python-version }}

CHANGELOG.md

@@ -2,7 +2,16 @@
 
 ## [Unreleased][]
 
-[Unreleased]: https://github.com/chaostoolkit/chaostoolkit-kubernetes/compare/0.32.1...HEAD
+[Unreleased]: https://github.com/chaostoolkit/chaostoolkit-kubernetes/compare/0.33.0...HEAD
+
+## [0.33.0][] - 2023-10-29
+
+[0.33.0]: https://github.com/chaostoolkit/chaostoolkit-kubernetes/compare/0.32.1...0.33.0
+
+### Added
+
+* Probes to retrieve and verify node statuses
+* Advertize support for Python 3.11 and Python 3.12
 
 ## [0.32.1][] - 2023-10-28
 

chaosk8s/__init__.py

@@ -12,7 +12,7 @@
 from logzero import logger
 
 __all__ = ["create_k8s_api_client", "discover", "__version__"]
-__version__ = "0.32.1"
+__version__ = "0.33.0"
 
 
 def get_config_path() -> str:

chaosk8s/node/probes.py

@@ -1,11 +1,17 @@
 import json
+from typing import Dict, List
 
 from chaoslib.types import Configuration, Secrets
 from kubernetes import client
+from logzero import logger
 
 from chaosk8s import create_k8s_api_client
 
-__all__ = ["get_nodes"]
+__all__ = [
+ "get_nodes",
+ "all_nodes_must_be_ready_to_schedule",
+ "get_all_node_status_conditions",
+]
 
 
 def get_nodes(
@@ -26,3 +32,43 @@ def get_nodes(
  ret = v1.list_node(_preload_content=False)
 
  return json.loads(ret.read().decode("utf-8"))
+
+
+def get_all_node_status_conditions(
+ configuration: Configuration = None, secrets: Secrets = None
+) -> List[Dict[str, str]]:
+ """
+ Get all nodes conditions.
+ """
+ api = create_k8s_api_client(secrets)
+
+ v1 = client.CoreV1Api(api)
+ nodes = v1.list_node()
+
+ result = []
+
+ for node in nodes.items:
+ n = {"name": node.metadata.name}
+ for cond in node.status.conditions:
+ n[cond.type] = cond.status
+ result.append(n)
+
+ logger.debug(f"Nodes statuses: {result}")
+
+ return result
+
+
+def all_nodes_must_be_ready_to_schedule(
+ configuration: Configuration = None, secrets: Secrets = None
+) -> bool:
+ """
+ Verifies that all nodes in the cluster are in `Ready` condition and can
+ be scheduled.
+ """
+ result = get_all_node_status_conditions(configuration, secrets)
+
+ for statuses in result:
+ if statuses.get("Ready") != "True":
+ return False
+
+ return True

setup.py

@@ -38,6 +38,8 @@ def get_version_from_package() -> str:
  'Programming Language :: Python :: 3.8',
  'Programming Language :: Python :: 3.9',
  'Programming Language :: Python :: 3.10',
+ 'Programming Language :: Python :: 3.11',
+ 'Programming Language :: Python :: 3.12',
  'Programming Language :: Python :: Implementation',
  'Programming Language :: Python :: Implementation :: CPython'
 ]