Initial commitv3.3.4

1 files changed, 26 insertions, 0 deletions

diff --git a/tests/managers/thl/test_ledger/test_lm_tx_entries.py b/tests/managers/thl/test_ledger/test_lm_tx_entries.py
new file mode 100644
index 0000000..5bf1c48
--- /dev/null
+++ b/tests/managers/thl/test_ledger/test_lm_tx_entries.py
@@ -0,0 +1,26 @@
+from generalresearch.models.thl.ledger import LedgerEntry
+
+
+class TestLedgerEntryManager:
+
+    def test_get_tx_entries_by_tx(self, ledger_tx, lm):
+        # First confirm the Ledger TX exists with 2 Entries
+        res = lm.get_tx_by_id(transaction_id=ledger_tx.id)
+        assert len(res.entries) == 2
+
+        tx_entries = lm.get_tx_entries_by_tx(transaction=ledger_tx)
+        assert len(tx_entries) == 2
+
+        assert res.entries == tx_entries
+        assert isinstance(tx_entries[0], LedgerEntry)
+
+    def test_get_tx_entries_by_txs(self, ledger_tx, lm):
+        # First confirm the Ledger TX exists with 2 Entries
+        res = lm.get_tx_by_id(transaction_id=ledger_tx.id)
+        assert len(res.entries) == 2
+
+        tx_entries = lm.get_tx_entries_by_txs(transactions=[ledger_tx])
+        assert len(tx_entries) == 2
+
+        assert res.entries == tx_entries
+        assert isinstance(tx_entries[0], LedgerEntry)