openclaw/test/scripts/check-raw-socket-callsite-classification.test.ts

import { describe, expect, it } from "vitest";
import { findRawSocketClientCallLines } from "../../scripts/check-raw-socket-callsite-classification.mjs";

describe("check-raw-socket-callsite-classification", () => {
  it("finds raw net, tls, and http2 client calls", () => {
    const source = `
      import net from "node:net";
      import * as tls from "node:tls";
      import http2 from "node:http2";
      net.connect({ host: "example.com", port: 6667 });
      tls.connect({ host: "example.com", port: 6697 });
      http2.connect("https://api.example.com");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([5, 6, 7]);
  });

  it("ignores comments, strings, and unrelated connect methods", () => {
    const source = `
      // net.connect({ host: "example.com" });
      const text = "tls.connect({ host: 'example.com' })";
      client.connect(transport);
      websocket.connect();
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([]);
  });

  it("handles aliased imports, requires, and dynamic literal imports", () => {
    const source = `
      import * as rawNet from "node:net";
      const rawTls = require("node:tls");
      const rawHttp2 = await import("node:http2");
      rawNet.connect({ host: "127.0.0.1", port: 1 });
      rawTls.connect({ host: "127.0.0.1", port: 1 });
      rawHttp2.connect("https://example.com");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([5, 6, 7]);
  });

  it("finds element-access raw socket calls", () => {
    const source = `
      import net from "node:net";
      import tls from "node:tls";
      import http2 from "node:http2";
      net["connect"]({ host: "127.0.0.1", port: 1 });
      tls["createConnection"]({ host: "127.0.0.1", port: 1 });
      http2["connect"]("https://example.com");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([5, 6, 7]);
  });

  it("finds destructured dynamic-import default raw module aliases", () => {
    const source = `
      const { default: rawNet } = await import("node:net");
      const { default: rawTls } = await import("node:tls");
      const { default: rawHttp2 } = await import("node:http2");
      rawNet.connect({ host: "127.0.0.1", port: 1 });
      rawTls.connect({ host: "127.0.0.1", port: 1 });
      rawHttp2.connect("https://example.com");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([5, 6, 7]);
  });

  it("finds direct raw module receiver calls", () => {
    const source = `
      require("node:net").connect({ host: "127.0.0.1", port: 1 });
      require("node:tls").createConnection({ host: "127.0.0.1", port: 1 });
      (await import("node:http2")).connect("https://example.com");
      (await import("node:net")).default.connect({ host: "127.0.0.1", port: 1 });
      (await import("node:tls")).default.createConnection({ host: "127.0.0.1", port: 1 });
      (await import("node:http2")).default.connect("https://example.com");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([2, 3, 4, 5, 6, 7]);
  });

  it("finds named default raw module imports", () => {
    const source = `
      import { default as rawNet } from "node:net";
      import { default as rawTls } from "node:tls";
      import { default as rawHttp2 } from "node:http2";
      rawNet.connect({ host: "127.0.0.1", port: 1 });
      rawTls.connect({ host: "127.0.0.1", port: 1 });
      rawHttp2.connect("https://example.com");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([5, 6, 7]);
  });

  it("finds raw socket module object aliases", () => {
    const source = `
      import net from "node:net";
      import tls from "node:tls";
      import http2 from "node:http2";
      const rawNet = net;
      const rawTls = tls;
      const rawHttp2 = http2;
      rawNet.connect({ host: "127.0.0.1", port: 1 });
      rawTls.connect({ host: "127.0.0.1", port: 1 });
      rawHttp2.connect("https://example.com");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([8, 9, 10]);
  });

  it("finds aliases to raw socket module members", () => {
    const source = `
      import net from "node:net";
      import tls from "node:tls";
      import http2 from "node:http2";
      const netConnect = net.connect;
      const tlsConnect = tls.connect;
      const h2Connect = http2.connect;
      const Socket = net.Socket;
      const { createConnection } = net;
      netConnect({ host: "127.0.0.1", port: 1 });
      tlsConnect({ host: "127.0.0.1", port: 1 });
      h2Connect("https://example.com");
      createConnection({ host: "127.0.0.1", port: 1 });
      new Socket().connect("/tmp/socket");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([10, 11, 12, 13, 14]);
  });

  it("finds destructured require and dynamic import raw socket bindings", () => {
    const source = `
      const { connect, createConnection, Socket } = require("node:net");
      const { connect: tlsConnect } = await import("node:tls");
      connect({ host: "127.0.0.1", port: 1 });
      createConnection({ host: "127.0.0.1", port: 1 });
      tlsConnect({ host: "127.0.0.1", port: 1 });
      new Socket().connect("/tmp/socket");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([4, 5, 6, 7]);
  });

  it("finds stored Socket instance connect calls", () => {
    const source = `
      import net from "node:net";
      const client = new net.Socket();
      client.connect("/tmp/socket");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([4]);
  });

  it("finds named raw socket imports", () => {
    const source = `
      import { connect as netConnect, createConnection, Socket } from "node:net";
      import { connect as tlsConnect } from "node:tls";
      import { connect as http2Connect } from "node:http2";
      netConnect({ host: "127.0.0.1", port: 1 });
      createConnection({ host: "127.0.0.1", port: 1 });
      tlsConnect({ host: "127.0.0.1", port: 1 });
      http2Connect("https://example.com");
      new Socket().connect("/tmp/socket");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([5, 6, 7, 8, 9]);
  });

  it("finds createConnection and constructed Socket.connect client calls", () => {
    const source = `
      import net from "node:net";
      import tls from "node:tls";
      net.createConnection({ host: "127.0.0.1", port: 1 });
      tls.createConnection({ host: "127.0.0.1", port: 1 });
      new net.Socket().connect("/tmp/socket");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([4, 5, 6]);
  });

  it("handles parenthesized and asserted module identifiers", () => {
    const source = `
      import net from "node:net";
      import tls from "node:tls";
      import http2 from "node:http2";
      (net as typeof import("node:net")).connect({ host: "127.0.0.1", port: 1 });
      (tls as typeof import("node:tls")).connect({ host: "127.0.0.1", port: 1 });
      (http2 as typeof import("node:http2")).connect("https://example.com");
    `;

    expect(findRawSocketClientCallLines(source)).toEqual([5, 6, 7]);
  });
});