utxo_by_tx_id ogmios test

Author: juliusfrost

pycardano/backend/ogmios.py

@@ -384,6 +384,7 @@ def utxo_by_tx_id(self, tx_id: str, index: int) -> Optional[UTxO]:
             Optional[UTxO]: Return a UTxO if exists at the tx_id and index.
         """
         results = self._query_utxos_by_tx_id(tx_id, index)
+        assert len(results) < 2, f"Query for UTxO {tx_id}#{index} should be unique!"
 
         utxos = []
         for result in results:

test/pycardano/backend/test_ogmios.py

@@ -89,7 +89,17 @@ def override_request(method, args):
         elif args["query"] == "genesisConfig":
             return GENESIS_RESULT
         elif "utxo" in args["query"]:
-            return UTXOS
+            query = args["query"]["utxo"][0]
+            if isinstance(query, dict):
+                for utxo in UTXOS:
+                    if (
+                        utxo[0]["txId"] == query["txId"]
+                        and utxo[0]["index"] == query["index"]
+                    ):
+                        return [utxo]
+                return []
+            else:
+                return UTXOS
         else:
             return None
 
@@ -174,3 +184,19 @@ def test_utxo(self):
                 },
             }
         )
+
+    def test_utxo_by_tx_id(self):
+        utxo = self.chain_context.utxo_by_tx_id(
+            "3a42f652bd8dee788577e8c39b6217db3df659c33b10a2814c20fb66089ca167",
+            1,
+        )
+        assert utxo.input == TransactionInput.from_primitive(
+            ["3a42f652bd8dee788577e8c39b6217db3df659c33b10a2814c20fb66089ca167", 1]
+        )
+        assert utxo.output.amount == 764295183
+
+        not_utxo = self.chain_context.utxo_by_tx_id(
+            "3a42f652bd8dee788577e8c39b6217db3df659c33b10a2814c20fb66089ca167",
+            2,
+        )
+        assert not_utxo is None