Added unit test for 'get_proposal_id()' function

Author: tieneupin

tests/server/test_ispyb.py

@@ -2,7 +2,7 @@
 from sqlalchemy import select
 from sqlalchemy.orm import Session
 
-from murfey.server.ispyb import get_session_id
+from murfey.server.ispyb import get_proposal_id, get_session_id
 from tests.conftest import ExampleVisit
 
 
@@ -33,3 +33,25 @@ def test_get_session_id(
         db=ispyb_db,
     )
     assert query == result
+
+
+def test_get_proposal_id(
+    ispyb_db: Session,
+):
+    # Manually query the Proposal ID
+    query = (
+        ispyb_db.execute(
+            select(Proposal)
+            .where(Proposal.proposalCode == ExampleVisit.proposal_code)
+            .where(Proposal.proposalNumber == ExampleVisit.proposal_number)
+        )
+        .scalar_one()
+        .proposalId
+    )
+
+    # Test function
+    result = get_proposal_id(
+        proposal_code=ExampleVisit.proposal_code,
+        proposal_number=ExampleVisit.proposal_number,
+    )
+    assert query == result