示例#1
0
 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