# lint-amnesty, pylint: disable=missing-module-docstring

import unittest
import xml.sax.saxutils as saxutils

from lxml import etree

from xmodule.capa import customrender
from xmodule.capa.tests.helpers import mock_capa_system

# just a handy shortcut
lookup_tag = customrender.registry.get_class_for_tag


def extract_context(xml):
    """
    Given an xml element corresponding to the output of test_capa_system.render_template, get back the
    original context
    """
    return eval(xml.text)  # lint-amnesty, pylint: disable=eval-used


def quote_attr(s):
    return saxutils.quoteattr(s)[1:-1]  # don't want the outer quotes


class HelperTest(unittest.TestCase):
    '''
    Make sure that our helper function works!
    '''

    def check(self, d):
        xml = etree.XML(mock_capa_system().render_template('blah', d))
        assert d == extract_context(xml)

    def test_extract_context(self):
        self.check({})
        self.check({1, 2})
        self.check({'id', 'an id'})
        self.check({'with"quote', 'also"quote'})


class SolutionRenderTest(unittest.TestCase):
    '''
    Make sure solutions render properly.
    '''

    def test_rendering(self):
        solution = 'To compute unicorns, count them.'
        xml_str = """<solution id="solution_12">{s}</solution>""".format(s=solution)
        element = etree.fromstring(xml_str)

        renderer = lookup_tag('solution')(mock_capa_system(), element)

        assert renderer.id == 'solution_12'

        # Our test_capa_system "renders" templates to a div with the repr of the context.
        xml = renderer.get_html()
        context = extract_context(xml)
        assert context == {'id': 'solution_12'}


class MathRenderTest(unittest.TestCase):
    '''
    Make sure math renders properly.
    '''

    def check_parse(self, latex_in, mathjax_out):  # lint-amnesty, pylint: disable=missing-function-docstring
        xml_str = """<math>{tex}</math>""".format(tex=latex_in)
        element = etree.fromstring(xml_str)

        renderer = lookup_tag('math')(mock_capa_system(), element)

        assert renderer.mathstr == mathjax_out

    def test_parsing(self):
        self.check_parse('$abc$', '[mathjaxinline]abc[/mathjaxinline]')
        self.check_parse('$abc', '$abc')
        self.check_parse(r'$\displaystyle 2+2$', '[mathjax] 2+2[/mathjax]')

    # NOTE: not testing get_html yet because I don't understand why it's doing what it's doing.