"""
Tests for ErrorBlock
"""


import unittest

from opaque_keys.edx.locator import CourseLocator

from xmodule.error_block import ErrorBlock
from xmodule.modulestore.xml import CourseLocationManager
from xmodule.tests import get_test_system
from xmodule.x_module import STUDENT_VIEW


class SetupTestErrorBlock(unittest.TestCase):
    """Common setUp for use in ErrorBlock tests."""

    def setUp(self):
        super().setUp()
        self.system = get_test_system()
        self.course_id = CourseLocator('org', 'course', 'run')
        self.location = self.course_id.make_usage_key('foo', 'bar')
        self.valid_xml = "<problem>ABC \N{SNOWMAN}</problem>"
        self.error_msg = "Error"


class TestErrorBlock(SetupTestErrorBlock):
    """
    Tests for ErrorBlock
    """

    def test_error_block_xml_rendering(self):
        block = ErrorBlock.from_xml(
            self.valid_xml,
            self.system,
            CourseLocationManager(self.course_id),
            self.error_msg
        )
        assert isinstance(block, ErrorBlock)
        block.runtime = self.system
        context_repr = self.system.render(block, STUDENT_VIEW).content
        assert self.error_msg in context_repr
        assert repr(self.valid_xml) in context_repr