125 lines
4.2 KiB
TypeScript
125 lines
4.2 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright Google LLC 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.dev/license
|
|
*/
|
|
|
|
import {
|
|
ComponentHarness,
|
|
ComponentHarnessConstructor,
|
|
HarnessPredicate,
|
|
parallel,
|
|
} from '@angular/cdk/testing';
|
|
import {
|
|
_MatCellHarnessBase,
|
|
MatCellHarness,
|
|
MatFooterCellHarness,
|
|
MatHeaderCellHarness,
|
|
} from './cell-harness';
|
|
import {CellHarnessFilters, RowHarnessFilters} from './table-harness-filters';
|
|
|
|
/** Text extracted from a table row organized by columns. */
|
|
export interface MatRowHarnessColumnsText {
|
|
[columnName: string]: string;
|
|
}
|
|
|
|
export abstract class _MatRowHarnessBase<
|
|
CellType extends ComponentHarnessConstructor<Cell> & {
|
|
with: (options?: CellHarnessFilters) => HarnessPredicate<Cell>;
|
|
},
|
|
Cell extends _MatCellHarnessBase,
|
|
> extends ComponentHarness {
|
|
protected abstract _cellHarness: CellType;
|
|
|
|
/** Gets a list of `MatCellHarness` for all cells in the row. */
|
|
async getCells(filter: CellHarnessFilters = {}): Promise<Cell[]> {
|
|
return this.locatorForAll(this._cellHarness.with(filter))();
|
|
}
|
|
|
|
/** Gets the text of the cells in the row. */
|
|
async getCellTextByIndex(filter: CellHarnessFilters = {}): Promise<string[]> {
|
|
const cells = await this.getCells(filter);
|
|
return parallel(() => cells.map(cell => cell.getText()));
|
|
}
|
|
|
|
/** Gets the text inside the row organized by columns. */
|
|
async getCellTextByColumnName(): Promise<MatRowHarnessColumnsText> {
|
|
const output: MatRowHarnessColumnsText = {};
|
|
const cells = await this.getCells();
|
|
const cellsData = await parallel(() =>
|
|
cells.map(cell => {
|
|
return parallel(() => [cell.getColumnName(), cell.getText()]);
|
|
}),
|
|
);
|
|
cellsData.forEach(([columnName, text]) => (output[columnName] = text));
|
|
return output;
|
|
}
|
|
}
|
|
|
|
/** Harness for interacting with an Angular Material table row. */
|
|
export class MatRowHarness extends _MatRowHarnessBase<typeof MatCellHarness, MatCellHarness> {
|
|
/** The selector for the host element of a `MatRowHarness` instance. */
|
|
static hostSelector = '.mat-mdc-row';
|
|
protected _cellHarness = MatCellHarness;
|
|
|
|
/**
|
|
* Gets a `HarnessPredicate` that can be used to search for a table row with specific attributes.
|
|
* @param options Options for narrowing the search
|
|
* @return a `HarnessPredicate` configured with the given options.
|
|
*/
|
|
static with<T extends MatRowHarness>(
|
|
this: ComponentHarnessConstructor<T>,
|
|
options: RowHarnessFilters = {},
|
|
): HarnessPredicate<T> {
|
|
return new HarnessPredicate(this, options);
|
|
}
|
|
}
|
|
|
|
/** Harness for interacting with an Angular Material table header row. */
|
|
export class MatHeaderRowHarness extends _MatRowHarnessBase<
|
|
typeof MatHeaderCellHarness,
|
|
MatHeaderCellHarness
|
|
> {
|
|
/** The selector for the host element of a `MatHeaderRowHarness` instance. */
|
|
static hostSelector = '.mat-mdc-header-row';
|
|
protected _cellHarness = MatHeaderCellHarness;
|
|
|
|
/**
|
|
* Gets a `HarnessPredicate` that can be used to search for a table header row with specific
|
|
* attributes.
|
|
* @param options Options for narrowing the search
|
|
* @return a `HarnessPredicate` configured with the given options.
|
|
*/
|
|
static with<T extends MatHeaderRowHarness>(
|
|
this: ComponentHarnessConstructor<T>,
|
|
options: RowHarnessFilters = {},
|
|
): HarnessPredicate<T> {
|
|
return new HarnessPredicate(this, options);
|
|
}
|
|
}
|
|
|
|
/** Harness for interacting with an Angular Material table footer row. */
|
|
export class MatFooterRowHarness extends _MatRowHarnessBase<
|
|
typeof MatFooterCellHarness,
|
|
MatFooterCellHarness
|
|
> {
|
|
/** The selector for the host element of a `MatFooterRowHarness` instance. */
|
|
static hostSelector = '.mat-mdc-footer-row';
|
|
protected _cellHarness = MatFooterCellHarness;
|
|
|
|
/**
|
|
* Gets a `HarnessPredicate` that can be used to search for a table footer row cell with specific
|
|
* attributes.
|
|
* @param options Options for narrowing the search
|
|
* @return a `HarnessPredicate` configured with the given options.
|
|
*/
|
|
static with<T extends MatFooterRowHarness>(
|
|
this: ComponentHarnessConstructor<T>,
|
|
options: RowHarnessFilters = {},
|
|
): HarnessPredicate<T> {
|
|
return new HarnessPredicate(this, options);
|
|
}
|
|
}
|