fix more tests

getnikola · Jul 17, 2015 · 72083ce · 72083ce
1 parent ac2fb93
commit 72083ce
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/tests/test_command_import_wordpress.py b/tests/test_command_import_wordpress.py
@@ -2,8 +2,6 @@
 from __future__ import unicode_literals, absolute_import
 
 import os
-import sys
-
 
 import unittest
 import mock

diff --git a/tests/test_rss_feeds.py b/tests/test_rss_feeds.py
@@ -91,9 +91,12 @@ def setUp(self):
                     # lxml will complain if the encoding is specified in the
                     # xml when running with unicode strings.
                     # We do not include this in our content.
-                    open_handle = opener_mock()
-                    file_content = [call[1][0]
-                                    for call in open_handle.mock_calls[1:-1]][0]
+                    try:
+                        file_content = [
+                            call[1][0]
+                            for call in opener_mock.mock_calls[2:-1]][0]
+                    except IndexError:
+                        import pdb; pdb.set_trace()
                     splitted_content = file_content.split('\n')
                     self.encoding_declaration = splitted_content[0]
                     content_without_encoding_declaration = splitted_content[1:]