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