test(ivy): get ViewRef.rootNodes should get all root nodes from projectable nodes (#33647)

PR Close #33647
This commit is contained in:
Pawel Kozlowski
2019-11-07 12:23:17 +01:00
committed by Kara Erickson
parent 86d656bc67
commit e6f383fe20
2 changed files with 13 additions and 12 deletions

View File

@ -19,7 +19,7 @@ import {CONTEXT, FLAGS, HOST, LView, LViewFlags, TVIEW, T_HOST} from './interfac
import {assertNodeOfPossibleTypes} from './node_assert';
import {destroyLView, renderDetachView} from './node_manipulation';
import {findComponentView, getLViewParent} from './util/view_traversal_utils';
import {getNativeByTNode, unwrapRNode} from './util/view_utils';
import {unwrapRNode} from './util/view_utils';
@ -302,7 +302,8 @@ export class RootViewRef<T> extends ViewRef<T> {
get context(): T { return null !; }
}
function collectNativeNodes(lView: LView, tNode: TNode | null, result: any[]): any[] {
function collectNativeNodes(
lView: LView, tNode: TNode | null, result: any[], isProjection: boolean = false): any[] {
while (tNode !== null) {
ngDevMode && assertNodeOfPossibleTypes(
tNode, TNodeType.Element, TNodeType.Container, TNodeType.Projection,
@ -333,15 +334,13 @@ function collectNativeNodes(lView: LView, tNode: TNode | null, result: any[]): a
const componentView = findComponentView(lView);
const componentHost = componentView[T_HOST] as TElementNode;
const parentView = getLViewParent(componentView);
let currentProjectedNode: TNode|null =
let firstProjectedNode: TNode|null =
(componentHost.projection as(TNode | null)[])[tNode.projection as number];
while (currentProjectedNode !== null && parentView !== null) {
result.push(getNativeByTNode(currentProjectedNode, parentView));
currentProjectedNode = currentProjectedNode.next;
if (firstProjectedNode !== null && parentView !== null) {
collectNativeNodes(parentView, firstProjectedNode, result, true);
}
}
tNode = tNode.next;
tNode = isProjection ? tNode.projectionNext : tNode.next;
}
return result;