Add test for encoding amf3 booleans

hydralabs · Aug 1, 2010 · 568f854 · 568f854
1 parent 26ef90a
commit 568f854
Show file tree

Hide file tree

Showing 6 changed files with 676 additions and 600 deletions.
diff --git a/cpyamf/amf0.c b/cpyamf/amf0.c
diff --git a/cpyamf/amf3.c b/cpyamf/amf3.c
diff --git a/cpyamf/amf3.pyx b/cpyamf/amf3.pyx
@@ -687,6 +687,12 @@ cdef class Encoder(codec.Encoder):
     cdef inline int writeUndefined(self, n) except -1:
         return self.writeType(TYPE_UNDEFINED)
 
+    cdef inline int writeBoolean(self, n) except -1:
+        if n is True:
+            return self.writeType(TYPE_BOOL_TRUE)
+
+        return self.writeType(TYPE_BOOL_FALSE)
+
     cpdef int serialiseString(self, s) except -1:
         """
         Similar to L{writeString} but does not encode a type byte.

diff --git a/cpyamf/codec.c b/cpyamf/codec.c
diff --git a/cpyamf/util.c b/cpyamf/util.c
diff --git a/pyamf/tests/test_amf3.py b/pyamf/tests/test_amf3.py
@@ -278,6 +278,10 @@ def test_dict(self):
             '\n\x0b\x01', ('\x03c\x06\x03g', '\x03b\x06\x03f', '\x03a\x06\x03e',
             '\x03d\x06\x03h'), '\x01')
 
+    def test_boolean(self):
+        self.assertEncoded(True, '\x03')
+        self.assertEncoded(False, '\x02')
+
     def test_mixed_array(self):
         x = pyamf.MixedArray()
         x.update({0:u'hello', 'spam': u'eggs'})