HtmlDataExtractor.test.ts

import { HtmlDataExtractor } from '@shared/visualServer/HtmlDataExtractor';

describe('HtmlDataExtractor', () => {
  let target: HtmlDataExtractor;

  beforeEach(() => {
    target = new HtmlDataExtractor();
  });

  describe('extractArray', () => {
    it('should extract a simple array', () => {
      const html = 'let TEST_DATA = [1, 2, 3];';
      const result = target.extractArray(html, 'TEST_DATA');
      expect(result).toBe('[1, 2, 3]');
    });

    it('should extract nested arrays', () => {
      const html = 'let TEST_DATA = [[1], [2, 3]];';
      const result = target.extractArray(html, 'TEST_DATA');
      expect(result).toBe('[[1], [2, 3]]');
    });

    it('should handle whitespace after equals sign', () => {
      const html = 'let TEST_DATA =    [true];';
      const result = target.extractArray(html, 'TEST_DATA');
      expect(result).toBe('[true]');
    });

    it('should return null if variable not found', () => {
      const html = 'let OTHER_DATA = [];';
      const result = target.extractArray(html, 'TEST_DATA');
      expect(result).toBeNull();
    });

    it('should return null if does not start with bracket', () => {
      const html = 'let TEST_DATA = "string";';
      const result = target.extractArray(html, 'TEST_DATA');
      expect(result).toBeNull();
    });

    it('should return null if brackets are unbalanced', () => {
      const html = 'let TEST_DATA = [1, 2';
      const result = target.extractArray(html, 'TEST_DATA');
      expect(result).toBeNull();
    });

    it('should return null if brackets are unbalanced (extra closing)', () => {
      const html = 'let TEST_DATA = [1, 2';
      expect(target.extractArray(html, 'TEST_DATA')).toBeNull();
    });

    it('should handle array with objects', () => {
      const html = 'let TEST_DATA = [{a: 1}, {b: [2]}];';
      const result = target.extractArray(html, 'TEST_DATA');
      expect(result).toBe('[{a: 1}, {b: [2]}]');
    });
  });

  describe('extractGlobalDefinitions', () => {
    it('should extract definition with let', () => {
      const html = 'let GLOBAL_DEFINITIONS = { "key": "value" };';
      const result = target.extractGlobalDefinitions(html);
      expect(result).toBe('{ "key": "value" }');
    });

    it('should extract definition with const', () => {
      const html = 'const GLOBAL_DEFINITIONS = { "key": "data" };';
      const result = target.extractGlobalDefinitions(html);
      expect(result).toBe('{ "key": "data" }');
    });

    it('should return null if not found', () => {
      const html = 'let OTHER = {};';
      const result = target.extractGlobalDefinitions(html);
      expect(result).toBeNull();
    });

    it('should return null if regex match has no group', () => {
      const html = 'let GLOBAL_DEFINITIONS = ;';
      expect(target.extractGlobalDefinitions(html)).toBeNull();
    });
  });
});