test cases for add registered attrs

py_tests/test_dataflow.py

@@ -1,6 +1,10 @@
 """Test dataflow.py"""
 
+import requests
+import requests_mock
+
 from fusion.dataflow import InputDataFlow, OutputDataFlow
+from fusion.fusion import Fusion
 
 
 def test_inputdataflow_class_object_representation() -> None:
@@ -13,3 +17,23 @@ def test_outputdataflow_class_object_representation() -> None:
     dataflow = OutputDataFlow(identifier="my_dataflow", flow_details={"key": "value"})
     assert repr(dataflow)
 
+
+def test_add_registered_attribute(requests_mock: requests_mock.Mocker, fusion_obj: Fusion) -> None:
+    """Test the add_registered_attribute method."""
+    catalog = "my_catalog"
+    dataflow = "TEST_DATAFLOW"
+    attribute_identifier = "my_attribute"
+    url = f"{fusion_obj.root_url}catalogs/{catalog}/datasets/{dataflow}/attributes/{attribute_identifier}/registration"
+
+    requests_mock.post(url, json={"isCriticalDataElement": False})
+
+    dataflow_obj = InputDataFlow(identifier="TEST_DATAFLOW")
+    dataflow_obj.client = fusion_obj
+    resp = dataflow_obj.add_registered_attribute(
+        attribute_identifier="my_attribute",
+        catalog=catalog,
+        return_resp_obj=True
+    )
+    assert isinstance(resp, requests.Response)
+    status_code = 200
+    assert resp.status_code == status_code

py_tests/test_report.py

@@ -1,9 +1,34 @@
 """Test file for report.py"""
 
+import requests
+import requests_mock
+
+from fusion.fusion import Fusion
 from fusion.report import Report
 
 
 def test_report_class_object_representation() -> None:
     """Test the object representation of the Report class."""
     report = Report(identifier="my_report", report={"key": "value"})
-    assert repr(report)
+    assert repr(report)
+
+def test_add_registered_attribute(requests_mock: requests_mock.Mocker, fusion_obj: Fusion) -> None:
+    """Test the add_registered_attribute method."""
+    catalog = "my_catalog"
+    report = "TEST_REPORT"
+    attribute_identifier = "my_attribute"
+    url = f"{fusion_obj.root_url}catalogs/{catalog}/datasets/{report}/attributes/{attribute_identifier}/registration"
+
+    requests_mock.post(url, json={"isCriticalDataElement": True})
+
+    report_obj = Report(identifier="TEST_REPORT")
+    report_obj.client = fusion_obj
+    resp = report_obj.add_registered_attribute(
+        attribute_identifier="my_attribute",
+        is_kde=True,
+        catalog=catalog,
+        return_resp_obj=True
+    )
+    assert isinstance(resp, requests.Response)
+    status_code = 200
+    assert resp.status_code == status_code