????
Your IP : 216.73.216.75
"use strict";
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.PythonLocatorFactory = exports.JavaScriptLocatorFactory = exports.JavaLocatorFactory = exports.CSharpLocatorFactory = void 0;
exports.asLocator = asLocator;
exports.isRegExp = isRegExp;
var _stringUtils = require("./stringUtils");
var _selectorParser = require("./selectorParser");
/**
* Copyright (c) Microsoft Corporation.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
function asLocator(lang, selector, isFrameLocator = false, playSafe = false) {
if (playSafe) {
try {
return innerAsLocator(generators[lang], (0, _selectorParser.parseSelector)(selector), isFrameLocator);
} catch (e) {
// Tolerate invalid input.
return selector;
}
} else {
return innerAsLocator(generators[lang], (0, _selectorParser.parseSelector)(selector), isFrameLocator);
}
}
function innerAsLocator(factory, parsed, isFrameLocator = false) {
const parts = [...parsed.parts];
// frameLocator('iframe').first is actually "iframe >> nth=0 >> internal:control=enter-frame"
// To make it easier to parse, we turn it into "iframe >> internal:control=enter-frame >> nth=0"
for (let index = 0; index < parts.length - 1; index++) {
if (parts[index].name === 'nth' && parts[index + 1].name === 'internal:control' && parts[index + 1].body === 'enter-frame') {
// Swap nth and enter-frame.
const [nth] = parts.splice(index, 1);
parts.splice(index + 1, 0, nth);
}
}
const tokens = [];
let nextBase = isFrameLocator ? 'frame-locator' : 'page';
for (let index = 0; index < parts.length; index++) {
const part = parts[index];
const base = nextBase;
nextBase = 'locator';
if (part.name === 'nth') {
if (part.body === '0') tokens.push(factory.generateLocator(base, 'first', ''));else if (part.body === '-1') tokens.push(factory.generateLocator(base, 'last', ''));else tokens.push(factory.generateLocator(base, 'nth', part.body));
continue;
}
if (part.name === 'internal:text') {
const {
exact,
text
} = detectExact(part.body);
tokens.push(factory.generateLocator(base, 'text', text, {
exact
}));
continue;
}
if (part.name === 'internal:has-text') {
const {
exact,
text
} = detectExact(part.body);
// There is no locator equivalent for strict has-text, leave it as is.
if (!exact) {
tokens.push(factory.generateLocator(base, 'has-text', text, {
exact
}));
continue;
}
}
if (part.name === 'internal:has') {
const inner = innerAsLocator(factory, part.body.parsed);
tokens.push(factory.generateLocator(base, 'has', inner));
continue;
}
if (part.name === 'internal:label') {
const {
exact,
text
} = detectExact(part.body);
tokens.push(factory.generateLocator(base, 'label', text, {
exact
}));
continue;
}
if (part.name === 'internal:role') {
const attrSelector = (0, _selectorParser.parseAttributeSelector)(part.body, true);
const options = {
attrs: []
};
for (const attr of attrSelector.attributes) {
if (attr.name === 'name') {
options.exact = attr.caseSensitive;
options.name = attr.value;
} else {
if (attr.name === 'level' && typeof attr.value === 'string') attr.value = +attr.value;
options.attrs.push({
name: attr.name === 'include-hidden' ? 'includeHidden' : attr.name,
value: attr.value
});
}
}
tokens.push(factory.generateLocator(base, 'role', attrSelector.name, options));
continue;
}
if (part.name === 'internal:testid') {
const attrSelector = (0, _selectorParser.parseAttributeSelector)(part.body, true);
const {
value
} = attrSelector.attributes[0];
tokens.push(factory.generateLocator(base, 'test-id', value));
continue;
}
if (part.name === 'internal:attr') {
const attrSelector = (0, _selectorParser.parseAttributeSelector)(part.body, true);
const {
name,
value,
caseSensitive
} = attrSelector.attributes[0];
const text = value;
const exact = !!caseSensitive;
if (name === 'placeholder') {
tokens.push(factory.generateLocator(base, 'placeholder', text, {
exact
}));
continue;
}
if (name === 'alt') {
tokens.push(factory.generateLocator(base, 'alt', text, {
exact
}));
continue;
}
if (name === 'title') {
tokens.push(factory.generateLocator(base, 'title', text, {
exact
}));
continue;
}
}
let locatorType = 'default';
const nextPart = parts[index + 1];
if (nextPart && nextPart.name === 'internal:control' && nextPart.body === 'enter-frame') {
locatorType = 'frame';
nextBase = 'frame-locator';
index++;
}
const p = {
parts: [part]
};
tokens.push(factory.generateLocator(base, locatorType, (0, _selectorParser.stringifySelector)(p)));
}
return tokens.join('.');
}
function detectExact(text) {
let exact = false;
const match = text.match(/^\/(.*)\/([igm]*)$/);
if (match) return {
text: new RegExp(match[1], match[2])
};
if (text.endsWith('"')) {
text = JSON.parse(text);
exact = true;
} else if (text.endsWith('"s')) {
text = JSON.parse(text.substring(0, text.length - 1));
exact = true;
} else if (text.endsWith('"i')) {
text = JSON.parse(text.substring(0, text.length - 1));
exact = false;
}
return {
exact,
text
};
}
class JavaScriptLocatorFactory {
generateLocator(base, kind, body, options = {}) {
switch (kind) {
case 'default':
return `locator(${this.quote(body)})`;
case 'frame':
return `frameLocator(${this.quote(body)})`;
case 'nth':
return `nth(${body})`;
case 'first':
return `first()`;
case 'last':
return `last()`;
case 'role':
const attrs = [];
if (isRegExp(options.name)) {
attrs.push(`name: ${options.name}`);
} else if (typeof options.name === 'string') {
attrs.push(`name: ${this.quote(options.name)}`);
if (options.exact) attrs.push(`exact: true`);
}
for (const {
name,
value
} of options.attrs) attrs.push(`${name}: ${typeof value === 'string' ? this.quote(value) : value}`);
const attrString = attrs.length ? `, { ${attrs.join(', ')} }` : '';
return `getByRole(${this.quote(body)}${attrString})`;
case 'has-text':
return `filter({ hasText: ${this.toHasText(body)} })`;
case 'has':
return `filter({ has: ${body} })`;
case 'test-id':
return `getByTestId(${this.quote(body)})`;
case 'text':
return this.toCallWithExact('getByText', body, !!options.exact);
case 'alt':
return this.toCallWithExact('getByAltText', body, !!options.exact);
case 'placeholder':
return this.toCallWithExact('getByPlaceholder', body, !!options.exact);
case 'label':
return this.toCallWithExact('getByLabel', body, !!options.exact);
case 'title':
return this.toCallWithExact('getByTitle', body, !!options.exact);
default:
throw new Error('Unknown selector kind ' + kind);
}
}
toCallWithExact(method, body, exact) {
if (isRegExp(body)) return `${method}(${body})`;
return exact ? `${method}(${this.quote(body)}, { exact: true })` : `${method}(${this.quote(body)})`;
}
toHasText(body) {
if (isRegExp(body)) return String(body);
return this.quote(body);
}
quote(text) {
return (0, _stringUtils.escapeWithQuotes)(text, '\'');
}
}
exports.JavaScriptLocatorFactory = JavaScriptLocatorFactory;
class PythonLocatorFactory {
generateLocator(base, kind, body, options = {}) {
switch (kind) {
case 'default':
return `locator(${this.quote(body)})`;
case 'frame':
return `frame_locator(${this.quote(body)})`;
case 'nth':
return `nth(${body})`;
case 'first':
return `first`;
case 'last':
return `last`;
case 'role':
const attrs = [];
if (isRegExp(options.name)) {
attrs.push(`name=${this.regexToString(options.name)}`);
} else if (typeof options.name === 'string') {
attrs.push(`name=${this.quote(options.name)}`);
if (options.exact) attrs.push(`exact=True`);
}
for (const {
name,
value
} of options.attrs) {
let valueString = typeof value === 'string' ? this.quote(value) : value;
if (typeof value === 'boolean') valueString = value ? 'True' : 'False';
attrs.push(`${(0, _stringUtils.toSnakeCase)(name)}=${valueString}`);
}
const attrString = attrs.length ? `, ${attrs.join(', ')}` : '';
return `get_by_role(${this.quote(body)}${attrString})`;
case 'has-text':
return `filter(has_text=${this.toHasText(body)})`;
case 'has':
return `filter(has=${body})`;
case 'test-id':
return `get_by_test_id(${this.quote(body)})`;
case 'text':
return this.toCallWithExact('get_by_text', body, !!options.exact);
case 'alt':
return this.toCallWithExact('get_by_alt_text', body, !!options.exact);
case 'placeholder':
return this.toCallWithExact('get_by_placeholder', body, !!options.exact);
case 'label':
return this.toCallWithExact('get_by_label', body, !!options.exact);
case 'title':
return this.toCallWithExact('get_by_title', body, !!options.exact);
default:
throw new Error('Unknown selector kind ' + kind);
}
}
regexToString(body) {
const suffix = body.flags.includes('i') ? ', re.IGNORECASE' : '';
return `re.compile(r"${body.source.replace(/\\\//, '/').replace(/"/g, '\\"')}"${suffix})`;
}
toCallWithExact(method, body, exact) {
if (isRegExp(body)) return `${method}(${this.regexToString(body)})`;
if (exact) return `${method}(${this.quote(body)}, exact=True)`;
return `${method}(${this.quote(body)})`;
}
toHasText(body) {
if (isRegExp(body)) return this.regexToString(body);
return `${this.quote(body)}`;
}
quote(text) {
return (0, _stringUtils.escapeWithQuotes)(text, '\"');
}
}
exports.PythonLocatorFactory = PythonLocatorFactory;
class JavaLocatorFactory {
generateLocator(base, kind, body, options = {}) {
let clazz;
switch (base) {
case 'page':
clazz = 'Page';
break;
case 'frame-locator':
clazz = 'FrameLocator';
break;
case 'locator':
clazz = 'Locator';
break;
}
switch (kind) {
case 'default':
return `locator(${this.quote(body)})`;
case 'frame':
return `frameLocator(${this.quote(body)})`;
case 'nth':
return `nth(${body})`;
case 'first':
return `first()`;
case 'last':
return `last()`;
case 'role':
const attrs = [];
if (isRegExp(options.name)) {
attrs.push(`.setName(${this.regexToString(options.name)})`);
} else if (typeof options.name === 'string') {
attrs.push(`.setName(${this.quote(options.name)})`);
if (options.exact) attrs.push(`.setExact(true)`);
}
for (const {
name,
value
} of options.attrs) attrs.push(`.set${(0, _stringUtils.toTitleCase)(name)}(${typeof value === 'string' ? this.quote(value) : value})`);
const attrString = attrs.length ? `, new ${clazz}.GetByRoleOptions()${attrs.join('')}` : '';
return `getByRole(AriaRole.${(0, _stringUtils.toSnakeCase)(body).toUpperCase()}${attrString})`;
case 'has-text':
return `filter(new ${clazz}.FilterOptions().setHasText(${this.toHasText(body)}))`;
case 'has':
return `filter(new ${clazz}.FilterOptions().setHas(${body}))`;
case 'test-id':
return `getByTestId(${this.quote(body)})`;
case 'text':
return this.toCallWithExact(clazz, 'getByText', body, !!options.exact);
case 'alt':
return this.toCallWithExact(clazz, 'getByAltText', body, !!options.exact);
case 'placeholder':
return this.toCallWithExact(clazz, 'getByPlaceholder', body, !!options.exact);
case 'label':
return this.toCallWithExact(clazz, 'getByLabel', body, !!options.exact);
case 'title':
return this.toCallWithExact(clazz, 'getByTitle', body, !!options.exact);
default:
throw new Error('Unknown selector kind ' + kind);
}
}
regexToString(body) {
const suffix = body.flags.includes('i') ? ', Pattern.CASE_INSENSITIVE' : '';
return `Pattern.compile(${this.quote(body.source)}${suffix})`;
}
toCallWithExact(clazz, method, body, exact) {
if (isRegExp(body)) return `${method}(${this.regexToString(body)})`;
if (exact) return `${method}(${this.quote(body)}, new ${clazz}.${(0, _stringUtils.toTitleCase)(method)}Options().setExact(true))`;
return `${method}(${this.quote(body)})`;
}
toHasText(body) {
if (isRegExp(body)) return this.regexToString(body);
return this.quote(body);
}
quote(text) {
return (0, _stringUtils.escapeWithQuotes)(text, '\"');
}
}
exports.JavaLocatorFactory = JavaLocatorFactory;
class CSharpLocatorFactory {
generateLocator(base, kind, body, options = {}) {
switch (kind) {
case 'default':
return `Locator(${this.quote(body)})`;
case 'frame':
return `FrameLocator(${this.quote(body)})`;
case 'nth':
return `Nth(${body})`;
case 'first':
return `First`;
case 'last':
return `Last`;
case 'role':
const attrs = [];
if (isRegExp(options.name)) {
attrs.push(`NameRegex = ${this.regexToString(options.name)}`);
} else if (typeof options.name === 'string') {
attrs.push(`Name = ${this.quote(options.name)}`);
if (options.exact) attrs.push(`Exact = true`);
}
for (const {
name,
value
} of options.attrs) attrs.push(`${(0, _stringUtils.toTitleCase)(name)} = ${typeof value === 'string' ? this.quote(value) : value}`);
const attrString = attrs.length ? `, new() { ${attrs.join(', ')} }` : '';
return `GetByRole(AriaRole.${(0, _stringUtils.toTitleCase)(body)}${attrString})`;
case 'has-text':
return `Filter(new() { ${this.toHasText(body)} })`;
case 'has':
return `Filter(new() { Has = ${body} })`;
case 'test-id':
return `GetByTestId(${this.quote(body)})`;
case 'text':
return this.toCallWithExact('GetByText', body, !!options.exact);
case 'alt':
return this.toCallWithExact('GetByAltText', body, !!options.exact);
case 'placeholder':
return this.toCallWithExact('GetByPlaceholder', body, !!options.exact);
case 'label':
return this.toCallWithExact('GetByLabel', body, !!options.exact);
case 'title':
return this.toCallWithExact('GetByTitle', body, !!options.exact);
default:
throw new Error('Unknown selector kind ' + kind);
}
}
regexToString(body) {
const suffix = body.flags.includes('i') ? ', RegexOptions.IgnoreCase' : '';
return `new Regex(${this.quote(body.source)}${suffix})`;
}
toCallWithExact(method, body, exact) {
if (isRegExp(body)) return `${method}(${this.regexToString(body)})`;
if (exact) return `${method}(${this.quote(body)}, new() { Exact = true })`;
return `${method}(${this.quote(body)})`;
}
toHasText(body) {
if (isRegExp(body)) return `HasTextRegex = ${this.regexToString(body)}`;
return `HasText = ${this.quote(body)}`;
}
quote(text) {
return (0, _stringUtils.escapeWithQuotes)(text, '\"');
}
}
exports.CSharpLocatorFactory = CSharpLocatorFactory;
const generators = {
javascript: new JavaScriptLocatorFactory(),
python: new PythonLocatorFactory(),
java: new JavaLocatorFactory(),
csharp: new CSharpLocatorFactory()
};
function isRegExp(obj) {
return obj instanceof RegExp;
}