File size: 791 Bytes
a67789e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
const fs = require("fs");
const path = require("path");
const { extractFromHTML } = require("../../data_extraction/extractor");

describe("extractor", () => {
  const sampleFile = path.join(__dirname, "sample.html");

  beforeAll(() => {
    fs.writeFileSync(
      sampleFile,
      `
      <h1>Q: What is service?</h1>
      <p>A: Service is love made visible.</p>
    `
    );
  });

  afterAll(() => {
    fs.unlinkSync(sampleFile);
  });

  test("extracts paragraphs and roles", () => {
    const turns = extractFromHTML(sampleFile);

    expect(turns.length).toBe(2);

    expect(turns[0].role).toBe("user");
    expect(turns[0].content).toContain("What is service");

    expect(turns[1].role).toBe("assistant");
    expect(turns[1].content).toContain("love made visible");
  });
});