fix(tree-grid): update selection state correctly when grid is filtered

- with primaryKey - foreignKey configuration

Author: mmart1n

projects/igniteui-angular/src/lib/grids/tree-grid/tree-grid.component.ts

@@ -478,7 +478,7 @@ export class IgxTreeGridComponent extends IgxGridBaseDirective implements GridTy
                     // if a row has been added and before commiting the transaction deleted
                     const leafRowsDirectParents = new Set<any>();
                     this.records.forEach(record => {
-                        if (record && !record.children && record.parent) {
+                        if (record && (!record.children || record.children.length === 0) && record.parent) {
                             leafRowsDirectParents.add(record.parent);
                         }
                     });
@@ -495,7 +495,7 @@ export class IgxTreeGridComponent extends IgxGridBaseDirective implements GridTy
             if (this.rowSelection === GridSelectionMode.multipleCascade) {
                 const leafRowsDirectParents = new Set<any>();
                 this.records.forEach(record => {
-                    if (record && !record.children && record.parent) {
+                    if (record && (!record.children || record.children.length === 0) && record.parent) {
                         leafRowsDirectParents.add(record.parent);
                     }
                 });