mirror of
https://github.com/openshift/installer.git
synced 2026-02-05 06:46:36 +01:00
95 lines
3.4 KiB
Python
Executable File
95 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import unittest
|
|
|
|
import sys
|
|
import glob
|
|
import yaml
|
|
|
|
ASSETS_DIR = ""
|
|
|
|
EXPECTED_MASTER_REPLICAS = 10
|
|
EXPECTED_MASTER_ZONE_NAMES = ["masterzone", "masterztwo", "masterzthree"]
|
|
|
|
EXPECTED_WORKER_REPLICAS = 1000
|
|
EXPECTED_WORKER_ZONE_NAMES = ["zone", "ztwo", "zthree"]
|
|
|
|
class NovaAvailabilityZonesMachines(unittest.TestCase):
|
|
def setUp(self):
|
|
"""Parse the Machines into a Python data structure."""
|
|
self.machines = []
|
|
for machine_path in glob.glob(
|
|
f'{ASSETS_DIR}/openshift/99_openshift-cluster-api_master-machines-*.yaml'
|
|
):
|
|
with open(machine_path) as f:
|
|
self.machines.append(yaml.load(f, Loader=yaml.FullLoader))
|
|
|
|
def test_zone_names(self):
|
|
"""Assert that all machines have one valid availability zone."""
|
|
for machine in self.machines:
|
|
zone = machine["spec"]["providerSpec"]["value"]["availabilityZone"]
|
|
self.assertIn(zone, EXPECTED_MASTER_ZONE_NAMES)
|
|
|
|
def test_total_instance_number(self):
|
|
"""Assert that there are as many Machines as required ControlPlane replicas."""
|
|
self.assertEqual(len(self.machines), EXPECTED_MASTER_REPLICAS)
|
|
|
|
def test_replica_distribution(self):
|
|
"""Assert that machines are evenly distributed across zones."""
|
|
zones = {}
|
|
for machine in self.machines:
|
|
zone = machine["spec"]["providerSpec"]["value"]["availabilityZone"]
|
|
zones[zone] = zones.get(zone, 0) + 1
|
|
|
|
setpoint = 0
|
|
for replicas in zones.values():
|
|
if setpoint == 0:
|
|
setpoint = replicas
|
|
else:
|
|
self.assertTrue(-2 < replicas - setpoint < 2)
|
|
|
|
|
|
class NovaAvailabilityZonesMachinesets(unittest.TestCase):
|
|
def setUp(self):
|
|
"""Parse the MachineSets into a Python data structure."""
|
|
self.machinesets = []
|
|
for machineset_path in glob.glob(
|
|
f'{ASSETS_DIR}/openshift/99_openshift-cluster-api_worker-machineset-*.yaml'
|
|
):
|
|
with open(machineset_path) as f:
|
|
self.machinesets.append(yaml.load(f, Loader=yaml.FullLoader))
|
|
|
|
def test_machineset_zone_name(self):
|
|
"""Assert that there is exactly one MachineSet per availability zone."""
|
|
found = []
|
|
for machineset in self.machinesets:
|
|
zone = machineset["spec"]["template"]["spec"]["providerSpec"][
|
|
"value"]["availabilityZone"]
|
|
self.assertIn(zone, EXPECTED_WORKER_ZONE_NAMES)
|
|
self.assertNotIn(zone, found)
|
|
found.append(zone)
|
|
self.assertEqual(len(self.machinesets), len(EXPECTED_WORKER_ZONE_NAMES))
|
|
|
|
def test_total_replica_number(self):
|
|
"""Assert that replicas spread across the MachineSets add up to the expected number."""
|
|
total_found = 0
|
|
for machineset in self.machinesets:
|
|
total_found += machineset["spec"]["replicas"]
|
|
self.assertEqual(total_found, EXPECTED_WORKER_REPLICAS)
|
|
|
|
def test_replica_distribution(self):
|
|
"""Assert that replicas are evenly distributed across machinesets."""
|
|
setpoint = 0
|
|
for machineset in self.machinesets:
|
|
replicas = machineset["spec"]["replicas"]
|
|
if setpoint == 0:
|
|
setpoint = replicas
|
|
else:
|
|
self.assertTrue(-2 < replicas - setpoint < 2)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
ASSETS_DIR = sys.argv.pop()
|
|
unittest.main(verbosity=2)
|