New upstream version 7.8.1upstream/7.8.1

1 files changed, 18 insertions, 45 deletions

diff --git a/tests/base.py b/tests/base.py
index 14af18a..2322c15 100644
--- a/tests/base.py
+++ b/tests/base.py
@@ -20,6 +20,7 @@ import unittest
 import logbook
 
 import nikola.utils
+import nikola.shortcodes
 nikola.utils.LOGGER.handlers.append(logbook.TestHandler())
 
 from yapsy.PluginManager import PluginManager
@@ -35,51 +36,7 @@ from nikola.plugin_categories import (
     RestExtension
 )
 
-
-if sys.version_info < (2, 7):
-
-    try:
-        import unittest2
-        _unittest2 = True
-    except ImportError:
-        _unittest2 = False
-
-    if _unittest2:
-        BaseTestCase = unittest2.TestCase
-
-    else:
-
-        class BaseTestCase(unittest.TestCase):
-            """ Base class for providing 2.6 compatibility """
-
-            def assertIs(self, first, second, msg=None):
-                self.assertTrue(first is second)
-
-            def assertIsNot(self, first, second, msg=None):
-                self.assertTrue(first is not second)
-
-            def assertIsNone(self, expr, msg=None):
-                self.assertTrue(expr is None)
-
-            def assertIsNotNone(self, expr, msg=None):
-                self.assertTrue(expr is not None)
-
-            def assertIn(self, first, second, msg=None):
-                self.assertTrue(first in second)
-
-            def assertNotIn(self, first, second, msg=None):
-                self.assertTrue(first not in second)
-
-            def assertIsInstance(self, obj, cls, msg=None):
-                self.assertTrue(isinstance(obj, cls))
-
-            def assertNotIsInstance(self, obj, cls, msg=None):
-                self.assertFalse(isinstance(obj, cls))
-
-
-else:
-    BaseTestCase = unittest.TestCase
-
+BaseTestCase = unittest.TestCase
 
 @contextmanager
 def cd(path):
@@ -183,6 +140,8 @@ class FakePost(object):
         self._title = title
         self._slug = slug
         self._meta = {'slug': slug}
+        self.default_lang = 'en'
+        self._depfile = {}
 
     def title(self):
         return self._title
@@ -219,6 +178,7 @@ class FakeSite(object):
             "RestExtension": RestExtension
         })
         self.loghandlers = nikola.utils.STDERR_HANDLER  # TODO remove on v8
+        self.shortcode_registry = {}
         self.plugin_manager.setPluginInfoExtension('plugin')
         if sys.version_info[0] == 3:
             places = [
@@ -238,6 +198,7 @@ class FakeSite(object):
         ]
         self.debug = True
         self.rst_transforms = []
+        self.post_per_input_file = {}
         # This is to make plugin initialization happy
         self.template_system = self
         self.name = 'mako'
@@ -257,3 +218,15 @@ class FakeSite(object):
 
     def render_template(self, name, _, context):
         return('<img src="IMG.jpg">')
+
+    # this code duplicated in nikola/nikola.py
+    def register_shortcode(self, name, f):
+        """Register function f to handle shortcode "name"."""
+        if name in self.shortcode_registry:
+            nikola.utils.LOGGER.warn('Shortcode name conflict: %s', name)
+            return
+        self.shortcode_registry[name] = f
+
+    def apply_shortcodes(self, data, *a, **kw):
+        """Apply shortcodes from the registry on data."""
+        return nikola.shortcodes.apply_shortcodes(data, self.shortcode_registry, **kw)