Merge pull request #80 from 3D-Beacons/ensembl-fix

HOTFIX: Cast transcript_id to string

Author: sreenathnair

app/ensembl/ensembl.py

@@ -83,9 +83,11 @@ async def get_ensembl_summary_helper(
 
         uniprot_accession = mapping["uniprotEntry"]["uniprotAccession"]
         uniprot_set.add(uniprot_accession)
-
         mapping["ensemblTranscript"].update(
-            {"alignment_difference": mapping["alignment_difference"]}
+            {
+                "alignment_difference": mapping["alignment_difference"],
+                "transcript_id": str(mapping["ensemblTranscript"]["transcript_id"]),
+            }
         )
 
         if not transcript_dict.get(uniprot_accession):

tests/test_hub.py

@@ -65,18 +65,30 @@ async def test_get_ensembl_summaries_api(
         "app.ensembl.ensembl.get_ensembl_mappings", return_value=valid_gifts_response
     )
 
-    uniprot_list_future = asyncio.Future()
-    uniprot_list_future.set_result(uniprot_summary_obj_list)
     mocker.patch(
         "app.ensembl.ensembl.get_list_of_uniprot_summary_helper",
-        return_value=uniprot_list_future,
+        return_value=uniprot_summary_obj_list,
+    )
+    mocker.patch(
+        "app.ensembl.ensembl.get_uniprot_api_results",
+        return_value={},
+    )
+    mocker.patch(
+        "app.ensembl.ensembl.get_uniprot_name",
+        return_value="Test protein",
     )
 
     response = await client.get(
         "/ensembl/summary/ENSG00000288864.json",
     )
 
     assert response.status_code == status.HTTP_200_OK
+    response_json = response.json()
+    assert response_json["uniprot_mappings"]
+    assert all(
+        isinstance(mapping["ensembl_transcript"]["transcript_id"], str)
+        for mapping in response_json["uniprot_mappings"]
+    )
 
 
 @pytest.mark.asyncio