Ruff typing from this morning. WIP

1 files changed, 29 insertions, 11 deletions

diff --git a/tests/managers/thl/test_ledger/test_lm_tx_metadata.py b/tests/managers/thl/test_ledger/test_lm_tx_metadata.py
index 5d12633..f63efa4 100644
--- a/tests/managers/thl/test_ledger/test_lm_tx_metadata.py
+++ b/tests/managers/thl/test_ledger/test_lm_tx_metadata.py
@@ -1,34 +1,52 @@
+from __future__ import annotations
+
+from generalresearch.managers.thl.ledger_manager.ledger import (
+    LedgerManager,
+    LedgerTransaction,
+)
+
+
 class TestLedgerMetadataManager:
 
-    def test_get_tx_metadata_by_txs(self, ledger_tx, lm):
+    def test_get_tx_metadata_by_txs(
+        self, ledger_tx: LedgerTransaction, ledger_manager: LedgerManager
+    ):
         # First confirm the Ledger TX exists with 2 Entries
-        res = lm.get_tx_by_id(transaction_id=ledger_tx.id)
+        res = ledger_manager.get_tx_by_id(transaction_id=ledger_tx.id)
         assert isinstance(res.metadata, dict)
 
-        tx_metadatas = lm.get_tx_metadata_by_txs(transactions=[ledger_tx])
+        tx_metadatas = ledger_manager.get_tx_metadata_by_txs(transactions=[ledger_tx])
         assert isinstance(tx_metadatas, dict)
         assert isinstance(tx_metadatas[ledger_tx.id], dict)
 
         assert res.metadata == tx_metadatas[ledger_tx.id]
 
-    def test_get_tx_metadata_ids_by_tx(self, ledger_tx, lm):
+    def test_get_tx_metadata_ids_by_tx(
+        self, ledger_tx: LedgerTransaction, ledger_manager: LedgerManager
+    ):
         # First confirm the Ledger TX exists with 2 Entries
-        res = lm.get_tx_by_id(transaction_id=ledger_tx.id)
+        res = ledger_manager.get_tx_by_id(transaction_id=ledger_tx.id)
         tx_metadata_cnt = len(res.metadata.keys())
 
-        tx_metadata_ids = lm.get_tx_metadata_ids_by_tx(transaction=ledger_tx)
+        tx_metadata_ids = ledger_manager.get_tx_metadata_ids_by_tx(
+            transaction=ledger_tx
+        )
         assert isinstance(tx_metadata_ids, set)
-        assert isinstance(list(tx_metadata_ids)[0], int)
+        assert isinstance(next(iter(tx_metadata_ids)), int)
 
         assert tx_metadata_cnt == len(tx_metadata_ids)
 
-    def test_get_tx_metadata_ids_by_txs(self, ledger_tx, lm):
+    def test_get_tx_metadata_ids_by_txs(
+        self, ledger_tx: LedgerTransaction, ledger_manager: LedgerManager
+    ):
         # First confirm the Ledger TX exists with 2 Entries
-        res = lm.get_tx_by_id(transaction_id=ledger_tx.id)
+        res = ledger_manager.get_tx_by_id(transaction_id=ledger_tx.id)
         tx_metadata_cnt = len(res.metadata.keys())
 
-        tx_metadata_ids = lm.get_tx_metadata_ids_by_txs(transactions=[ledger_tx])
+        tx_metadata_ids = ledger_manager.get_tx_metadata_ids_by_txs(
+            transactions=[ledger_tx]
+        )
         assert isinstance(tx_metadata_ids, set)
-        assert isinstance(list(tx_metadata_ids)[0], int)
+        assert isinstance(next(iter(tx_metadata_ids)), int)
 
         assert tx_metadata_cnt == len(tx_metadata_ids)