fix: improve based on code review feedback

Author: MichalKalita

src/errors.ts

@@ -0,0 +1,3 @@
+export class TimeoutError extends Error {
+    override readonly name = 'TimeoutError';
+}

src/mcp/client.ts

@@ -4,13 +4,10 @@ import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/
 
 import log from '@apify/log';
 
+import { TimeoutError } from '../errors.js';
 import { ACTORIZED_MCP_CONNECTION_TIMEOUT_MSEC } from './const.js';
 import { getMCPServerID } from './utils.js';
 
-class TimeoutError extends Error {
-    override readonly name = 'TimeoutError';
-}
-
 /**
  * Creates and connects a ModelContextProtocol client.
  * First tries streamable HTTP transport, then falls back to SSE transport.
@@ -91,7 +88,7 @@ async function createMCPSSEClient(
                     headers.set('authorization', `Bearer ${token}`);
                     return fetch(input, { ...init, headers });
                 },
-            // We have to cast to "any" to use it, since it's non-standard
+                // We have to cast to "any" to use it, since it's non-standard
             } as any, // eslint-disable-line @typescript-eslint/no-explicit-any
         });
 

tests/integration/suite.ts

@@ -398,7 +398,7 @@ export function createIntegrationTestsSuite(
                     limit: 5,
                 },
             });
-            const content = result.content as {text: string}[];
+            const content = result.content as { text: string }[];
             expect(content.some((item) => item.text.includes(ACTOR_PYTHON_EXAMPLE))).toBe(true);
         });
 
@@ -415,7 +415,7 @@ export function createIntegrationTestsSuite(
                     limit: 100,
                 },
             });
-            const content = result.content as {text: string}[];
+            const content = result.content as { text: string }[];
             expect(content.length).toBe(1);
             const outputText = content[0].text;
 
@@ -941,18 +941,8 @@ export function createIntegrationTestsSuite(
             await client.close();
         });
 
-        // Skipping tests, MCP actors are not available in staging environment
-        // TODO: run this tests only for local environment
-        it.skip.for([
-            'mcp-servers/slidespeak-mcp-server',
-            'mcp-servers/brave-search-mcp-server',
-            'jiri.spilka/weather-mcp-server',
-            'apify/actors-mcp-server',
-            'jakub.kopecky/browserbase-mcp-server',
-            'jakub.kopecky/arxiv-mcp-server',
-            'jiri.spilka/playwright-mcp-server',
-        ])('should connect to "%s" MCP server and at least one tool is available', async (mcpServer) => {
-            client = await createClientFn({ tools: [mcpServer] });
+        it('should connect to MCP server and at least one tool is available', async () => {
+            client = await createClientFn({ tools: [ACTOR_MCP_SERVER_ACTOR_NAME] });
             const tools = await client.listTools();
             expect(tools.tools.length).toBeGreaterThan(0);
         });