playwright/src/client/video.ts
Dmitry Gozman b29b7df47e
fix(connect): handle disconnect in various situations (#6276)
There are a few ways for `connect()` to finish:
- `Browser.close()` from the client side.
- Browser on the server side did exit (e.g. crashed).
- Connection was dropped by either of the sides.

We reduce all the cases to the last one by dropping the
connection when client wants calls `Browser.close()` or
server-side browser exits.

In all these cases we should properly cleanup on the server side,
and ensure that all promises reject on the client side.
2021-05-06 09:34:06 -07:00

62 lines
2 KiB
TypeScript

/**
* 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.
*/
import { Page } from './page';
import * as api from '../../types/types';
import { Artifact } from './artifact';
export class Video implements api.Video {
private _artifact: Promise<Artifact | null> | null = null;
private _artifactCallback = (artifact: Artifact) => {};
private _isRemote = false;
constructor(page: Page) {
const browser = page.context()._browser;
this._isRemote = !!browser && !!browser._remoteType;
this._artifact = Promise.race([
new Promise<Artifact>(f => this._artifactCallback = f),
page._closedOrCrashedPromise.then(() => null),
]);
}
_artifactReady(artifact: Artifact) {
artifact._isRemote = this._isRemote;
this._artifactCallback(artifact);
}
async path(): Promise<string> {
if (this._isRemote)
throw new Error(`Path is not available when using browserType.connect(). Use saveAs() to save a local copy.`);
const artifact = await this._artifact;
if (!artifact)
throw new Error('Page did not produce any video frames');
return artifact._initializer.absolutePath;
}
async saveAs(path: string): Promise<void> {
const artifact = await this._artifact;
if (!artifact)
throw new Error('Page did not produce any video frames');
return artifact.saveAs(path);
}
async delete(): Promise<void> {
const artifact = await this._artifact;
if (artifact)
await artifact.delete();
}
}