import sys from unittest.mock import Mock import pyscript class TestElement: def test_id_is_correct(self): el = pyscript.Element("something") assert el.id == "something" def test_element(self, monkeypatch): el = pyscript.Element("something") document_mock = Mock() call_result = "some_result" document_mock.querySelector = Mock(return_value=call_result) monkeypatch.setattr(pyscript, "document", document_mock) assert not el._element real_element = el.element assert real_element assert pyscript.document.querySelector.call_count == 1 pyscript.document.querySelector.assert_called_with("#something") assert real_element == call_result def test_format_mime_str(): obj = "just a string" out, mime = pyscript.format_mime(obj) assert out == obj assert mime == "text/plain" def test_format_mime_str_escaping(): obj = "
hello
" out, mime = pyscript.format_mime(obj) assert out == "<p>hello</p>" assert mime == "text/plain" def test_format_mime_repr_escaping(): out, mime = pyscript.format_mime(sys) assert out == "<module 'sys' (built-in)>" assert mime == "text/plain" def test_format_mime_HTML(): obj = pyscript.HTML("hello
") out, mime = pyscript.format_mime(obj) assert out == "hello
" assert mime == "text/html"