import sys
import unittest
import weakref

sys.path.insert(0, "..")
import pycparser.c_ast as c_ast
from pycparser.c_parser import Coord


class Test_c_ast(unittest.TestCase):
    def test_BinaryOp(self):
        b1 = c_ast.BinaryOp(
            op="+",
            left=c_ast.Constant(type="int", value="6"),
            right=c_ast.ID(name="joe"),
        )

        self.assertIsInstance(b1.left, c_ast.Constant)
        self.assertEqual(b1.left.type, "int")
        self.assertEqual(b1.left.value, "6")

        self.assertIsInstance(b1.right, c_ast.ID)
        self.assertEqual(b1.right.name, "joe")

    def test_weakref_works_on_nodes(self):
        c1 = c_ast.Constant(type="float", value="3.14")
        wr = weakref.ref(c1)
        cref = wr()
        self.assertIsNotNone(cref)
        assert cref is not None
        self.assertEqual(cref.type, "float")
        self.assertEqual(weakref.getweakrefcount(c1), 1)

    def test_weakref_works_on_coord(self):
        coord = Coord(file="a", line=2)
        wr = weakref.ref(coord)
        cref = wr()
        self.assertIsNotNone(cref)
        assert cref is not None
        self.assertEqual(cref.line, 2)
        self.assertEqual(weakref.getweakrefcount(coord), 1)


class TestNodeVisitor(unittest.TestCase):
    class ConstantVisitor(c_ast.NodeVisitor):
        def __init__(self):
            self.values = []

        def visit_Constant(self, node):
            self.values.append(node.value)

    def test_scalar_children(self):
        b1 = c_ast.BinaryOp(
            op="+",
            left=c_ast.Constant(type="int", value="6"),
            right=c_ast.ID(name="joe"),
        )

        cv = self.ConstantVisitor()
        cv.visit(b1)

        self.assertEqual(cv.values, ["6"])

        b2 = c_ast.BinaryOp(
            op="*", left=c_ast.Constant(type="int", value="111"), right=b1
        )

        b3 = c_ast.BinaryOp(op="^", left=b2, right=b1)

        cv = self.ConstantVisitor()
        cv.visit(b3)

        self.assertEqual(cv.values, ["111", "6", "6"])

    def tests_list_children(self):
        c1 = c_ast.Constant(type="float", value="5.6")
        c2 = c_ast.Constant(type="char", value="t")

        b1 = c_ast.BinaryOp(op="+", left=c1, right=c2)

        b2 = c_ast.BinaryOp(op="-", left=b1, right=c2)

        comp = c_ast.Compound(block_items=[b1, b2, c1, c2])

        cv = self.ConstantVisitor()
        cv.visit(comp)

        self.assertEqual(cv.values, ["5.6", "t", "5.6", "t", "t", "5.6", "t"])

    def test_repr(self):
        c1 = c_ast.Constant(type="float", value="5.6")
        c2 = c_ast.Constant(type="char", value="t")

        b1 = c_ast.BinaryOp(op="+", left=c1, right=c2)

        b2 = c_ast.BinaryOp(op="-", left=b1, right=c2)

        comp = c_ast.Compound(block_items=[b1, b2, c1, c2])

        expected = (
            "Compound(block_items=[BinaryOp(op='+',\n"
            "                               left=Constant(type='float',\n"
            "                                             value='5.6'\n"
            "                                             ),\n"
            "                               right=Constant(type='char',\n"
            "                                              value='t'\n"
            "                                              )\n"
            "                               ),\n"
            "                      BinaryOp(op='-',\n"
            "                               left=BinaryOp(op='+',\n"
            "                                             left=Constant(type='float',\n"
            "                                                           value='5.6'\n"
            "                                                           ),\n"
            "                                             right=Constant(type='char',\n"
            "                                                            value='t'\n"
            "                                                            )\n"
            "                                             ),\n"
            "                               right=Constant(type='char',\n"
            "                                              value='t'\n"
            "                                              )\n"
            "                               ),\n"
            "                      Constant(type='float',\n"
            "                               value='5.6'\n"
            "                               ),\n"
            "                      Constant(type='char',\n"
            "                               value='t'\n"
            "                               )\n"
            "                     ]\n"
            "         )"
        )

        self.assertEqual(repr(comp), expected)


if __name__ == "__main__":
    unittest.main()
