Changeset 1675

Timestamp:

09/14/08 23:01:45 (3 months ago)

Author:

nick

Message:

Adding tests for amf0

Files:

• pyamf/branches/slots-347/pyamf/tests/test_amf0.py (modified) (2 diffs)

pyamf/branches/slots-347/pyamf/tests/test_amf0.py

@@ -400 +400 @@
             '\x00\x0bfamily_name\x02\x00\x03Doe\x00\ngiven_name\x02\x00\x04'
             'Jane\x00\x00\t')
+    def test_slots(self):
+        class Person(object):
+            __slots__ = ('family_name', 'given_name')
+
+        u = Person()
+        u.family_name = 'Doe'
+        u.given_name = 'Jane'
+
+        self.encoder.writeElement(u)
+
+        self.assertEquals(self.buf.getvalue(), '\x10\x00\x10spam.eggs.Person'
+            '\x00\x0bfamily_name\x02\x00\x03Doe\x00\ngiven_name\x02\x00\x04'
+            'Jane\x00\x00\t')
+
 
     def test_getstate(self):
@@ -737 +751 @@
         self.assertEquals(foo.foo, 'bar')
 
+    def test_slots(self):
+        class Person(object):
+            __slots__ = ('family_name', 'given_name')
+
+        pyamf.register_class(Person, 'spam.eggs.Person')
+        self.buf.write('\x10\x00\x10spam.eggs.Person\x00\x0bfamily_name\x02'
+            '\x00\x03Doe\x00\ngiven_name\x02\x00\x04Jane\x00\x00\t')
+        self.buf.seek(0)
+
+        foo = self.decoder.readElement()
+
+        self.assertTrue(isinstance(foo, Person))
+        self.assertEquals(foo.family_name, 'Doe')
+        self.assertEquals(foo.given_name, 'Jane')
+
 class HelperTestCase(unittest.TestCase):
     def test_encode(self):