def test_delete_servers_by_name(self, delete_hosts_mock, admin_mock,
kill_node_mock):
"""Test deleting servers by name."""
admin_srv_mock = admin_mock.server.return_value
autoscale.delete_servers_by_name([
'test-partition-dq2opb2qrfj.foo.com',
'test-partition-dq2opbqskkq.foo.com',
'test-partition-dq2opc7ao37.foo.com',
])
delete_hosts_mock.assert_called_once_with(
ipa_client=mock.ANY,
ec2_conn=mock.ANY,
hostnames=[
'test-partition-dq2opb2qrfj.foo.com',
'test-partition-dq2opbqskkq.foo.com',
'test-partition-dq2opc7ao37.foo.com',
],
)
admin_srv_mock.delete.assert_has_calls([
mock.call('test-partition-dq2opb2qrfj.foo.com'),
mock.call('test-partition-dq2opbqskkq.foo.com'),
mock.call('test-partition-dq2opc7ao37.foo.com'),
])
kill_node_mock.assert_has_calls([
mock.call(mock.ANY, 'test-partition-dq2opb2qrfj.foo.com'),
mock.call(mock.ANY, 'test-partition-dq2opbqskkq.foo.com'),
mock.call(mock.ANY, 'test-partition-dq2opc7ao37.foo.com'),
])
def autoscale_cmd(timeout, max_count, min_count, batch_count,
app_srv_ratio):
"""Autoscale Treadmill cell based on scheduler queue."""
while True:
create_cnt, extra_servers = autoscale.scale(
max_servers=max_count,
min_servers=min_count,
default_app_srv_ratio=app_srv_ratio,
max_batch=batch_count)
if create_cnt > 0:
autoscale.create_n_servers(create_cnt, partition=None)
if extra_servers:
autoscale.delete_servers_by_name(extra_servers)
time.sleep(timeout)
def delete_cmd(servers):
"""Delete servers by name."""
autoscale.delete_servers_by_name(servers)