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