Fix plugins tests. (#83)

Author: vb64

tests/test/test_plugins/__init__.py

@@ -1,6 +1,22 @@
 """Root class for testing plugins."""
+from plantuml import PlantUMLConnectionError
 from .. import TestBase
 
 
+class MockPlantUML:
+    """Mocked PlantUML class."""
+
+    def __init__(self, url=''):
+        """Make new instance."""
+        self.url = url
+
+    def processes(self, _text):
+        """Emulate call."""
+        if not self.url.startswith('http://'):
+            raise PlantUMLConnectionError("Only absolute URIs are allowed.")
+
+        return open("fixture/plantuml.png", "rb").read()
+
+
 class TestPlugin(TestBase):
     """Base class for tests plugins."""

tests/test/test_plugins/test_init.py

@@ -2,7 +2,7 @@
 
 make test T=test_plugins/test_init.py
 """
-from . import TestPlugin
+from . import TestPlugin, MockPlantUML
 
 
 class TestInit(TestPlugin):
@@ -12,6 +12,7 @@ def test_md(self):
         """Process md with plantuml content."""
         from markdown_pdf import Section, MarkdownPdf
         from markdown_pdf.pligins import Plugin
+        from markdown_pdf.pligins import plantuml
 
         pdf = MarkdownPdf()
         assert not pdf.plugins
@@ -20,6 +21,9 @@ def test_md(self):
         assert "@startuml" in html
         pdf.save(self.build("plantuml.pdf"))
 
+        saved = plantuml.PlantUML
+        plantuml.PlantUML = MockPlantUML
+
         plugins = {
           Plugin.Plantuml: {'url': 'http://www.plantuml.com/plantuml/img/'}
         }
@@ -29,6 +33,8 @@ def test_md(self):
         html = pdf.add_section(Section(text))
         assert "@startuml" not in html
 
+        plantuml.PlantUML = saved
+
     def test_get_plugin_chunks(self):
         """Check get_plugin_chunks function."""
         from markdown_pdf.pligins import Plugin, get_plugin_chunks

tests/test/test_plugins/test_plantuml.py

@@ -4,7 +4,7 @@
 """
 import pytest
 from plantuml import PlantUML, PlantUMLConnectionError
-from . import TestPlugin
+from . import TestPlugin, MockPlantUML
 
 UML_CODE = """
 @startuml
@@ -14,21 +14,6 @@
 """
 
 
-class MockPlantUML:
-    """Mocked PlantUML class."""
-
-    def __init__(self, url=''):
-        """Make new instance."""
-        self.url = url
-
-    def processes(self, _text):
-        """Emulate call."""
-        if not self.url.startswith('http://'):
-            raise PlantUMLConnectionError("Only absolute URIs are allowed.")
-
-        return open("fixture/plantuml.png", "rb").read()
-
-
 class TesPlantuml(TestPlugin):
     """Plantuml content."""