fix: 'Memory leak on running more than 11 tests' (close #7188) (#7199)

* fix 'Memory leak on running more than 11 tests' (close #7188) * fix tests in firefox
DevExpress · Jul 28, 2022 · 76e8e9f · 76e8e9f
1 parent 68cc8d6
commit 76e8e9f
Show file tree

Hide file tree

Showing 5 changed files with 41 additions and 6 deletions.
diff --git a/src/browser/provider/built-in/dedicated/chrome/cdp-client/index.ts b/src/browser/provider/built-in/dedicated/chrome/cdp-client/index.ts
@@ -51,8 +51,9 @@ export class BrowserClient {
     private readonly _runtimeInfo: RuntimeInfo;
     private _parentTarget?: remoteChrome.TargetInfo;
     private readonly debugLogger: debug.Debugger;
-    private readonly _videoFramesBuffer: VideoFrameData[];
+    private _videoFramesBuffer: VideoFrameData[];
     private _lastFrame: VideoFrameData | null;
+    private _screencastFrameListenerAttached = false;
 
     public constructor (runtimeInfo: RuntimeInfo) {
         this._runtimeInfo = runtimeInfo;
@@ -358,16 +359,32 @@ export class BrowserClient {
         if (!client)
             return;
 
-        client.Page.on('screencastFrame', (event: ScreencastFrameEvent) => {
-            this._videoFramesBuffer.push({
-                data:      event.data,
-                sessionId: event.sessionId,
+        if (!this._screencastFrameListenerAttached) {
+            client.Page.on('screencastFrame', (event: ScreencastFrameEvent) => {
+                this._videoFramesBuffer.push({
+                    data:      event.data,
+                    sessionId: event.sessionId,
+                });
             });
-        });
+
+            this._screencastFrameListenerAttached = true;
+        }
 
         await client.Page.startScreencast(SCREENCAST_OPTIONS as StartScreencastRequest);
     }
 
+    public async stopCapturingVideo (): Promise<void> {
+        const client = await this.getActiveClient();
+
+        if (!client)
+            return;
+
+        await client.Page.stopScreencast();
+
+        this._lastFrame         = null;
+        this._videoFramesBuffer = [];
+    }
+
     public async getVideoFrameData (): Promise<Buffer | null> {
         const currentVideoFrame = this._videoFramesBuffer.shift() || this._lastFrame;
 

diff --git a/src/browser/provider/built-in/dedicated/chrome/index.js b/src/browser/provider/built-in/dedicated/chrome/index.js
@@ -128,6 +128,12 @@ export default {
         await browserClient.startCapturingVideo();
     },
 
+    async stopCapturingVideo (browserId) {
+        const { browserClient } = this.openedBrowsers[browserId];
+
+        await browserClient.stopCapturingVideo();
+    },
+
     async getVideoFrameData (browserId) {
         const { browserClient } = this.openedBrowsers[browserId];
 

diff --git a/src/browser/provider/index.ts b/src/browser/provider/index.ts
@@ -417,6 +417,10 @@ export default class BrowserProvider {
         await this.plugin.startCapturingVideo(browserId);
     }
 
+    public async stopCapturingVideo (browserId: string): Promise<void> {
+        await this.plugin.stopCapturingVideo(browserId);
+    }
+
     public async hasCustomActionForBrowser (browserId: string): Promise<any> {
         return this.plugin.hasCustomActionForBrowser(browserId);
     }

diff --git a/src/browser/provider/plugin-host.js b/src/browser/provider/plugin-host.js
@@ -143,6 +143,9 @@ export default class BrowserProviderPluginHost {
     async startCapturingVideo (/*browserId*/) {
     }
 
+    async stopCapturingVideo (/*browserId*/) {
+    }
+
     async getVideoFrameData (browserId) {
         const browserAlias = BrowserConnection.getById(browserId).browserInfo.alias;
 

diff --git a/src/video-recorder/process.js b/src/video-recorder/process.js
@@ -130,6 +130,10 @@ export default class VideoRecorder extends AsyncEmitter {
         await this.connection.provider.startCapturingVideo(this.connection.id);
     }
 
+    async _stopCapturing () {
+        await this.connection.provider.stopCapturingVideo(this.connection.id);
+    }
+
     async init () {
         this.ffmpegProcess = spawn(this.ffmpegPath, this.optionsList, { stdio: 'pipe' });
 
@@ -183,6 +187,7 @@ export default class VideoRecorder extends AsyncEmitter {
 
         this.closed = true;
 
+        await this._stopCapturing();
         await this.capturingPromise;
         await this.dispose();
     }