123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
import {Page} from 'puppeteer';import {SELECTION_ID} from '../src/selection';import {embedFn, fill, hits as hitsMaster, pt, spec, testRenderFn} from './util';
declare const page: Page;for (const type of ['single', 'multi']) {  describe(`${type} selections at runtime in unit views`, () => {    beforeAll(async () => {      await page.goto('http://0.0.0.0:8000/test-runtime/');    });
    const hits = hitsMaster.discrete;    const embed = embedFn(page);    const testRender = testRenderFn(page, `${type}/unit`);
    it('should add values to the store', async () => {      for (let i = 0; i < hits.qq.length; i++) {        await embed(spec('unit', i, {type}));        const store = await page.evaluate(pt('qq', i));        expect(store).toHaveLength(1);        expect(store[0].fields).toHaveLength(1);        expect(store[0].values).toHaveLength(1);        expect(store[0].fields[0].field).toEqual(SELECTION_ID);        expect(store[0].fields[0].type).toEqual('E');        await testRender(`click_${i}`);      }    });
    it('should respect projections', async () => {      let values: number[][] = [];      let encodings: string[] = [];      let fields: string[] = [];      const t = async (emb: (i: number) => void) => {        for (let i = 0; i < hits.qq.length; i++) {          emb(i);          const store = await page.evaluate(pt('qq', i));          expect(store).toHaveLength(1);          expect(store[0].fields).toHaveLength(fields.length);          expect(store[0].values).toHaveLength(fields.length);          expect(store[0].fields.map((f: any) => f.field)).toEqual(fields);          expect(store[0].fields.map((f: any) => f.type)).toEqual(fill('E', fields.length));          expect(store[0].values).toEqual(values[i]);          await testRender(`${encodings}_${fields}_${i}`);        }      };
      encodings = ['x', 'color'];      fields = ['a', 'c'];      values = [        [2, 1],        [6, 0]      ];      await t(async (i: number) => await embed(spec('unit', i, {type, encodings})));
      encodings = [];      fields = ['c', 'a', 'b'];      values = [        [1, 2, 53],        [0, 6, 87]      ];      await t(async (i: number) => await embed(spec('unit', i, {type, fields})));    });
    it('should clear out the store', async () => {      for (let i = 0; i < hits.qq_clear.length; i++) {        await embed(spec('unit', i, {type}));        let store = await page.evaluate(pt('qq', i));        expect(store).toHaveLength(1);
        store = await page.evaluate(pt('qq_clear', i));        expect(store).toHaveLength(0);        await testRender(`clear_${i}`);      }    });
    it('should support selecting bins', async () => {      const encodings = ['x', 'color', 'y'];      const fields = ['a', 'c', 'b'];      const types = ['R-RE', 'E', 'R-RE'];      const values = [        [[1, 2], 0, [40, 50]],        [[8, 9], 1, [10, 20]]      ];
      for (let i = 0; i < hits.bins.length; i++) {        await embed(spec('unit', i, {type, encodings}, {x: {bin: true}, y: {bin: true}}));        const store = await page.evaluate(pt('bins', i));        expect(store).toHaveLength(1);        expect(store[0].fields).toHaveLength(fields.length);        expect(store[0].values).toHaveLength(fields.length);        expect(store[0].fields.map((f: any) => f.field)).toEqual(fields);        expect(store[0].fields.map((f: any) => f.type)).toEqual(types);        expect(store[0].values).toEqual(values[i]);        await testRender(`bins_${i}`);      }    });  });}