The sunExample test now works

Author: tcalmant

tests/tests.py

@@ -70,12 +70,13 @@ def setUpClass(cls):
         subprocess.call(['mvn', 'test'], shell=True)
         os.chdir(cwd)
 
-    def read_file(self, filename):
+    def read_file(self, filename, stream=False):
         """
         Reads the content of the given file in binary mode
 
         :param filename: Name of the file to read
-        :return: File content
+        :param stream: If True, return the file stream
+        :return: File content or stream
         """
         for subfolder in ('java', ''):
             found_file = os.path.join(
@@ -85,8 +86,11 @@ def read_file(self, filename):
         else:
             raise IOError("File not found: {0}".format(filename))
 
-        with open(found_file, 'rb') as filep:
-            return filep.read()
+        if stream:
+            return open(found_file, "rb")
+        else:
+            with open(found_file, 'rb') as filep:
+                return filep.read()
 
     def test_char_rw(self):
         """
@@ -271,18 +275,18 @@ def test_enums(self):
     #     # TODO: add some tests
     #     self.assertEqual(classdesc.name, "MyExceptionWhenDumping")
 
-    # def test_sun_example(self):
-    #    marshaller = javaobj.JavaObjectUnmarshaller(
-    #       open("sunExample.ser", "rb"))
-    #    pobj = marshaller.readObject()
-    #
-    #    self.assertEqual(pobj.value, 17)
-    #    self.assertTrue(pobj.next)
-    #
-    #    pobj = marshaller.readObject()
-    #
-    #    self.assertEqual(pobj.value, 19)
-    #    self.assertFalse(pobj.next)
+    def test_sun_example(self):
+       marshaller = javaobj.JavaObjectUnmarshaller(
+           self.read_file("sunExample.ser", stream=True))
+       pobj = marshaller.readObject()
+
+       self.assertEqual(pobj.value, 17)
+       self.assertTrue(pobj.next)
+
+       pobj = marshaller.readObject()
+
+       self.assertEqual(pobj.value, 19)
+       self.assertFalse(pobj.next)
 
     def test_collections(self):
         jobj = self.read_file("objCollections.ser")