def test_validate_good_cluster(self):
client_workers = [
ClientWorker('10.0.0.0', 'n1-standard-16', 'europe-west4-a'),
ClientWorker('10.0.0.1', 'n1-standard-16', 'europe-west4-a'),
ClientWorker('10.0.0.2', 'n1-standard-16', 'europe-west4-a'),
ClientWorker('10.0.0.3',
'n1-standard-16',
'europe-west4-a',
hostname='test'),
]
service_workers = [
ServiceWorker('10.0.0.0', 'v3-32', 'europe-west4-a'),
ServiceWorker('10.0.0.1', 'v3-32', 'europe-west4-a'),
ServiceWorker('10.0.0.2', 'v3-32', 'europe-west4-a'),
ServiceWorker('10.0.0.3', 'v3-32', 'europe-west4-a'),
]
cluster = Cluster(client_workers, service_workers)
cluster.validate() # Does not raise exception
def test_validate_machine_type_client_cluster(self):
client_workers = [
ClientWorker('10.0.0.0', 'n1-standard-16', 'europe-west4-a'),
ClientWorker('10.0.0.1', 'n1-standard-8', 'europe-west4-a'),
]
service_workers = [
ServiceWorker('10.0.0.0', 'v3-8', 'europe-west4-a'),
ServiceWorker('10.0.0.1', 'v3-8', 'europe-west4-a'),
]
no_check_cluster = Cluster(client_workers,
service_workers,
check_client_machine_type=False)
no_check_cluster.validate() # Does not raise exception
check_cluster = Cluster(client_workers, service_workers)
self.assertRaisesRegex(
RuntimeError, 'All client_workers must have the same machine_type',
check_cluster.validate)
