commit d97959c7ede6ed9371f01cd789b434385bc6378d
author Michal Vasko <mvasko@cesnet.cz> Thu Dec 10 12:18:28 2020 +0100
committer Michal Vasko <mvasko@cesnet.cz> Fri Dec 11 11:54:58 2020 +0100
tree 3d48e3803e1bded444b3d92cd0db0138dbf9ba05
parent 2f03d226e60245b57e5e5cc77eebfe46b590bd1e

schema tree BUGFIX atoms are all traversed nodes

src/tree_schema.c

diff --git a/src/tree_schema.c b/src/tree_schema.c
index 588fdd0..b7fb264 100644
--- a/src/tree_schema.c
+++ b/src/tree_schema.c
@@ -381,7 +381,9 @@
     (*set)->size = xp_set.used;
 
     for (i = 0; i < xp_set.used; ++i) {
-        if ((xp_set.val.scnodes[i].type == LYXP_NODE_ELEM) && (xp_set.val.scnodes[i].in_ctx == LYXP_SET_SCNODE_ATOM_CTX)) {
+        if ((xp_set.val.scnodes[i].type == LYXP_NODE_ELEM) && (xp_set.val.scnodes[i].in_ctx >= LYXP_SET_SCNODE_ATOM)) {
+            assert((xp_set.val.scnodes[i].in_ctx == LYXP_SET_SCNODE_ATOM) ||
+                    (xp_set.val.scnodes[i].in_ctx == LYXP_SET_SCNODE_ATOM_CTX));
             ret = ly_set_add(*set, xp_set.val.scnodes[i].scnode, 1, NULL);
             LY_CHECK_GOTO(ret, cleanup);
         }