123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138
'use strict';
const ohm = require('..');const test = require('ava');
const {makeGrammar} = require('./helpers/testUtil');
// --------------------------------------------------------------------// Tests// --------------------------------------------------------------------
test('rule definition source', t => {  const g = makeGrammar(['G {', '  foo = bar', '  bar = "a" | "b" -- baz', '}']);
  function definitionLoc(grammar, ruleName) {    const interval = grammar.rules[ruleName].source;    return [interval.startIdx, interval.endIdx];  }  t.deepEqual(definitionLoc(g, 'foo'), [6, 15], 'regular rule');  t.deepEqual(definitionLoc(g, 'bar'), [18, 40]);  t.deepEqual(definitionLoc(g, 'bar_baz'), [30, 40], 'inline rule');
  const g2 = makeGrammar(['G2 <: G {', '  foo += bar', '  bar := "a" | "b" -- baz', '}'], {    G: g,  });  t.deepEqual(definitionLoc(g2, 'foo'), [12, 22], 'extended rule');  t.deepEqual(definitionLoc(g2, 'bar'), [25, 48], 'overridden rule');  t.deepEqual(definitionLoc(g2, 'bar_baz'), [38, 48], 'inline rule in overridden rule');});
test('rule application source', t => {  const g = makeGrammar([    'G {',    '  foo = bar',    '  beep = letter bar',    '  bar = "a" | "blah" | "a".."z" -- baz',    '}',  ]);
  function fromLoc(pexpr) {    return [pexpr.source.startIdx, pexpr.source.endIdx];  }  const fooBody = g.rules.foo.body;  const beepBody = g.rules.beep.body;  const barBody = g.rules.bar.body;  t.deepEqual(fromLoc(fooBody), [12, 15], 'regular rule application');  t.deepEqual(fromLoc(beepBody.factors[1]), [32, 35]);
  t.deepEqual(fromLoc(beepBody.factors[0]), [25, 31], 'built-in rule application');  t.deepEqual(fromLoc(barBody.terms[0]), [44, 47], 'primitive (string)');  t.deepEqual(fromLoc(barBody.terms[1]), [50, 56], 'another string');
  const barBazBody = g.rules.bar_baz.body;  t.deepEqual(fromLoc(barBazBody), [59, 67], 'primitive (range)');  t.deepEqual(fromLoc(beepBody), [25, 35], 'works for seq');  t.deepEqual(fromLoc(barBody), [44, 74], 'works for alt');});
test('toDisplayString', t => {  const g = ohm.grammar('G { start = "ab" | letter* | "a".."z" }');  const seq = g.rules.start.body;  t.is(seq.toDisplayString(), '"ab" | letter* | "a".."z"');  t.is(seq.terms[0].toDisplayString(), '"ab"');
  const many = seq.terms[1];  t.is(many.toDisplayString(), 'letter*');  t.is(many.expr.toDisplayString(), 'letter');
  t.is(seq.terms[2].toDisplayString(), '"a".."z"');});
test('toString', t => {  const g = makeGrammar('G { start = &"a" ~("b" | #c?) "a".."z"  c = "c" }');  const e = g.rules.start.body;  t.is(e.toString(), '(&"a" ~("b" | #(c)?) "a".."z")');});
test('toArgumentNameList', t => {  const g = makeGrammar([    'G {',    ' Start = &((foo bars foo)+)',    ' foo = "ab" | letter* | "a".."z"',    ' MoreAlt = "a" "b" | "c" "c" ',    ' bars = ~"a" (letter "b"| digit "b") foo? "+"',    ' MoreSeq = &(foo "a" foo) foo "a" foo',    ' plus = foo "+" bars',    ' MoreOpts = ("+" Start)?',    ' ranges = "1".."9" | "!".."@"',    ' Pair<elem> = "(" elem "," elem ")"',    ' }',  ]);
  const iter = g.rules.Start.body;  t.deepEqual(iter.toArgumentNameList(1), ['foo_1s', 'barses', 'foo_2s']);
  const alt = g.rules.foo.body;  t.deepEqual(alt.toArgumentNameList(1), ['_ab_or_letters_or_a_to_z']);  t.deepEqual(alt.terms[0].toArgumentNameList(1), ['_ab']);
  const moreAlt = g.rules.MoreAlt.body;  t.deepEqual(moreAlt.toArgumentNameList(1), ['_a_or__c', '_b_or__c']);
  const many = alt.terms[1];  t.deepEqual(many.toArgumentNameList(1), ['letters']);  t.deepEqual(many.expr.toArgumentNameList(1), ['letter']);
  const range = alt.terms[2];  t.deepEqual(range.toArgumentNameList(1), ['a_to_z']);
  const seq = g.rules.bars.body;  t.deepEqual(seq.toArgumentNameList(1), ['letter_or_digit', '_b', 'optFoo', '$4']);  t.deepEqual(seq.factors[2].toArgumentNameList(1), ['optFoo']);  t.deepEqual(seq.factors[3].toArgumentNameList(1), ['$1']);
  const moreSeq = g.rules.MoreSeq.body;  t.deepEqual(moreSeq.toArgumentNameList(1), [    'foo_1',    '_a_1',    'foo_2',    'foo_3',    '_a_2',    'foo_4',  ]);
  const plus = g.rules.plus.body;  t.deepEqual(plus.toArgumentNameList(1), ['foo', '$2', 'bars']);
  const opts = g.rules.MoreOpts.body;  t.deepEqual(opts.toArgumentNameList(1), ['opt$1', 'optStart']);
  const ranges = g.rules.ranges.body;  t.deepEqual(ranges.terms[0].toArgumentNameList(1), ['_1_to_9']);  t.deepEqual(ranges.terms[1].toArgumentNameList(1), ['$1']);
  const pair = g.rules.Pair.body;  t.deepEqual(pair.toArgumentNameList(1), ['$1', 'param0_1', '$3', 'param0_2', '$5']);});