/** * @license * Copyright Google Inc. All Rights Reserved. * * Use of this source code is governed by an MIT-style license that can be * found in the LICENSE file at https://angular.io/license */ import {D, E, Q, QueryList, e, m, qR} from '../../src/render3/index'; import {createComponent, renderComponent} from './render_util'; describe('query', () => { it('should project query children', () => { const Child = createComponent('child', function(ctx: any, cm: boolean) {}); let child1 = null; let child2 = null; const Cmp = createComponent('cmp', function(ctx: any, cm: boolean) { /** * * * * * class Cmp { * @ViewChildren(Child) query0; * @ViewChildren(Child, {descend: true}) query1; * } */ let tmp: any; if (cm) { m(0, Q(Child, false)); m(1, Q(Child, true)); E(2, Child.ngComponentDef); { child1 = D(3, Child.ngComponentDef.n(), Child.ngComponentDef); E(4, Child.ngComponentDef); { child2 = D(5, Child.ngComponentDef.n(), Child.ngComponentDef); } e(); } e(); } qR(tmp = m>(0)) && (ctx.query0 = tmp as QueryList); qR(tmp = m>(1)) && (ctx.query1 = tmp as QueryList); }); const parent = renderComponent(Cmp); expect((parent.query0 as QueryList).toArray()).toEqual([child1]); expect((parent.query1 as QueryList).toArray()).toEqual([child1, child2]); }); describe('local names', () => { it('should query for a single element', () => { let elToQuery; /** *
*
* class Cmpt { * @ViewChildren('foo') query; * } */ const Cmpt = createComponent('cmpt', function(ctx: any, cm: boolean) { let tmp: any; if (cm) { m(0, Q(['foo'])); elToQuery = E(1, 'div', [], 'foo'); e(); E(2, 'div'); e(); } qR(tmp = m>(0)) && (ctx.query = tmp as QueryList); }); const cmptInstance = renderComponent(Cmpt); const query = (cmptInstance.query as QueryList); expect(query.length).toBe(1); expect(query.first.nativeElement).toEqual(elToQuery); }); it('should query for multiple elements', () => { let el1ToQuery; let el2ToQuery; /** *
*
*
* class Cmpt { * @ViewChildren('foo,bar') query; * } */ const Cmpt = createComponent('cmpt', function(ctx: any, cm: boolean) { let tmp: any; if (cm) { m(0, Q(['foo', 'bar'])); el1ToQuery = E(1, 'div', null, 'foo'); e(); E(2, 'div'); e(); el2ToQuery = E(3, 'div', null, 'bar'); e(); } qR(tmp = m>(0)) && (ctx.query = tmp as QueryList); }); const cmptInstance = renderComponent(Cmpt); const query = (cmptInstance.query as QueryList); expect(query.length).toBe(2); expect(query.first.nativeElement).toEqual(el1ToQuery); expect(query.last.nativeElement).toEqual(el2ToQuery); }); }); });