1 files changed, 128 insertions, 0 deletions
diff --git a/heat/tests/openstack/neutron/test_qos.py b/heat/tests/openstack/neutron/test_qos.py
index d46eb64f4..306b15b4f 100644
--- a/heat/tests/openstack/neutron/test_qos.py
+++ b/heat/tests/openstack/neutron/test_qos.py
@@ -70,6 +70,19 @@ resources:
       tenant_id: d66c74c01d6c41b9846088c1ad9634d0
 '''
 
+minimum_packet_rate_rule_template = '''
+heat_template_version: 2021-04-16
+description: This template to define a neutron minimum packet rate rule.
+resources:
+  my_minimum_packet_rate_rule:
+    type: OS::Neutron::QoSMinimumPacketRateRule
+    properties:
+      policy: 477e8273-60a7-4c41-b683-fdb0bc7cd151
+      min_kpps: 1000
+      direction: any
+      tenant_id: d66c74c01d6c41b9846088c1ad9634d0
+'''
+
 
 class NeutronQoSPolicyTest(common.HeatTestCase):
     def setUp(self):
@@ -515,3 +528,118 @@ class NeutronQoSMinimumBandwidthRuleTest(common.HeatTestCase):
 
         self.neutronclient.show_minimum_bandwidth_rule.assert_called_once_with(
             self.minimum_bandwidth_rule.resource_id, self.policy_id)
+
+
+class NeutronQoSMinimumPacketRateRuleTest(common.HeatTestCase):
+    def setUp(self):
+        super(NeutronQoSMinimumPacketRateRuleTest, self).setUp()
+
+        self.ctx = utils.dummy_context()
+        tpl = template_format.parse(minimum_packet_rate_rule_template)
+        self.stack = stack.Stack(
+            self.ctx,
+            'neutron_minimum_packet_rate_rule_test',
+            template.Template(tpl)
+        )
+
+        self.neutronclient = mock.MagicMock()
+        self.patchobject(neutron.NeutronClientPlugin, 'has_extension',
+                         return_value=True)
+        self.minimum_packet_rate_rule = self.stack[
+            'my_minimum_packet_rate_rule']
+        self.minimum_packet_rate_rule.client = mock.MagicMock(
+            return_value=self.neutronclient)
+        self.find_mock = self.patchobject(
+            neutron.neutronV20,
+            'find_resourceid_by_name_or_id')
+        self.policy_id = '477e8273-60a7-4c41-b683-fdb0bc7cd151'
+        self.find_mock.return_value = self.policy_id
+
+    def test_rule_handle_create(self):
+        rule = {
+            'minimum_packet_rate_rule': {
+                'id': 'cf0eab12-ef8b-4a62-98d0-70576583c17a',
+                'min_kpps': 1000,
+                'direction': 'any',
+                'tenant_id': 'd66c74c01d6c41b9846088c1ad9634d0'
+            }
+        }
+
+        create_props = {'min_kpps': 1000,
+                        'direction': 'any',
+                        'tenant_id': 'd66c74c01d6c41b9846088c1ad9634d0'}
+        self.neutronclient.create_minimum_packet_rate_rule.return_value = rule
+
+        self.minimum_packet_rate_rule.handle_create()
+        self.assertEqual('cf0eab12-ef8b-4a62-98d0-70576583c17a',
+                         self.minimum_packet_rate_rule.resource_id)
+        self.neutronclient.create_minimum_packet_rate_rule.\
+            assert_called_once_with(
+                self.policy_id,
+                {'minimum_packet_rate_rule': create_props})
+
+    def test_rule_handle_delete(self):
+        rule_id = 'cf0eab12-ef8b-4a62-98d0-70576583c17a'
+        self.minimum_packet_rate_rule.resource_id = rule_id
+        self.neutronclient.delete_minimum_packet_rate_rule.return_value = None
+
+        self.assertIsNone(self.minimum_packet_rate_rule.handle_delete())
+        self.neutronclient.delete_minimum_packet_rate_rule.\
+            assert_called_once_with(rule_id, self.policy_id)
+
+    def test_rule_handle_delete_not_found(self):
+        rule_id = 'cf0eab12-ef8b-4a62-98d0-70576583c17a'
+        self.minimum_packet_rate_rule.resource_id = rule_id
+        not_found = self.neutronclient.NotFound
+        self.neutronclient.delete_minimum_packet_rate_rule.side_effect =\
+            not_found
+
+        self.assertIsNone(self.minimum_packet_rate_rule.handle_delete())
+        self.neutronclient.delete_minimum_packet_rate_rule.\
+            assert_called_once_with(rule_id, self.policy_id)
+
+    def test_rule_handle_delete_resource_id_is_none(self):
+        self.minimum_packet_rate_rule.resource_id = None
+        self.assertIsNone(self.minimum_packet_rate_rule.handle_delete())
+        self.assertEqual(
+            0,
+            self.neutronclient.minimum_packet_rate_rule.call_count)
+
+    def test_rule_handle_update(self):
+        rule_id = 'cf0eab12-ef8b-4a62-98d0-70576583c17a'
+        self.minimum_packet_rate_rule.resource_id = rule_id
+
+        prop_diff = {
+            'min_kpps': 500
+        }
+
+        self.minimum_packet_rate_rule.handle_update(
+            json_snippet={},
+            tmpl_diff={},
+            prop_diff=prop_diff.copy())
+
+        self.neutronclient.update_minimum_packet_rate_rule.\
+            assert_called_once_with(
+                rule_id,
+                self.policy_id,
+                {'minimum_packet_rate_rule': prop_diff})
+
+    def test_rule_get_attr(self):
+        self.minimum_packet_rate_rule.resource_id = 'test rule'
+        rule = {
+            'minimum_packet_rate_rule': {
+                'id': 'cf0eab12-ef8b-4a62-98d0-70576583c17a',
+                'min_kpps': 1000,
+                'direction': 'egress',
+                'tenant_id': 'd66c74c01d6c41b9846088c1ad9634d0'
+            }
+        }
+        self.neutronclient.show_minimum_packet_rate_rule.return_value = rule
+
+        self.assertEqual(rule['minimum_packet_rate_rule'],
+                         self.minimum_packet_rate_rule.FnGetAtt('show'))
+
+        self.neutronclient.show_minimum_packet_rate_rule.\
+            assert_called_once_with(
+                self.minimum_packet_rate_rule.resource_id,
+                self.policy_id)