import { parser } from '@grafana/lezer-logql';

import { getLeftMostChild, getString, replaceVariables } from './parsingUtils';

describe('getLeftMostChild', () => {
  it('return left most child', () => {
    const tree = parser.parse('count_over_time({bar="baz"}[5m])');
    const child = getLeftMostChild(tree.topNode);
    expect(child).toBeDefined();
    expect(child!.name).toBe('CountOverTime');
  });
});

describe('replaceVariables', () => {
  it('should replace variables', () => {
    expect(replaceVariables('rate([{bar="${app}", baz="[[label_var]]"}[$__auto])')).toBe(
      'rate([{bar="__V_2__app__V__", baz="__V_1__label_var__V__"}[__V_0____auto__V__])'
    );
  });
});

describe('getString', () => {
  it('should return correct string representation of the node', () => {
    const expr = 'count_over_time({bar="baz"}[5m])';
    const tree = parser.parse(expr);
    const child = getLeftMostChild(tree.topNode);
    expect(getString(expr, child)).toBe('count_over_time');
  });

  it('should return string with correct variables', () => {
    const expr = 'count_over_time({bar="__V_2__app__V__"}[__V_0____auto__V__])';
    const tree = parser.parse(expr);
    expect(getString(expr, tree.topNode)).toBe('count_over_time({bar="${app}"}[$__auto])');
  });

  it('is symmetrical with replaceVariables', () => {
    const expr = 'count_over_time({bar="${app}", baz="[[label_var]]"}[$__auto])';
    const replaced = replaceVariables(expr);
    const tree = parser.parse(replaced);
    expect(getString(replaced, tree.topNode)).toBe(expr);
  });
});