def test_aggregates_with_names_and_ids(self, mock_aggregate, mock_detailed_aggregate): allowed_nodes = [] mock_aggregate.return_value = [ mock.Mock(id=i, name="HA_{0}".format(i)) for i in range(2) ] mock_collection = [ mock.Mock(id=i, hosts=['Node_{0}'.format(i)]) for i in range(2) ] mock_collection[0].name = 'HA_0' mock_collection[1].name = 'HA_1' mock_detailed_aggregate.side_effect = mock_collection default.DefaultScope([{ 'host_aggregates': [{ 'name': 'HA_1' }, { 'id': 0 }] }], osc=mock.Mock())._collect_aggregates( [{ 'name': 'HA_1' }, { 'id': 0 }], allowed_nodes) self.assertEqual(['Node_0', 'Node_1'], allowed_nodes)
def test_remove_nodes_from_model(self): model = self.fake_cluster.generate_scenario_1() default.DefaultScope([], osc=mock.Mock()).remove_nodes_from_model( ['Node_1', 'Node_2'], model) expected_edges = [('INSTANCE_0', 'Node_0'), ('INSTANCE_1', 'Node_0'), ('INSTANCE_6', 'Node_3'), ('INSTANCE_7', 'Node_4')] self.assertEqual(sorted(expected_edges), sorted(model.edges()))
def test_exclude_instances_with_given_metadata(self): cluster = self.fake_cluster.generate_scenario_1() instance_metadata = [{'optimize': True}] instances_to_remove = set() default.DefaultScope( [], mock.Mock(), osc=mock.Mock()).exclude_instances_with_given_metadata( instance_metadata, cluster, instances_to_remove) self.assertEqual(sorted(['INSTANCE_' + str(i) for i in range(35)]), sorted(instances_to_remove)) instance_metadata = [{'optimize': False}] instances_to_remove = set() default.DefaultScope( [], mock.Mock(), osc=mock.Mock()).exclude_instances_with_given_metadata( instance_metadata, cluster, instances_to_remove) self.assertEqual(set(), instances_to_remove)
def test_exclude_resource(self, mock_aggregate, mock_detailed_aggregate): mock_aggregate.return_value = [ mock.Mock(id=i, name="HA_{0}".format(i)) for i in range(2) ] mock_collection = [ mock.Mock(id=i, hosts=['Node_{0}'.format(i)]) for i in range(2) ] mock_collection[0].name = 'HA_0' mock_collection[1].name = 'HA_1' mock_detailed_aggregate.side_effect = mock_collection resources_to_exclude = [{ 'host_aggregates': [{ 'name': 'HA_1' }, { 'id': 0 }] }, { 'instances': [{ 'uuid': 'INSTANCE_1' }, { 'uuid': 'INSTANCE_2' }] }, { 'compute_nodes': [{ 'name': 'Node_2' }, { 'name': 'Node_3' }] }, { 'instance_metadata': [{ 'optimize': True }, { 'optimize1': False }] }] instances_to_exclude = [] nodes_to_exclude = [] instance_metadata = [] default.DefaultScope([], mock.Mock(), osc=mock.Mock()).exclude_resources( resources_to_exclude, instances=instances_to_exclude, nodes=nodes_to_exclude, instance_metadata=instance_metadata) self.assertEqual(['Node_0', 'Node_1', 'Node_2', 'Node_3'], sorted(nodes_to_exclude)) self.assertEqual(['INSTANCE_1', 'INSTANCE_2'], sorted(instances_to_exclude)) self.assertEqual([{ 'optimize': True }, { 'optimize1': False }], instance_metadata)
def test_get_scoped_model_with_zones_and_instances(self, mock_zone_list): cluster = self.fake_cluster.generate_scenario_1() audit_scope = fake_scopes.fake_scope_1 mock_zone_list.return_value = [ mock.Mock(zoneName='AZ{0}'.format(i), hosts={'Node_{0}'.format(i): {}}) for i in range(2) ] model = default.DefaultScope(audit_scope, osc=mock.Mock()).get_scoped_model(cluster) expected_edges = [('INSTANCE_2', 'Node_1')] self.assertEqual(sorted(expected_edges), sorted(model.edges()))
def test__remove_instance(self): cluster = self.fake_cluster.generate_scenario_1() default.DefaultScope([], osc=mock.Mock())._remove_instance( cluster, 'INSTANCE_2', 'Node_1') expected_map = { 'Node_4': set(['INSTANCE_7']), 'Node_1': set([]), 'Node_0': set(['INSTANCE_0', 'INSTANCE_1']), 'Node_3': set(['INSTANCE_6']), 'Node_2': set(['INSTANCE_4', 'INSTANCE_5', 'INSTANCE_3']) } self.assertEqual(expected_map, cluster.get_mapping().get_mapping())
def test__remove_node_from_model(self): cluster = self.fake_cluster.generate_scenario_1() default.DefaultScope([], osc=mock.Mock())._remove_node_from_model( ['Node_1', 'Node_2'], cluster) expected_cluster = { 'Node_0': set(['INSTANCE_0', 'INSTANCE_1']), 'Node_1': set([]), 'Node_2': set([]), 'Node_3': set(['INSTANCE_6']), 'Node_4': set(['INSTANCE_7']) } self.assertEqual(expected_cluster, cluster.get_mapping().get_mapping())
def test_remove_instance(self): model = self.fake_cluster.generate_scenario_1() default.DefaultScope([], osc=mock.Mock()).remove_instance( model, model.get_instance_by_uuid('INSTANCE_2'), 'Node_1') expected_edges = [ ('INSTANCE_0', 'Node_0'), ('INSTANCE_1', 'Node_0'), ('INSTANCE_3', 'Node_2'), ('INSTANCE_4', 'Node_2'), ('INSTANCE_5', 'Node_2'), ('INSTANCE_6', 'Node_3'), ('INSTANCE_7', 'Node_4'), ] self.assertEqual(sorted(expected_edges), sorted(model.edges()))
def test_aggregates_wildcard_is_used(self, mock_aggregate, mock_detailed_aggregate): allowed_nodes = [] mock_aggregate.return_value = [mock.Mock(id=i) for i in range(2)] mock_detailed_aggregate.side_effect = [ mock.Mock(id=i, hosts=['Node_{0}'.format(i)]) for i in range(2) ] default.DefaultScope([{ 'host_aggregates': [{ 'id': '*' }] }], osc=mock.Mock())._collect_aggregates([{ 'id': '*' }], allowed_nodes) self.assertEqual(['Node_0', 'Node_1'], allowed_nodes)
def test_aggregates_wildcard_with_other_ids(self, mock_aggregate): allowed_nodes = [] mock_aggregate.return_value = [mock.Mock(id=i) for i in range(2)] scope_handler = default.DefaultScope([{ 'host_aggregates': [{ 'id': '*' }, { 'id': 1 }] }], osc=mock.Mock()) self.assertRaises(exception.WildcardCharacterIsUsed, scope_handler._collect_aggregates, [{ 'id': '*' }, { 'id': 1 }], allowed_nodes)
def test_get_scoped_model_with_zones_and_instances(self, mock_zone_list): cluster = self.fake_cluster.generate_scenario_1() audit_scope = fake_scopes.fake_scope_1 mock_zone_list.return_value = [ mock.Mock(zoneName='AZ{0}'.format(i), hosts={'Node_{0}'.format(i): {}}) for i in range(2) ] model = default.DefaultScope(audit_scope, osc=mock.Mock()).get_scoped_model(cluster) nodes = { 'Node_4': set([]), 'Node_0': set([]), 'Node_3': set([]), 'Node_1': set(['INSTANCE_2']), 'Node_2': set([]) } self.assertEqual(nodes, model.get_mapping().get_mapping())
def test_collect_zones(self, mock_zone_list): allowed_nodes = [] mock_zone_list.return_value = [ mock.Mock(zoneName="AZ{0}".format(i + 1), hosts={ 'Node_{0}'.format(2 * i): 1, 'Node_{0}'.format(2 * i + 1): 2 }) for i in range(2) ] default.DefaultScope([{ 'availability_zones': [{ 'name': "AZ1" }] }], osc=mock.Mock())._collect_zones([{ 'name': "AZ1" }], allowed_nodes) self.assertEqual(['Node_0', 'Node_1'], sorted(allowed_nodes))
def test_exclude_resources(self): resources_to_exclude = [{ 'instances': [{ 'uuid': 'INSTANCE_1' }, { 'uuid': 'INSTANCE_2' }] }, { 'compute_nodes': [{ 'name': 'Node_1' }, { 'name': 'Node_2' }] }] instances_to_exclude = [] nodes_to_exclude = [] default.DefaultScope([], osc=mock.Mock()).exclude_resources( resources_to_exclude, instances=instances_to_exclude, nodes=nodes_to_exclude) self.assertEqual(['Node_1', 'Node_2'], sorted(nodes_to_exclude)) self.assertEqual(['INSTANCE_1', 'INSTANCE_2'], sorted(instances_to_exclude))
def test_zones_wildcard_with_other_ids(self, mock_zone_list): allowed_nodes = [] mock_zone_list.return_value = [ mock.Mock(zoneName="AZ{0}".format(i + 1), hosts={ 'Node_{0}'.format(2 * i): 1, 'Node_{0}'.format(2 * i + 1): 2 }) for i in range(2) ] scope_handler = default.DefaultScope([{ 'availability_zones': [{ 'name': "*" }, { 'name': 'AZ1' }] }], osc=mock.Mock()) self.assertRaises(exception.WildcardCharacterIsUsed, scope_handler._collect_zones, [{ 'name': "*" }, { 'name': 'AZ1' }], allowed_nodes)
def audit_scope_handler(self): if not self._audit_scope_handler: self._audit_scope_handler = default_scope.DefaultScope( self.audit_scope) return self._audit_scope_handler
def test_get_scoped_model_without_scope(self, mock_zone_list): model = self.fake_cluster.generate_scenario_1() default.DefaultScope([], osc=mock.Mock()).get_scoped_model(model) assert not mock_zone_list.called