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

import type { ContentExpectations } from "../../../../../src/server/checker/expect/types";

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

function value(matcher: ContentExpectations[number]["matcher"]): ContentExpectations[number] {
  return { kind: "value", matcher };
}

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", [value({ contains: "ESMTP" })]);
    expect(result.matched).toBe(true);
  });

  test("contains 不匹配", () => {
    const result = checkBanner("220 smtp.example.com ESMTP", [value({ 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", [value({ regex: "^220" })]);
    expect(result.matched).toBe(true);
  });

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

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

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