TST: entity state in queries

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():