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()