Merge pull request #2 from Python-Cardano/feat/cache_chain_tip

Fix Ogmios test

Author: nielstron

pycardano/backend/base.py

@@ -8,7 +8,7 @@
 from pycardano.address import Address
 from pycardano.network import Network
 from pycardano.plutus import ExecutionUnits
-from pycardano.transaction import UTxO, Transaction
+from pycardano.transaction import Transaction, UTxO
 
 __all__ = [
     "GenesisParameters",

pycardano/backend/ogmios.py

@@ -1,8 +1,8 @@
 import json
+import time
 from datetime import datetime, timezone
 from enum import Enum
 from typing import Any, Dict, List, Optional, Tuple, Union
-import time
 
 import cbor2
 import requests

test/pycardano/backend/test_ogmios.py

@@ -1,3 +1,5 @@
+from unittest.mock import patch
+
 from pycardano.backend.base import GenesisParameters, ProtocolParameters
 from pycardano.backend.ogmios import OgmiosChainContext
 from pycardano.network import Network
@@ -80,30 +82,31 @@
 ]
 
 
-class TestOgmiosChainContext:
-    chain_context = OgmiosChainContext("", Network.TESTNET)
-
-    def override_request(method, args):
-        if args["query"] == "currentProtocolParameters":
-            return PROTOCOL_RESULT
-        elif args["query"] == "genesisConfig":
-            return GENESIS_RESULT
-        elif "utxo" in args["query"]:
-            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
+def override_request(method, args):
+    if args["query"] == "currentProtocolParameters":
+        return PROTOCOL_RESULT
+    elif args["query"] == "genesisConfig":
+        return GENESIS_RESULT
+    elif "utxo" in args["query"]:
+        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 None
+            return UTXOS
+    else:
+        return None
 
-    chain_context._request = override_request
+
+class TestOgmiosChainContext:
+    def __init__(self):
+        with patch("OgmiosChainContext._request", return_value=override_request):
+            self.chain_context = OgmiosChainContext("", Network.TESTNET)
 
     def test_protocol_param(self):
         assert (