Skip to content
Snippets Groups Projects
Commit 805875ec authored by Henrik tom Wörden's avatar Henrik tom Wörden
Browse files

fix

parent 81fd4920
Branches
Tags
1 merge request!18ENH: add is_reference to db.Property
...@@ -92,22 +92,20 @@ def test_selected_reference_list(): ...@@ -92,22 +92,20 @@ def test_selected_reference_list():
assert isinstance(testrecord.get_property("Conductor").value[0], Entity) 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(): 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) p1 = Property(id=1, datatype=db.INTEGER)
p2 = Property(id=2, datatype=db.DOUBLE) p2 = Property(id=2, datatype=db.DOUBLE)
p3 = Property(id=3, datatype=db.TEXT) p3 = Property(id=3, datatype=db.TEXT)
...@@ -133,3 +131,6 @@ def test_is_reference(): ...@@ -133,3 +131,6 @@ def test_is_reference():
assert p10.is_reference(server_retrieval=True) is False assert p10.is_reference(server_retrieval=True) is False
assert p20.is_reference(server_retrieval=True) is True assert p20.is_reference(server_retrieval=True) is True
assert p30.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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment