Add TextContent Class

DSD-DBS · Dec 18, 2023 · 30b161a · 30b161a
1 parent 2622e9c
commit 30b161a
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 8 deletions.
diff --git a/polarion_rest_api_client/base_client.py b/polarion_rest_api_client/base_client.py
@@ -9,7 +9,6 @@
 from polarion_rest_api_client import data_models as dm
 
 WorkItemType = t.TypeVar("WorkItemType", bound=dm.WorkItem)
-DocumentType = t.TypeVar("DocumentType", bound=dm.Document)
 
 
 class DefaultFields:

diff --git a/polarion_rest_api_client/client.py b/polarion_rest_api_client/client.py
@@ -694,10 +694,10 @@ def get_document(
                     module_name=unset_str_builder(attributes.module_name),
                     type=unset_str_builder(attributes.type),
                     status=unset_str_builder(attributes.status),
-                    home_page_content={
-                        "type": attributes.home_page_content.type,
-                        "value": attributes.home_page_content.value,
-                    }
+                    home_page_content=dm.TextContent(
+                        type=attributes.home_page_content.type,
+                        value=attributes.home_page_content.value,
+                    )
                     if not isinstance(
                         attributes.home_page_content, oa_types.Unset
                     )

diff --git a/polarion_rest_api_client/data_models.py b/polarion_rest_api_client/data_models.py
@@ -147,7 +147,7 @@ class Document(BaseItem):
 
     module_folder: str | None = None
     module_name: str | None = None
-    home_page_content: dict | None = None
+    home_page_content: TextContent | None = None
 
     def __init__(
         self,
@@ -156,9 +156,22 @@ def __init__(
         module_name: str | None = None,
         type: str | None = None,
         status: str | None = None,
-        home_page_content: dict | None = None,
+        home_page_content: TextContent | None = None,
     ):
         super().__init__(id, type, status)
         self.module_folder = module_folder
         self.module_name = module_name
         self.home_page_content = home_page_content
+
+
+class TextContent:
+    type: str | None = None
+    value: str | None = None
+
+    def __init__(
+        self,
+        type: str | None = None,
+        value: str | None = None,
+    ):
+        self.type = type
+        self.value = value
diff --git a/tests/test_client_documents.py b/tests/test_client_documents.py
@@ -33,5 +33,7 @@ def test_get_document_with_all_fields(
         "MyDocumentName",
         "standardSpecification",
         "open",
-        {"type": "text/html", "value": "<h1>My text value</h1>"},
+        polarion_api.data_models.TextContent(
+            type="text/html", value="<h1>My text value</h1>"
+        ),
     )