From ecc407f84e63922725a83218c13edbaa56d41887 Mon Sep 17 00:00:00 2001 From: Keith Walsh Date: Thu, 12 Sep 2024 08:53:16 -0400 Subject: [PATCH] Update workspace serializer tests --- rbac/management/workspace/serializer.py | 11 ++++++++++- tests/management/workspace/test_serializer.py | 7 +++++++ 2 files changed, 17 insertions(+), 1 deletion(-) diff --git a/rbac/management/workspace/serializer.py b/rbac/management/workspace/serializer.py index 8bbbf5172..5164728af 100644 --- a/rbac/management/workspace/serializer.py +++ b/rbac/management/workspace/serializer.py @@ -28,12 +28,21 @@ class WorkspaceSerializer(serializers.ModelSerializer): uuid = serializers.UUIDField(read_only=True, required=False) description = serializers.CharField(allow_null=True, required=False, max_length=255) parent_id = serializers.UUIDField(allow_null=True, required=False) + created = serializers.DateTimeField(read_only=True) + modified = serializers.DateTimeField(read_only=True) class Meta: """Metadata for the serializer.""" model = Workspace - fields = ("name", "uuid", "parent_id", "description") + fields = ( + "name", + "uuid", + "parent_id", + "description", + "created", + "modified", + ) def create(self, validated_data): """Create the workspace object in the database.""" diff --git a/tests/management/workspace/test_serializer.py b/tests/management/workspace/test_serializer.py index 0afbf806e..791e260e5 100644 --- a/tests/management/workspace/test_serializer.py +++ b/tests/management/workspace/test_serializer.py @@ -40,6 +40,9 @@ def tearDown(self): Workspace.objects.update(parent=None) Workspace.objects.all().delete() + def _format_timestamps(self, timestamp): + return timestamp.isoformat(timespec="microseconds").replace("+00:00", "Z") + def test_get_workspace_detail_child(self): """Return GET /workspace// serializer response for child""" serializer = WorkspaceSerializer(self.child) @@ -48,6 +51,8 @@ def test_get_workspace_detail_child(self): "name": self.child.name, "description": self.child.description, "parent_id": str(self.parent.uuid), + "created": self._format_timestamps(self.child.created), + "modified": self._format_timestamps(self.child.modified), } self.assertDictEqual(serializer.data, expected_data) @@ -60,6 +65,8 @@ def test_get_workspace_detail_parent(self): "name": self.parent.name, "description": self.parent.description, "parent_id": None, + "created": self._format_timestamps(self.parent.created), + "modified": self._format_timestamps(self.parent.modified), } self.assertDictEqual(serializer.data, expected_data)