import { expect } from "chai";
import { documentContext, inline } from "./testUtils";
import { calculateReplaceRangeByBracketStack } from "./calculateReplaceRangeByBracketStack";

describe("postprocess", () => {
  describe("calculateReplaceRangeByBracketStack", () => {
    it("should handle auto closing quotes", () => {
      const context = {
        ...documentContext`
        const hello = "║"
        `,
        language: "typescript",
      };
      const response = {
        id: "",
        choices: [
          {
            index: 0,
            text: inline`
                       ├hello";┤
            `,
            replaceRange: {
              start: context.position,
              end: context.position,
            },
          },
        ],
      };
      const expected = {
        id: "",
        choices: [
          {
            index: 0,
            text: inline`
                       ├hello";┤
            `,
            replaceRange: {
              start: context.position,
              end: context.position + 1,
            },
          },
        ],
      };
      expect(calculateReplaceRangeByBracketStack(response, context)).to.deep.equal(expected);
    });

    it("should handle auto closing quotes", () => {
      const context = {
        ...documentContext`
        let htmlMarkup = \`║\`
        `,
        language: "typescript",
      };
      const response = {
        id: "",
        choices: [
          {
            index: 0,
            text: inline`
                           ├<h1>\${message}</h1>\`;┤
            `,
            replaceRange: {
              start: context.position,
              end: context.position,
            },
          },
        ],
      };
      const expected = {
        id: "",
        choices: [
          {
            index: 0,
            text: inline`
                           ├<h1>\${message}</h1>\`;┤
            `,
            replaceRange: {
              start: context.position,
              end: context.position + 1,
            },
          },
        ],
      };
      expect(calculateReplaceRangeByBracketStack(response, context)).to.deep.equal(expected);
    });

    it("should handle multiple auto closing brackets", () => {
      const context = {
        ...documentContext`
        process.on('data', (data) => {║})
        `,
        language: "typescript",
      };
      const response = {
        id: "",
        choices: [
          {
            index: 0,
            text: inline`
                                      ├
              console.log(data);
            });┤
            `,
            replaceRange: {
              start: context.position,
              end: context.position,
            },
          },
        ],
      };
      const expected = {
        id: "",
        choices: [
          {
            index: 0,
            text: inline`
                                      ├
              console.log(data);
            });┤
            `,
            replaceRange: {
              start: context.position,
              end: context.position + 2,
            },
          },
        ],
      };
      expect(calculateReplaceRangeByBracketStack(response, context)).to.deep.equal(expected);
    });

    it("should handle multiple auto closing brackets", () => {
      const context = {
        ...documentContext`
        let mat: number[][][] = [[[║]]]
        `,
        language: "typescript",
      };
      const response = {
        id: "",
        choices: [
          {
            index: 0,
            text: inline`
                                   ├1, 2], [3, 4]], [[5, 6], [7, 8]]];┤
            `,
            replaceRange: {
              start: context.position,
              end: context.position,
            },
          },
        ],
      };
      const expected = {
        id: "",
        choices: [
          {
            index: 0,
            text: inline`
                                   ├1, 2], [3, 4]], [[5, 6], [7, 8]]];┤
            `,
            replaceRange: {
              start: context.position,
              end: context.position + 3,
            },
          },
        ],
      };
      expect(calculateReplaceRangeByBracketStack(response, context)).to.deep.equal(expected);
    });
  });

  describe("calculateReplaceRangeByBracketStack: bad cases", () => {
    it("cannot handle the case of completion bracket stack is same with suffix but should not be replaced", () => {
      const context = {
        ...documentContext`
        function clamp(n: number, max: number, min: number): number {
          return Math.max(Math.min(║);
        }
        `,
        language: "typescript",
      };
      const response = {
        id: "",
        choices: [
          {
            index: 0,
            text: inline`
                                   ├n, max), min┤
            `,
            replaceRange: {
              start: context.position,
              end: context.position,
            },
          },
        ],
      };
      const expected = {
        id: "",
        choices: [
          {
            index: 0,
            text: inline`
                                   ├n, max), min┤
            `,
            replaceRange: {
              start: context.position,
              end: context.position,
            },
          },
        ],
      };
      expect(calculateReplaceRangeByBracketStack(response, context)).not.to.deep.equal(expected);
    });
  });
});