Fix tests

Signed-off-by: Chris Warrick <kwpolska@gmail.com>
getnikola · Jul 30, 2016 · b6ce43f · b6ce43f
1 parent 51244bd
commit b6ce43f
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/nikola/plugins/compile/rest/__init__.py b/nikola/plugins/compile/rest/__init__.py
@@ -89,8 +89,15 @@ def compile_html(self, source, dest, is_two_file=True):
                 output, error_level, deps = self.compile_html_string(data, source, is_two_file)
                 output = apply_shortcodes(output, self.site.shortcode_registry, self.site, source)
                 out_file.write(output)
-            post = self.site.post_per_input_file[source]
-            post._depfile[dest] += deps.list
+            try:
+                post = self.site.post_per_input_file[source]
+            except KeyError:
+                if deps.list:
+                    self.logger.error(
+                        "Cannot save dependencies for post {0} due to unregistered source file name",
+                        source)
+            else:
+                post._depfile[dest] += deps.list
         if error_level < 3:
             return True
         else:

diff --git a/nikola/plugins/task/galleries.py b/nikola/plugins/task/galleries.py
@@ -416,6 +416,8 @@ def parse_index(self, gallery, input_folder, output_folder):
             #  may break)
             if post.title == 'index':
                 post.title = os.path.split(gallery)[1]
+            # Register the post (via #2417)
+            self.site.post_per_input_file[index_path] = post
         else:
             post = None
         return post

diff --git a/tests/base.py b/tests/base.py
@@ -197,6 +197,7 @@ def __init__(self):
         ]
         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'