From 47a17f3bc8c9c01d09c4932a0531cf835af7bf06 Mon Sep 17 00:00:00 2001
From: Florian Spreckelsen <f.spreckelsen@indiscale.com>
Date: Thu, 13 Mar 2025 14:24:00 +0100
Subject: [PATCH] TST: Mock get_entity_by_name in unit test
---
loanpy/unittests/test_manual_return.py | 16 +++++++++++++---
1 file changed, 13 insertions(+), 3 deletions(-)
diff --git a/loanpy/unittests/test_manual_return.py b/loanpy/unittests/test_manual_return.py
index 2360515..5ae66ea 100644
--- a/loanpy/unittests/test_manual_return.py
+++ b/loanpy/unittests/test_manual_return.py
@@ -1,8 +1,17 @@
-from linkahead import Record
+from linkahead import Record, Property
+from unittest.mock import Mock, patch
+
from loan.box_loan import BOX, BOX_RETURNED, BOX_BORROWED
-from loan.manual_return import _set_returned_box
+from loan.manual_return import _set_returned_box, db
+
+
+def mock_get_entity_by_name(name, role):
+ if name == BOX_RETURNED.name and role == "PROPERTY":
+ return Property(name=BOX_RETURNED.name, id=12345)
+ raise ValueError(f"Mock not defined for {name} with role {role}.")
+@patch("loan.manual_return.db.get_entity_by_name", new=Mock(side_effect=mock_get_entity_by_name))
def test_set_returned_box():
loan = Record()
loan.add_property(BOX, name=BOX_BORROWED.name, value="1234@abcd")
@@ -15,4 +24,5 @@ def test_set_returned_box():
assert len(loan.get_properties()) == 2
assert loan.get_property(BOX_BORROWED.name).value == "1234@abcd"
- assert loan.get_property(BOX_RETURNED.name).value == "1234@HEAD"
+ print(loan)
+ assert loan.get_property(12345).value == "1234@HEAD"
--
GitLab