Changeset 1100 for pyamf/branches/logging-173/pyamf/tests/test_amf0.py

Timestamp:

03/01/08 21:38:44 (10 months ago)

Author:

nick

Message:

Merging forward source:pyamf/trunk r1088:r1099

Files:

• pyamf/branches/logging-173/pyamf/tests/test_amf0.py (modified) (3 diffs)

pyamf/branches/logging-173/pyamf/tests/test_amf0.py

@@ -163 +163 @@
         self._run(data)
 
+    def test_bytestring(self):
+        class UnicodeObject:
+            def __unicode__(self):
+                return u'Mötley CrÃŒe'
+
+        class StrObject:
+            def __str__(self):
+                return u'Mötley CrÃŒe'
+
+        class ReprObject:
+            def __repr__(self):
+                return u'Mötley CrÃŒe'
+
+        self.encoder.writeString(UnicodeObject())
+        self.assertEquals(self.buf.getvalue(), '\x02\x00\x15M\xc3\x83\xc3\x82'
+            '\xc2\xb6tley Cr\xc3\x83\xc3\x82\xc2\xbce')
+        self.buf.truncate()
+
+        self.encoder.writeString(StrObject())
+        self.assertEquals(self.buf.getvalue(), '\x02\x00\x15M\xc3\x83\xc3\x82'
+            '\xc2\xb6tley Cr\xc3\x83\xc3\x82\xc2\xbce')
+        self.buf.truncate()
+
+        self.encoder.writeString(ReprObject())
+        self.assertEquals(self.buf.getvalue(), '\x02\x00\x15M\xc3\x83\xc3\x82'
+            '\xc2\xb6tley Cr\xc3\x83\xc3\x82\xc2\xbce')
+        self.buf.truncate()
+
+        self.encoder.writeString('M\xc3\x83\xc3\x82\xc2\xb6tley Cr\xc3\x83\xc3\x82\xc2\xbce')
+        self.assertEquals(self.buf.getvalue(), '\x02\x00\x15M\xc3\x83\xc3\x82'
+            '\xc2\xb6tley Cr\xc3\x83\xc3\x82\xc2\xbce')
+
     def test_null(self):
         self._run([(None, '\x05')])
@@ -380 +412 @@
 
         pyamf.unregister_class(Person)
+
+    def test_getstate(self):
+        tc = self
+        tc.executed = False
+
+        class Foo(object):
+            def __getstate__(self):
+                tc.executed = True
+                return {'spam': 'hello', 'eggs': True}
+
+        pyamf.register_class(Foo, 'foo')
+
+        foo = Foo()
+        self.encoder.writeElement(foo)
+
+        self.assertEquals(self.buf.getvalue(), '\x10\x00\x03\x66\x6f\x6f\x00'
+            '\x04\x65\x67\x67\x73\x01\x01\x00\x04\x73\x70\x61\x6d\x02\x00\x05'
+            '\x68\x65\x6c\x6c\x6f\x00\x00\x09')
+        self.assertTrue(self.executed)
+
+        pyamf.unregister_class(Foo)
 
 class DecoderTestCase(unittest.TestCase):
@@ -612 +665 @@
         pyamf.unregister_class(Foo)
 
+    def test_setstate_newstyle(self):
+        self.executed = False
+
+        class Foo(object):
+            tc = self
+
+            def __init__(self, *args, **kwargs):
+                self.tc.fail("__init__ called")
+
+            def __setstate__(self, state):
+                self.tc.executed = True
+                self.__dict__.update(state)
+
+        pyamf.register_class(Foo, 'foo')
+
+        self.buf.write('\x10\x00\x03\x66\x6f\x6f\x00\x04\x65\x67\x67\x73\x01\x01\x00\x04')
+        self.buf.write('\x73\x70\x61\x6d\x02\x00\x05\x68\x65\x6c\x6c\x6f\x00\x00\x09')
+        self.buf.seek(0)
+
+        foo = self.decoder.readElement()
+
+        self.assertEquals(foo.spam, 'hello')
+        self.assertEquals(foo.eggs, True)
+        self.assertTrue(self.executed)
+
+        pyamf.unregister_class(Foo)
+
+    def test_setstate_classic(self):
+        self.executed = False
+
+        class Foo:
+            tc = self
+
+            def __init__(self, *args, **kwargs):
+                self.tc.fail("__init__ called")
+
+            def __setstate__(self, state):
+                self.tc.executed = True
+                self.__dict__.update(state)
+
+        pyamf.register_class(Foo, 'foo')
+
+        self.buf.write('\x10\x00\x03\x66\x6f\x6f\x00\x04\x65\x67\x67\x73\x01\x01\x00\x04')
+        self.buf.write('\x73\x70\x61\x6d\x02\x00\x05\x68\x65\x6c\x6c\x6f\x00\x00\x09')
+        self.buf.seek(0)
+
+        foo = self.decoder.readElement()
+
+        self.assertTrue(self.executed)
+        self.assertTrue(isinstance(foo, Foo))
+        self.assertEquals(foo.spam, 'hello')
+        self.assertEquals(foo.eggs, True)
+
+        pyamf.unregister_class(Foo)
+
 class HelperTestCase(unittest.TestCase):
     def test_encode(self):