Changeset 7080 for main/waeup.sirp/trunk/src/waeup/sirp/utils/tests/test_helpers.py

Timestamp:

11 Nov 2011, 10:02:30 (13 years ago)

Author:

uli

Message:

Add tests for new helper.

File:

• main/waeup.sirp/trunk/src/waeup/sirp/utils/tests/test_helpers.py (modified) (3 diffs)

main/waeup.sirp/trunk/src/waeup/sirp/utils/tests/test_helpers.py

@@ -27 +27 @@
 import unittest
 import doctest
+from cStringIO import StringIO
 from zope.security.testing import Principal, Participation
 from zope.security.management import newInteraction, endInteraction
@@ -219 +220 @@
         f2 = open(p2, 'wb').write('Ho!')
         assert helpers.cmp_files(open(p1, 'r'), open(p2, 'r')) is False
+
+class FileSizeTestCase(unittest.TestCase):
+
+    def setUp(self):
+        self.workdir = tempfile.mkdtemp()
+
+    def tearDown(self):
+        shutil.rmtree(self.workdir)
+
+    def test_real_file(self):
+        # we can get the size of real files
+        path = os.path.join(self.workdir, 'sample.txt')
+        open(path, 'wb').write('My content')
+        self.assertEqual(
+            int(helpers.file_size(open(path, 'rb'))), 10)
+        return
+
+    def test_stringio_file(self):
+        # we can get the size of file-like objects
+        self.assertEqual(
+            helpers.file_size(StringIO('my sample content')), 17)
 
 def test_suite():
@@ -230 +252 @@
         CurrentPrincipalTestCase,
         CmpFilesTestCase,
+        FileSizeTestCase,
         ]:
         suite.addTests(