zan-proxy/test/ProxyServer/handler/connect.spec.ts

前端代理

import chai from "chai";
import http from "http";
import LRU from "lru-cache";
import "mocha";
import net from "net";
import { ConnectHandler } from "./../../../src/ProxyServer/impl";

const should = chai.should();

describe("ConnectHandler", () => {
    const cache = LRU();
    const connectHandler = new ConnectHandler(2019, cache);
    const s = http.createServer();
    const ts = net.createServer();
    beforeEach(() => {
        s.listen(2018);
        ts.listen(2019);
    });
    it("should connect to the proxy port if request port is 443", (done) => {
        s.on("connect", connectHandler.handle.bind(connectHandler));
        const req = http.request({
            hostname: "127.0.0.1",
            method: "CONNECT",
            path: "www.example.com:443",
            port: 2018,
        });
        req.end();
        req.on("connect", () => {
            req.destroy(null);
            done();
        });
    });
    afterEach(() => {
        s.close();
        ts.close();
    });
});