123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
import {Page} from 'puppeteer';import {stringValue} from 'vega-util';import {compositeTypes, embedFn, parentSelector, spec, testRenderFn} from './util';
declare const page: Page;
const hits = {  qq: [8, 19, 13, 21],  qq_clear: [5, 16],  bins: [4, 29, 16, 9],  bins_clear: [18, 7],  composite: [1, 3, 5, 7, 8, 9]};
function toggle(key: string, idx: number, shiftKey: boolean, parent?: string) {  const fn = key.match('_clear') ? 'clear' : 'pt';  return `${fn}(${hits[key][idx]}, ${stringValue(parent)}, ${!!shiftKey})`;}
describe('Toggle multi selections at runtime', () => {  beforeAll(async () => {    await page.goto('http://0.0.0.0:8000/test-runtime/');  });
  const type = 'multi';  const embed = embedFn(page);  const testRender = testRenderFn(page, 'multi/toggle');
  it('should toggle values into/out of the store', async () => {    await embed(spec('unit', 0, {type}));    await page.evaluate(toggle('qq', 0, false));    await page.evaluate(toggle('qq', 1, true));    let store = await page.evaluate(toggle('qq', 2, true));    expect(store).toHaveLength(3);    await testRender('click_0');
    store = await page.evaluate(toggle('qq', 2, true));    expect(store).toHaveLength(2);    await testRender('click_1');
    store = await page.evaluate(toggle('qq', 3, false));    expect(store).toHaveLength(1);    await testRender('click_2');  });
  it('should clear out the store w/o shiftKey', async () => {    await embed(spec('unit', 1, {type}));    await page.evaluate(toggle('qq', 0, false));    await page.evaluate(toggle('qq', 1, true));    await page.evaluate(toggle('qq', 2, true));    await page.evaluate(toggle('qq', 3, true));    await testRender(`clear_0`);
    let store = await page.evaluate(toggle('qq_clear', 0, true));    expect(store).toHaveLength(4);    await testRender(`clear_1`);
    store = await page.evaluate(toggle('qq_clear', 1, false));    expect(store).toHaveLength(0);    await testRender(`clear_2`);  });
  it('should toggle binned fields', async () => {    await embed(spec('unit', 0, {type, encodings: ['x', 'y']}, {x: {bin: true}, y: {bin: true}}));
    await page.evaluate(toggle('bins', 0, false));    await page.evaluate(toggle('bins', 1, true));    let store = await page.evaluate(toggle('bins', 2, true));    expect(store).toHaveLength(3);    await testRender('bins_0');
    store = await page.evaluate(toggle('bins', 2, true));    expect(store).toHaveLength(2);    await testRender('bins_1');
    store = await page.evaluate(toggle('bins', 3, false));    expect(store).toHaveLength(1);    await testRender('bins_2');  });
  compositeTypes.forEach((specType, idx) => {    it(`should toggle in ${specType} views`, async () => {      await embed(spec(specType, idx, {type, resolve: 'union'}));      let length = 0;      for (let i = 0; i < hits.composite.length; i++) {        const parent = parentSelector(specType, i % 3);        const store = (await page.evaluate(toggle('composite', i, true, parent))) as string;        expect((length = store.length)).toEqual(i + 1);        if (i % 3 === 2) {          await testRender(`${specType}_${i}`);        }      }
      for (let i = 0; i < hits.composite.length; i++) {        const even = i % 2 === 0;        const parent = parentSelector(specType, ~~(i / 2));        const store = await page.evaluate(toggle('qq_clear', 0, even, parent));        expect(store).toHaveLength(even ? length : (length = length - 2));        if (!even) {          await testRender(`${specType}_clear_${i}`);        }      }    });  });});