61 lines
2.4 KiB
TypeScript
61 lines
2.4 KiB
TypeScript
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
import { EOL } from "os";
|
|
import { ILogger, LogLevel } from "../src/ILogger";
|
|
import { HttpTransportType } from "../src/ITransport";
|
|
|
|
export function eachTransport(action: (transport: HttpTransportType) => void) {
|
|
const transportTypes = [
|
|
HttpTransportType.WebSockets,
|
|
HttpTransportType.ServerSentEvents,
|
|
HttpTransportType.LongPolling ];
|
|
transportTypes.forEach((t) => action(t));
|
|
}
|
|
|
|
export function eachEndpointUrl(action: (givenUrl: string, expectedUrl: string) => void) {
|
|
const urls = [
|
|
[ "http://tempuri.org/endpoint/?q=my/Data", "http://tempuri.org/endpoint/negotiate?q=my/Data" ],
|
|
[ "http://tempuri.org/endpoint?q=my/Data", "http://tempuri.org/endpoint/negotiate?q=my/Data" ],
|
|
[ "http://tempuri.org/endpoint", "http://tempuri.org/endpoint/negotiate" ],
|
|
[ "http://tempuri.org/endpoint/", "http://tempuri.org/endpoint/negotiate" ],
|
|
];
|
|
|
|
urls.forEach((t) => action(t[0], t[1]));
|
|
}
|
|
|
|
type ErrorMatchFunction = (error: string) => boolean;
|
|
|
|
export class VerifyLogger implements ILogger {
|
|
public unexpectedErrors: string[];
|
|
private expectedErrors: ErrorMatchFunction[];
|
|
|
|
public constructor(...expectedErrors: Array<RegExp | string | ErrorMatchFunction>) {
|
|
this.unexpectedErrors = [];
|
|
this.expectedErrors = [];
|
|
expectedErrors.forEach((element) => {
|
|
if (element instanceof RegExp) {
|
|
this.expectedErrors.push((e) => element.test(e));
|
|
} else if (typeof element === "string") {
|
|
this.expectedErrors.push((e) => element === e);
|
|
} else {
|
|
this.expectedErrors.push(element);
|
|
}
|
|
}, this);
|
|
}
|
|
|
|
public static async run(fn: (logger: VerifyLogger) => Promise<void>, ...expectedErrors: Array<RegExp | string | ErrorMatchFunction>): Promise<void> {
|
|
const logger = new VerifyLogger(...expectedErrors);
|
|
await fn(logger);
|
|
expect(logger.unexpectedErrors.join(EOL)).toBe("");
|
|
}
|
|
|
|
public log(logLevel: LogLevel, message: string): void {
|
|
if (logLevel >= LogLevel.Error) {
|
|
if (!this.expectedErrors.some((fn) => fn(message))) {
|
|
this.unexpectedErrors.push(message);
|
|
}
|
|
}
|
|
}
|
|
}
|