def test_add_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.add_node(Node.frontend_type) assert frontend_amount == (len(cluster.frontend_nodes) - 1) assert frontend_node == cluster.frontend_nodes[-1] assert frontend_node.type == Node.frontend_type compute_amount = len(cluster.compute_nodes) compute_node = cluster.add_node(Node.compute_type) assert compute_amount == (len(cluster.compute_nodes) - 1) assert compute_node == cluster.compute_nodes[-1] assert compute_node.type == Node.compute_type