From 805875ec15a044081bd4c4ae6fd4eb8a7d113aec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Henrik=20tom=20W=C3=B6rden?= <henrik@trineo.org>
Date: Tue, 13 Jul 2021 09:45:10 +0200
Subject: [PATCH] fix
---
unittests/test_property.py | 31 ++++++++++++++++---------------
1 file changed, 16 insertions(+), 15 deletions(-)
diff --git a/unittests/test_property.py b/unittests/test_property.py
index 8c8ee3a6..834b1be5 100644
--- a/unittests/test_property.py
+++ b/unittests/test_property.py
@@ -92,22 +92,20 @@ def test_selected_reference_list():
assert isinstance(testrecord.get_property("Conductor").value[0], Entity)
-PROPS = {
- 10: db.INTEGER,
- 20: db.REFERENCE,
- 30: "SomeRT",
-}
-
-
-def dummy_retrieve(self):
- self.datatype = PROPS[self.id]
- self.is_valid = lambda: True
-
-
-Entity.retrieve = dummy_retrieve
-
-
def test_is_reference():
+ PROPS = {
+ 10: db.INTEGER,
+ 20: db.REFERENCE,
+ 30: "SomeRT",
+ }
+
+ def dummy_retrieve(self):
+ self.datatype = PROPS[self.id]
+ self.is_valid = lambda: True
+ # replace retrieve function by dummy
+ real_retrieve = Entity.retrieve
+ Entity.retrieve = dummy_retrieve
+
p1 = Property(id=1, datatype=db.INTEGER)
p2 = Property(id=2, datatype=db.DOUBLE)
p3 = Property(id=3, datatype=db.TEXT)
@@ -133,3 +131,6 @@ def test_is_reference():
assert p10.is_reference(server_retrieval=True) is False
assert p20.is_reference(server_retrieval=True) is True
assert p30.is_reference(server_retrieval=True) is True
+
+ # restore retrieve function with original
+ Entity.retrieve = real_retrieve
--
GitLab