import { LinePartitionChunker } from "../../code-search/chunk/LinePartitionChunker";
import { ChunkWithoutID } from "../../code-search/chunk/_base/Chunk";
import { TextRange } from "../../code-search/scope-graph/model/TextRange";

describe('LinePartitionChunker', () => {
  let chunker: LinePartitionChunker;

  beforeEach(() => {
    chunker = new LinePartitionChunker();
  });

  describe('chunk', () => {
    it('should return 2 chunks when return 3 lines', async () => {
      const filepath = 'test.txt';
      const contents = 'This is a test content for the LinePartitionChunker class.\nLine 2\nLine3';
      const maxChunkSize = 2;

      const chunks: ChunkWithoutID[] = [];
      let chunkWithoutIDS = chunker.chunk(filepath, contents, maxChunkSize);
      for await (const chunk of chunkWithoutIDS) {
        chunks.push(chunk);
      }

      expect(chunks.length).equal(2);
    });
  });
});