@mablhq/mabl-cli 2.46.4 → 2.49.0

package/http/httpUtil.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.postProcessHeaders = exports.getHttpAgent = exports.GENERAL_API_STEP_EXECUTION_ERROR_CODE = exports.EXECUTION_ENGINE_SSL_VERIFY = exports.USER_AGENT_HEADER = exports.USER_AGENT = void 0;
+const agent_base_1 = require("agent-base");
 const socks_proxy_agent_1 = require("socks-proxy-agent");
 const pureUtil_1 = require("../util/pureUtil");
 const MablHttpAgent_1 = require("./MablHttpAgent");
@@ -9,8 +10,8 @@ exports.USER_AGENT = `${(0, pureUtil_1.getCliName)()}@${(0, pureUtil_1.getCliVer
 exports.USER_AGENT_HEADER = 'User-Agent';
 exports.EXECUTION_ENGINE_SSL_VERIFY = false;
 exports.GENERAL_API_STEP_EXECUTION_ERROR_CODE = '-1';
-function getHttpAgent(shouldFilterHttpRequests, linkServer, proxyConfig) {
-    const linkAgent = maybeGetSocksAgent(linkServer);
+function getHttpAgent(shouldFilterHttpRequests, linkServers, proxyConfig) {
+    const linkAgent = maybeGetSocksAgent(linkServers);
     if (linkAgent) {
         return MablHttpAgent_1.MablHttpAgent.createNewInstance({
             httpAgent: linkAgent,
@@ -23,11 +24,11 @@ function getHttpAgent(shouldFilterHttpRequests, linkServer, proxyConfig) {
     return MablHttpAgent_1.MablHttpAgent.createNewInstance({ proxyConfig });
 }
 exports.getHttpAgent = getHttpAgent;
-function maybeGetSocksAgent(linkServer) {
-    if (!linkServer) {
+function maybeGetSocksAgent(linkServers) {
+    if (!(linkServers === null || linkServers === void 0 ? void 0 : linkServers.length)) {
         return;
     }
-    const agentOptions = {
+    const agentOptions = linkServers.map((linkServer) => ({
         host: linkServer.forwarding_host,
         port: linkServer.forwarding_port,
         userId: linkServer.workspace_id,
@@ -35,8 +36,26 @@ function maybeGetSocksAgent(linkServer) {
         tls: {
             rejectUnauthorized: false,
         },
-    };
-    return new socks_proxy_agent_1.SocksProxyAgent(agentOptions);
+    }));
+    if (agentOptions.length === 1) {
+        return new socks_proxy_agent_1.SocksProxyAgent(agentOptions[0]);
+    }
+    return new MultiSocksProxyAgent(agentOptions);
+}
+class MultiSocksProxyAgent extends agent_base_1.Agent {
+    constructor(options) {
+        super();
+        this.agents = options.map((proxy) => new socks_proxy_agent_1.SocksProxyAgent(proxy));
+    }
+    async callback(req, opts) {
+        return this.selectAgent().callback(req, opts);
+    }
+    selectAgent() {
+        return this.agents[Math.floor(Math.random() * this.agents.length)];
+    }
+    destroy() {
+        this.agents.forEach((agent) => agent.destroy());
+    }
 }
 function postProcessHeaders(response) {
     var _a, _b, _c;

package/index.d.ts

@@ -1,4 +1,5 @@
 import {BrowserContext, Locator, Page, Frame} from 'playwright-core';
+import {TestType} from '@playwright/test';
 
 declare module '@mablhq/mabl-cli' {
   export enum TestResultStatus {
@@ -147,8 +148,10 @@ export declare enum MobilePlatformEnum {
 // Playwright external tool types
 
 export interface MablToolsetOptions {
+  apiKey: string;
   browserContext?: BrowserContext;
   workspaceId?: string;
+  test: TestType<any, any>;
 }
 
 export interface GenAiAssertionResponse {
@@ -181,7 +184,7 @@ export interface DatabaseQueryResultSet {
 }
 
 export interface MablToolset {
-  constructor(options?: MablToolsetOptions);
+  constructor(options: MablToolsetOptions);
   evaluateGenAiAssertion(
     // TODO this is any because it needs to align with the @playwright/test package's Page and we don't have that as a dependency of the CLI right now
     page: any,
@@ -205,6 +208,7 @@ export interface MablToolset {
     environmentId?: string,
     timeoutMillis?: number,
   ): Promise<DatabaseQueryResultSet>;
+  openPdfFile(filePath: string, page: Page): Promise<void>;
 }
 
 export interface MablFixtures {