mirror of
https://github.com/openshift/installer.git
synced 2026-02-05 06:46:36 +01:00
feebe4c70c
* instrument the manifest-generation test-framework to execute Python testfiles * test machinesets with availability zones
58 lines
1.9 KiB
Python
Executable File
58 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import unittest
|
|
|
|
import sys
|
|
import glob
|
|
import yaml
|
|
|
|
ASSETS_DIR = ""
|
|
|
|
EXPECTED_MACHINES_NUMBER = 1000
|
|
EXPECTED_ZONE_NAMES = ["zone", "ztwo", "zthree"]
|
|
|
|
|
|
class TestAZMachinesets(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_ZONE_NAMES)
|
|
self.assertNotIn(zone, found)
|
|
found.append(zone)
|
|
self.assertEqual(len(self.machinesets), len(EXPECTED_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_MACHINES_NUMBER)
|
|
|
|
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()
|