def test_remove_node(self):
cloud_provider = BotoCloudProvider(config_cloud_ec2_url, config_cloud_ec2_region, config_cloud_ec2_access_key, config_cloud_ec2_secret_key)
setup_provider = AnsibleSetupProvider(config_login_user_key_private, config_login_image_user, config_login_image_user_sudo, config_login_image_sudo, config_setup_playbook_path, config_setup_frontend_groups, config_setup_compute_groups)
cluster = Cluster(config_cluster_name, config_cloud_name, cloud_provider, setup_provider, 1, 2, Configurator())
frontend_amount = len(cluster.frontend_nodes)
frontend_node = cluster.frontend_nodes[-1]
frontend_node_name = frontend_node.name
cluster.remove_node(frontend_node)
assert frontend_amount == (len(cluster.frontend_nodes) + 1)
for node in cluster.frontend_nodes:
assert node.name != frontend_node_name
compute_amount = len(cluster.compute_nodes)
compute_node = cluster.compute_nodes[-1]
compute_node_name = compute_node.name
cluster.remove_node(compute_node)
assert compute_amount == (len(cluster.compute_nodes) + 1)
for node in cluster.compute_nodes:
assert node.name != compute_node_name