import pytest

try:
    import rply
except ImportError:
    pytest.skip("rply not installed")

from rpython.jit.metainterp.ruleopt.parse import *

import os

with open(
    os.path.join(
        os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "real.rules"
    )
) as f:
    ALLRULES = f.read()


def test_parse_int_add_zero():
    s = """\
add_zero: int_add(x, 0)
    => x
"""
    ast = parse(s)
    assert str(ast.rules[0]) == 'add_zero: int_add(x, 0)\n    => x'

def test_parse_source_positions():
    s = """\
add_zero: int_add(x, 0)
    => x
"""
    ast = parse(s)
    rule = ast.rules[0]
    assert rule.sourcepos.lineno == 1
    assert rule.target.sourcepos.lineno == 2
    assert rule.target.sourcepos.colno == 8

def test_parse_function_many_args():
    s = """\
n: op(C)
    C1 = f(C, C, C, C)
    => C
    """
    ast = parse(s)


def test_sorry():
    s = """\
eq_different_knownbits: int_eq(x, y)
    SORRY_Z3
    => 0
    """
    ast = parse(s)
    assert ast.rules[0].cantproof == True

def test_parse_lshift_rshift():
    s = """\
int_lshift_int_rshift_consts: int_lshift(int_rshift(x, C1), C1)
    C = (-1 >> C1) << C1
    => int_and(x, C)
    """
    ast = parse(s)


def test_parse_all():
    ast = parse(ALLRULES)  # also typechecks


def test_undefined_name():
    s = """\
n: op(C)
    => x
    """
    with pytest.raises(TypeCheckError) as info:
        ast = parse(s)
    assert str(info.value) == "variable 'x' is not defined"
    assert info.value.format(s) == '''\
Type error: variable 'x' is not defined
in line 2
        => x
           ^'''

def test_doubly_defined_name():
    s = """\
n: op(C)
    C = C + 1
    => C
    """
    with pytest.raises(TypeCheckError) as info:
        ast = parse(s)
    assert str(info.value) == "'C' is already defined"


def test_check_not_bool():
    s = """\
n: op(C)
    check C
    => C
    """
    with pytest.raises(TypeCheckError) as info:
        ast = parse(s)
    assert str(info.value) == "expected check expression to return a bool, got int"
    assert info.value.format(s) == '''\
Type error: expected check expression to return a bool, got int
in line 2
        check C
              ^'''