From 9308e9456e12f78e490905b4c58de797d22a00f5 Mon Sep 17 00:00:00 2001 From: Timm Fitschen <t.fitschen@indiscale.com> Date: Wed, 18 Nov 2020 02:33:16 +0100 Subject: [PATCH] TST: entity state in queries --- tests/test_state.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/test_state.py b/tests/test_state.py index 87a284c..36b786b 100644 --- a/tests/test_state.py +++ b/tests/test_state.py @@ -121,9 +121,11 @@ def test_state_query(): rec.insert() assert rec.get_property("State") is None - assert db.execute_query( - "FIND TestRT WITH State = State1", - unique=True).id == rec.id + result = db.execute_query("FIND TestRT WITH State = State1", unique=True) + assert result.id == rec.id + + assert result.get_property("State") is None + assert result.state == db.State(model="Model1", name="State1") def test_state_transition(): -- GitLab