feat: method to register codeocean result assets to docdb

src/aind_data_access_api/document_db.py

@@ -655,6 +655,34 @@ def register_asset(self, s3_location: str) -> Dict[str, Any]:
         response.raise_for_status()
         return response.json()
 
+    def register_co_result(
+        self,
+        s3_location: str,
+        name: str,
+        co_asset_id: str,
+        co_computation_id: str,
+    ) -> Dict[str, Any]:
+        """Register a Code Ocean result asset to the DocDB metadata index."""
+
+        data = json.dumps(
+            {
+                "s3_location": s3_location,
+                "name": name,
+                "co_asset_id": co_asset_id,
+                "co_computation_id": co_computation_id,
+            }
+        )
+        signed_header = self._signed_request(
+            method="POST", url=self._register_asset_url, data=data
+        )
+        response = self.session.post(
+            url=self._register_asset_url,
+            headers=dict(signed_header.headers),
+            data=data,
+        )
+        response.raise_for_status()
+        return response.json()
+
     def deregister_asset(self, s3_location: str) -> Dict[str, Any]:
         """De-register (delete) a data asset in Code Ocean and the
         DocDB metadata index."""

tests/test_document_db.py

@@ -979,6 +979,52 @@ def test_register_asset(
         )
         self.assertEqual(response_message, response)
 
+    @patch("boto3.session.Session")
+    @patch("botocore.auth.SigV4Auth.add_auth")
+    @patch("requests.Session.post")
+    def test_register_co_result(
+        self,
+        mock_post: MagicMock,
+        mock_auth: MagicMock,
+        mock_session: MagicMock,
+    ):
+        """Tests register_co_result method"""
+        mock_creds = MagicMock()
+        mock_creds.access_key = "abc"
+        mock_creds.secret_key = "efg"
+        mock_session.return_value.region_name = "us-west-2"
+        mock_session.get_credentials.return_value = mock_creds
+        mock_response = Response()
+        mock_response.status_code = 201
+        response_message = {
+            "message": (
+                "Inserted new DocDB record for result asset "
+                "s3://codeocean-bucket/123-456."
+            ),
+        }
+        mock_response._content = json.dumps(response_message).encode("utf-8")
+        mock_post.return_value = mock_response
+
+        client = MetadataDbClient(**self.example_client_args)
+        response = client.register_co_result(
+            s3_location="s3://codeocean-bucket/123-456",
+            name="co_internal_result",
+            co_asset_id="123-456",
+            co_computation_id="comp-789",
+        )
+        mock_auth.assert_called_once()
+        mock_post.assert_called_once_with(
+            url="https://example.com/v1/assets/register",
+            headers={"Content-Type": "application/json"},
+            data=(
+                '{"s3_location": "s3://codeocean-bucket/123-456", '
+                '"name": "co_internal_result", '
+                '"co_asset_id": "123-456", '
+                '"co_computation_id": "comp-789"}'
+            ),
+        )
+        self.assertEqual(response_message, response)
+
     @patch("boto3.session.Session")
     @patch("botocore.auth.SigV4Auth.add_auth")
     @patch("requests.Session.delete")