Added files writer tests

IPython/nbconvert/writers/tests/test_files.py

@@ -0,0 +1,159 b''
+"""
+Module with tests for files
+"""
+
+#-----------------------------------------------------------------------------
+# Copyright (c) 2013, the IPython Development Team.
+#
+# Distributed under the terms of the Modified BSD License.
+#
+# The full license is in the file COPYING.txt, distributed with this software.
+#-----------------------------------------------------------------------------
+
+#-----------------------------------------------------------------------------
+# Imports
+#-----------------------------------------------------------------------------
+
+import sys
+import os
+from StringIO import StringIO
+
+from ...tests.base import TestsBase
+from ..files import FilesWriter
+
+
+#-----------------------------------------------------------------------------
+# Class
+#-----------------------------------------------------------------------------
+
+class Testfiles(TestsBase):
+    """Contains test functions for files.py"""
+
+    def test_basic_output(self):
+        """Is FilesWriter basic output correct?"""
+
+        # Work in a temporary directory.
+        with self.create_temp_cwd():
+
+            # Create the resoruces dictionary
+            res = {}
+
+            # Create files writer, test output
+            writer = FilesWriter()
+            writer.write(u'y', res, notebook_name="z")
+
+            # Check the output of the file
+            with open('z', 'r') as f:
+                output = f.read()
+                self.assertEqual(output, u'y')
+
+    def test_ext(self):
+        """Does the FilesWriter add the correct extension to the output?"""
+
+        # Work in a temporary directory.
+        with self.create_temp_cwd():
+
+            # Create the resoruces dictionary
+            res = {'output_extension': 'txt'}
+
+            # Create files writer, test output
+            writer = FilesWriter()
+            writer.write(u'y', res, notebook_name="z")
+
+            # Check the output of the file
+            assert os.path.isfile('z.txt')
+            with open('z.txt', 'r') as f:
+                output = f.read()
+                self.assertEqual(output, u'y')
+
+
+    def test_extract(self):
+        """Can FilesWriter write extracted figures correctly?"""
+
+        # Work in a temporary directory.
+        with self.create_temp_cwd():
+
+            # Create the resoruces dictionary
+            res = {'outputs': {os.path.join('z_files', 'a'): b'b'}}
+
+            # Create files writer, test output
+            writer = FilesWriter()
+            writer.write(u'y', res, notebook_name="z")
+
+            # Check the output of the file
+            with open('z', 'r') as f:
+                output = f.read()
+                self.assertEqual(output, u'y')
+
+            # Check the output of the extracted file
+            extracted_file_dest = os.path.join('z_files', 'a')
+            assert os.path.isfile(extracted_file_dest)
+            with open(extracted_file_dest, 'r') as f:
+                output = f.read()
+                self.assertEqual(output, 'b')
+
+
+    def test_builddir(self):
+        """Can FilesWriter write to a build dir correctly?"""
+
+        # Work in a temporary directory.
+        with self.create_temp_cwd():
+
+            # Create the resoruces dictionary
+            res = {'outputs': {os.path.join('z_files', 'a'): b'b'}}
+
+            # Create files writer, test output
+            writer = FilesWriter()
+            writer.build_directory = u'build'
+            writer.write(u'y', res, notebook_name="z")
+
+            # Check the output of the file
+            assert os.path.isdir(writer.build_directory)
+            dest = os.path.join(writer.build_directory, 'z')
+            with open(dest, 'r') as f:
+                output = f.read()
+                self.assertEqual(output, u'y')
+
+            # Check the output of the extracted file
+            extracted_file_dest = os.path.join(writer.build_directory, 'z_files', 'a')
+            assert os.path.isfile(extracted_file_dest)
+            with open(extracted_file_dest, 'r') as f:
+                output = f.read()
+                self.assertEqual(output, 'b')
+
+
+    def test_links(self):
+        """Can the FilesWriter handle linked files correctly?"""
+
+        # Work in a temporary directory.
+        with self.create_temp_cwd():
+
+            # Create test file
+            os.mkdir('sub')
+            with open(os.path.join('sub', 'c'), 'w') as f:
+                f.write('d')
+
+            # Create the resoruces dictionary
+            res = {}
+
+            # Create files writer, test output
+            writer = FilesWriter()
+            writer.files = [os.path.join('sub', 'c')]
+            writer.build_directory = u'build'
+            writer.write(u'y', res, notebook_name="z")
+
+            # Check the output of the file
+            assert os.path.isdir(writer.build_directory)
+            dest = os.path.join(writer.build_directory, 'z')
+            with open(dest, 'r') as f:
+                output = f.read()
+                self.assertEqual(output, u'y')
+
+            # Check to make sure the linked file was copied
+            path = os.path.join(writer.build_directory, 'sub')
+            assert os.path.isdir(path)
+            dest = os.path.join(path, 'c')
+            assert os.path.isfile(dest)
+            with open(dest, 'r') as f:
+                output = f.read()
+                self.assertEqual(output, 'd')