from __future__ import absolute_import

import sys
from unittest import TestCase, main

from lark import Lark, Token
from lark.load_grammar import GrammarLoader, GrammarError


class TestGrammar(TestCase):
    def setUp(self):
        pass

    def test_errors(self):
        for msg, examples in GrammarLoader.ERRORS:
            for example in examples:
                try:
                    p = Lark(example)
                except GrammarError as e:
                    assert msg in str(e)
                else:
                    assert False, "example did not raise an error"

    def test_override_rule(self):
        # Overrides the 'sep' template in existing grammar to add an optional terminating delimiter
        # Thus extending it beyond its original capacity
        p = Lark("""
            %import .test_templates_import (start, sep)

            %override sep{item, delim}: item (delim item)* delim?
            %ignore " "
        """, source_path=__file__)

        a = p.parse('[1, 2, 3]')
        b = p.parse('[1, 2, 3, ]')
        assert a == b

    def test_override_terminal(self):
        p = Lark("""
            
            %import .grammars.ab (startab, A, B)
            
            %override A: "C"
            %override B: "D"
        """, start='startab', source_path=__file__)

        a = p.parse('CD')
        self.assertEqual(a.children[0].children, [Token('A', 'C'), Token('B', 'D')])



if __name__ == '__main__':
    main()