import { describe, expect, test } from "bun:test";

import { checkBanner, checkConnected } from "../../../../../src/server/checker/runner/tcp/expect";

describe("checkConnected", () => {
  test("connected=true 期望 true 匹配", () => {
    const result = checkConnected(true, true);
    expect(result.matched).toBe(true);
    expect(result.failure).toBeNull();
  });

  test("connected=false 期望 false 匹配", () => {
    const result = checkConnected(false, false);
    expect(result.matched).toBe(true);
    expect(result.failure).toBeNull();
  });

  test("connected=false 期望 true 不匹配", () => {
    const result = checkConnected(false, true);
    expect(result.matched).toBe(false);
    expect(result.failure!.kind).toBe("mismatch");
    expect(result.failure!.phase).toBe("connected");
  });

  test("connected=true 期望 false 不匹配", () => {
    const result = checkConnected(true, false);
    expect(result.matched).toBe(false);
    expect(result.failure!.kind).toBe("mismatch");
    expect(result.failure!.phase).toBe("connected");
  });
});

describe("checkBanner", () => {
  test("contains 匹配", () => {
    const result = checkBanner("220 smtp.example.com ESMTP", [{ contains: "ESMTP" }]);
    expect(result.matched).toBe(true);
  });

  test("contains 不匹配", () => {
    const result = checkBanner("220 smtp.example.com ESMTP", [{ contains: "POSTFIX" }]);
    expect(result.matched).toBe(false);
    expect(result.failure!.kind).toBe("mismatch");
    expect(result.failure!.phase).toBe("banner");
  });

  test("regex 正则匹配", () => {
    const result = checkBanner("220 smtp.example.com ESMTP", [{ regex: "^220" }]);
    expect(result.matched).toBe(true);
  });

  test("空 banner 与 contains 空字符串", () => {
    const result = checkBanner("", [{ contains: "" }]);
    expect(result.matched).toBe(true);
  });

  test("多 operator 同时匹配", () => {
    const result = checkBanner("220 ESMTP", [{ contains: "ESMTP", regex: "^220" }]);
    expect(result.matched).toBe(true);
  });

  test("多 operator 部分不匹配", () => {
    const result = checkBanner("220 ESMTP", [{ contains: "ESMTP", regex: "^250" }]);
    expect(result.matched).toBe(false);
  });
});