gristlabs_grist-core/test/nbrowser/chartViewTestUtils.ts

100 lines
3.8 KiB
TypeScript
Raw Normal View History

import {assert, driver, WebElement} from 'mocha-webdriver';
import {Layout, LayoutAxis, PlotData} from 'plotly.js';
import * as gu from 'test/nbrowser/gristUtils';
import isString = require('lodash/isString');
import isUndefined = require('lodash/isUndefined');
export interface ChartData {
data: Partial<PlotData>[];
layout: Layout;
}
export async function selectChartType(chartType: string) {
await driver.find('.test-chart-type').click();
await driver.findContent('.test-select-row', chartType).click();
return gu.waitForServer();
}
export async function getChartData(chartElem?: WebElement|string): Promise<ChartData> {
if (isString(chartElem) || isUndefined(chartElem)) {
const section = isString(chartElem) ?
await gu.getSection(chartElem) :
await driver.findWait('.active_section', 4000);
chartElem = await section.find('.test-chart-container');
}
return driver.executeScript((el: any) => ({data: el.data, layout: el.layout}), chartElem);
}
export function checkAxisRange({layout}: ChartData, xMin: number, xMax: number, yMin: number, yMax: number) {
assert.closeTo(layout.xaxis.range![0], xMin, xMin * 0.1);
assert.closeTo(layout.xaxis.range![1], xMax, xMax * 0.1);
assert.closeTo(layout.yaxis.range![0], yMin, yMin * 0.1);
assert.closeTo(layout.yaxis.range![1], yMax, yMax * 0.1);
}
export function getAxisTitle(axis: Partial<LayoutAxis>): string|undefined {
return axis.title && (axis.title as any).text;
}
export function findYAxis(name: string) {
return driver.findContent('.test-chart-y-axis', name);
}
export async function removeYAxis(name: string) {
await findYAxis(name).mouseMove().find('.test-chart-ref-select-remove').click();
await gu.waitForServer();
}
export async function checkAxisConfig(expected: {groupingByColumn?: string|false,
xaxis: string|undefined, yaxis: string[]}) {
const isGroupByPresent = await driver.find('.test-chart-group-by-column').isPresent();
let groupingByColumn = isGroupByPresent ? await driver.find('.test-chart-group-by-column').getText() : false;
if (groupingByColumn === 'Pick a column') {
groupingByColumn = false;
}
const xaxis = await driver.find('.test-chart-x-axis').getText();
assert.deepEqual({
groupingByColumn,
xaxis: xaxis === 'Pick a column' ? undefined : xaxis,
yaxis: await driver.findAll('.test-chart-y-axis', (e) => e.getText()),
}, {...expected, groupingByColumn: expected.groupingByColumn || false});
}
export async function setSplitSeries(name: string|false, section?: string) {
await gu.openSectionMenu('viewLayout', section);
await driver.findContent('.grist-floating-menu li', 'Widget options').click();
const isChecked = await driver.findContent('label', /Split series/).find('input').matches(':checked');
if (name === false && isChecked === true ||
name && isChecked === false) {
await driver.findContent('label', /Split series/).click();
}
if (name) {
await driver.find('.test-chart-group-by-column').click();
await driver.findContent('.test-select-menu li', name || 'Pick a column').click();
}
await gu.waitForServer();
}
export async function selectXAxis(name: string, opt: {noWait?: boolean} = {}) {
await driver.find('.test-chart-x-axis').click();
await driver.findContent('.test-select-menu li', name).click();
if (!opt.noWait) {
await gu.waitForServer();
}
}
export async function setYAxis(names: string[]) {
// let's first remove all yaxis and then add new ones
const toRemove = await driver.findAll('.test-chart-y-axis', (e) => e.getText());
for (const n of toRemove) { await removeYAxis(n); }
for (const n of names) { await addYAxis(n); }
}
export async function addYAxis(name: string) {
await driver.find('.test-chart-add-y-axis').click();
await driver.findContent('.grist-floating-menu li', name).click();
await gu.waitForServer();
}