def test_create_network_with_default_policy(self, dvs_update_mock,
update_bindings_mock):
"""Test the DVS update when default policy attached to a network"""
# Create a default policy with a rule
policy_data = copy.deepcopy(self.policy_data['policy'])
policy_data['is_default'] = True
_policy = policy_object.QosPolicy(self.ctxt, **policy_data)
setattr(_policy, "rules", [self.rule, self.dscp_rule])
default_policy = policy_object.QosPolicyDefault(
qos_policy_id=policy_data['id'])
with mock.patch('neutron.services.qos.qos_plugin.QoSPlugin.'
'get_policy',
return_value=_policy) as get_rules_mock,\
mock.patch('neutron.objects.qos.policy.QosPolicyDefault.'
'get_object', return_value=default_policy):
# create the network (with no specific qos policy)
net_data = copy.deepcopy(self._net_data)
del net_data['network']['qos_policy_id']
net = self._create_net(net_data=net_data)
# make sure the network-policy binding was updated
update_bindings_mock.assert_called_once_with(
self.ctxt, net['id'], self.policy.id)
# make sure the qos rule was found
get_rules_mock.assert_called_with(self.ctxt, self.policy.id)
# make sure the dvs was updated
self.assertTrue(dvs_update_mock.called)
def test_unset_default_default_policy_exists(self, mock_default_delete,
mock_default_get):
obj = self._create_test_policy()
with mock.patch.object(obj, 'get_default', return_value=obj.id):
mock_default_get.return_value = policy.QosPolicyDefault()
obj.unset_default()
mock_default_get.assert_called_once_with(obj.obj_context,
project_id=obj.project_id)
mock_default_delete.assert_called_once_with()
