import { describe, it, expect } from 'vitest';
import { highlightTs } from './highlight-ts';

describe('highlightTs', () => {
  it('wraps keywords, strings and comments in the styling spans', () => {
    const out = highlightTs(`const x = 'hi'; // note`);
    expect(out).toContain('<span class="k">const</span>');
    expect(out).toContain(`<span class="s">'hi'</span>`);
    expect(out).toContain('<span class="c">// note</span>');
  });

  it('escapes HTML metacharacters so the [innerHTML] sink is safe', () => {
    const out = highlightTs(`type T = A<B> & C;`);
    expect(out).toContain('&lt;B&gt;');
    expect(out).toContain('&amp;');
    expect(out).not.toContain('<B>');
  });

  it('treats the whole // tail as one comment (keywords after // are not re-highlighted)', () => {
    const out = highlightTs(`x(); // return here`);
    expect(out).toContain('<span class="c">// return here</span>');
    expect(out).not.toContain('<span class="k">return</span>');
  });
});