new entity for NotificationRecipients (#1125)

Added new entity `NotificationRecipients` for `/notification_recipients` endpoint. (cherry picked from commit e6e6ec6) Co-authored-by: Pavel Novotny <[email protected]>
SatelliteQE · Apr 9, 2024 · 0172314 · 0172314
1 parent e53b19f
commit 0172314
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 0 deletions.
diff --git a/nailgun/entities.py b/nailgun/entities.py
@@ -8916,3 +8916,17 @@ def search(self, fields=None, query=None, filters=None):
         return super().search(
             fields=fields, query=query, filters=filters, path_fields={'user': self.user}
         )
+
+
+class NotificationRecipients(Entity, EntityReadMixin):
+    """A representation of /notification_recipients endpoint."""
+
+    def __init__(self, server_config=None, **kwargs):
+        self._fields = {
+            'notifications': entity_fields.ListField(),
+        }
+        self._meta = {
+            'api_path': '/notification_recipients',
+            'read_type': 'base',
+        }
+        super().__init__(server_config, **kwargs)
diff --git a/tests/test_entities.py b/tests/test_entities.py
@@ -3999,3 +3999,34 @@ def test_scap_tailoring_file(self):
             _get_required_field_names(entity),
             set(entity.get_values().keys()),
         )
+
+
+class NotificationRecipientsTestCase(TestCase):
+    """Tests for :class:`nailgun.entities.NotificationRecipients`."""
+
+    def test_to_json(self):
+        """Check json serialisation on nested entities."""
+        notifications_kwargs = {
+            "notifications": [
+                {
+                    "id": 28,
+                    "seen": False,
+                    "level": "info",
+                    "text": "The fastest guide to configuring Red Hat Satellite ever",
+                    "created_at": "2024-03-20T17:24:33.596Z",
+                    "group": "Community",
+                    "actions": {
+                        "links": [
+                            {
+                                "href": "https://www.redhat.com/en/blog/fastest-guide-configuring-red-hat-satellite-ever",
+                                "title": "Open",
+                                "external": True,
+                            }
+                        ]
+                    },
+                }
+            ]
+        }
+        cfg = config.ServerConfig(url='https://foo.bar', verify=False, auth=('foo', 'bar'))
+        notifications = entities.NotificationRecipients(cfg, **notifications_kwargs)
+        self.assertDictEqual(notifications_kwargs, json.loads(notifications.to_json()))