fix(testplane): fix ff and add logs

Author: rocketraccoon

src/browser/commands/restoreState/index.ts

@@ -1,25 +1,26 @@
 import fs from "fs-extra";
+import _ from "lodash";
 
 import { clearAllIndexedDB } from "./clearAllIndexedDB";
 import { restoreIndexedDB } from "./restoreIndexedDB";
 import { restoreStorage } from "./restoreStorage";
 
-import _ from "lodash";
+import * as logger from "../../../utils/logger";
 import type { Browser } from "../../types";
 import { DEVTOOLS_PROTOCOL, WEBDRIVER_PROTOCOL } from "../../../constants/config";
 import { defaultOptions, getWebdriverFrames, SaveStateData, SaveStateOptions } from "../saveState";
 import { Protocol } from "devtools-protocol";
 
-type RestoreStateOptions = SaveStateOptions;
-
 export default (browser: Browser): void => {
     const { publicAPI: session } = browser;
 
-    session.addCommand("restoreState", async (_options: RestoreStateOptions) => {
+    session.addCommand("restoreState", async (_options: SaveStateOptions) => {
         const options = { ...defaultOptions, ..._options };
 
         const restoreState: SaveStateData = await fs.readJson(options.path);
 
+        logger.log("Restore state", options);
+
         switch (browser.config.automationProtocol) {
             case WEBDRIVER_PROTOCOL: {
                 if (restoreState.cookies && options.cookies) {
@@ -126,5 +127,7 @@ export default (browser: Browser): void => {
                 break;
             }
         }
+
+        logger.log("State restored");
     });
 };

src/browser/commands/saveState/index.ts

@@ -1,5 +1,6 @@
 import fs from "fs-extra";
 
+import * as logger from "../../../utils/logger";
 import type { Browser } from "../../types";
 import { DumpIndexDB, dumpIndexedDB } from "./dumpIndexedDB";
 import { dumpStorage, StorageData } from "./dumpStorage";
@@ -48,6 +49,8 @@ export default (browser: Browser): void => {
             framesData: {},
         };
 
+        logger.log("Save state", options);
+
         switch (browser.config.automationProtocol) {
             case WEBDRIVER_PROTOCOL: {
                 if (options.cookies) {
@@ -165,6 +168,7 @@ export default (browser: Browser): void => {
         }
 
         if (options && options.path) {
+            logger.log("State saved");
             await fs.writeJson(options.path, data, { spaces: 2 });
         }
     });

src/browser/existing-browser.ts

@@ -10,7 +10,7 @@ import { Camera, PageMeta } from "./camera";
 import { type ClientBridge, build as buildClientBridge } from "./client-bridge";
 import * as history from "./history";
 import * as logger from "../utils/logger";
-import { WEBDRIVER_PROTOCOL } from "../constants/config";
+import { DEVTOOLS_PROTOCOL, WEBDRIVER_PROTOCOL } from "../constants/config";
 import { MIN_CHROME_VERSION_SUPPORT_ISOLATION } from "../constants/browser";
 import { isSupportIsolation } from "../utils/browser";
 import { isRunInNodeJsEnv } from "../utils/config";
@@ -98,7 +98,9 @@ export class ExistingBrowser extends Browser {
 
                 await isolationPromise;
 
-                await this.startCollectCookies();
+                if (this.config.automationProtocol === DEVTOOLS_PROTOCOL) {
+                    await this.startCollectCookies();
+                }
 
                 this._callstackHistory?.clear();