78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
|
from test import test_support
|
||
|
|
import unittest
|
||
|
|
from java.io import ByteArrayInputStream, ByteArrayOutputStream, ObjectInputStream, ObjectOutputStream
|
||
|
|
from org.python.util import PythonObjectInputStream
|
||
|
|
|
||
|
|
def serialize(o, special=False):
|
||
|
|
b = ByteArrayOutputStream()
|
||
|
|
objs = ObjectOutputStream(b)
|
||
|
|
objs.writeObject(o)
|
||
|
|
if not special:
|
||
|
|
OIS = ObjectInputStream
|
||
|
|
else:
|
||
|
|
OIS = PythonObjectInputStream
|
||
|
|
objs = OIS(ByteArrayInputStream(b.toByteArray()))
|
||
|
|
return objs.readObject()
|
||
|
|
|
||
|
|
from jser2_classes import A, AJ, N, NL, NT
|
||
|
|
|
||
|
|
class TestJavaSerialisation(unittest.TestCase):
|
||
|
|
|
||
|
|
def serialize_and_check(self, obj, special=False):
|
||
|
|
obj1 = serialize(obj, special)
|
||
|
|
self.assertEqual(obj, obj1)
|
||
|
|
|
||
|
|
def test_list(self):
|
||
|
|
self.serialize_and_check([1,"a", 3.0])
|
||
|
|
|
||
|
|
def test_dict(self):
|
||
|
|
self.serialize_and_check({'a': 3.0})
|
||
|
|
|
||
|
|
def test_tuple(self):
|
||
|
|
self.serialize_and_check((1, 'a'))
|
||
|
|
|
||
|
|
def test_oldstyle(self):
|
||
|
|
self.serialize_and_check(A('x'))
|
||
|
|
|
||
|
|
def test_jsubcl(self):
|
||
|
|
self.serialize_and_check(AJ('x'), special=True)
|
||
|
|
|
||
|
|
def test_singletons(self):
|
||
|
|
for v in (None, Ellipsis):
|
||
|
|
v1 = serialize(v)
|
||
|
|
self.assert_(v is v1)
|
||
|
|
v1 = serialize((v,))[0]
|
||
|
|
self.assert_(v is v1)
|
||
|
|
|
||
|
|
def test_NotImplemented(self):
|
||
|
|
# XXX serialize(NotImplemented) is None because of __tojava__
|
||
|
|
v1 = serialize((NotImplemented,))[0]
|
||
|
|
self.assert_(v1 is NotImplemented)
|
||
|
|
|
||
|
|
def test_type(self):
|
||
|
|
list1 = serialize(list)
|
||
|
|
self.assert_(list1 is list)
|
||
|
|
list1 = serialize((list,))[0]
|
||
|
|
self.assert_(list1 is list)
|
||
|
|
|
||
|
|
def test_user_type(self):
|
||
|
|
N1 = serialize(N)
|
||
|
|
self.assert_(N1 is N)
|
||
|
|
N1 = serialize((N,))[0]
|
||
|
|
self.assert_(N1 is N)
|
||
|
|
|
||
|
|
def test_newstyle(self):
|
||
|
|
self.serialize_and_check(N('x'))
|
||
|
|
|
||
|
|
def test_newstyle_list(self):
|
||
|
|
self.serialize_and_check(NL('x',1,2,3))
|
||
|
|
|
||
|
|
def test_newstyle_tuple(self):
|
||
|
|
self.serialize_and_check(NT('x',1,2,3))
|
||
|
|
|
||
|
|
def test_main():
|
||
|
|
test_support.run_unittest(TestJavaSerialisation)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
test_main()
|