[Zope-Checkins] CVS: Zope/lib/python/ZPublisher/tests - test_xmlrpc.py:1.1.2.1

[Tres Seaver]

Update of /cvs-repository/Zope/lib/python/ZPublisher/tests
In directory cvs.zope.org:/tmp/cvs-serv1752/lib/python/ZPublisher/tests

Added Files:
      Tag: Zope-2_6-branch
	test_xmlrpc.py 
Log Message:


  - XML-RPC marshalling of class instances used the instance 
    __dict__ to marshal the object, and could include attributes 
    prefixed with an underscore name. These attributes are considered 
    private in Zope and should generally not be disclosed.


=== Added File Zope/lib/python/ZPublisher/tests/test_xmlrpc.py ===
import unittest

class FauxResponse:

    def __init__(self):
        self._headers = {}
        self._body = None

    def setBody(self, body):
        self._body = body

    def setHeader(self, name, value):
        self._headers[name] = value

class FauxInstance:
    def __init__(self, **kw):
        self.__dict__.update(kw)

class XMLRPCResponseTests(unittest.TestCase):

    def _getTargetClass(self):
        from ZPublisher.xmlrpc import Response
        return Response

    def _makeOne(self, *args, **kw):
        return self._getTargetClass()(*args, **kw)

    def test_setBody(self):

        import xmlrpclib

        body = FauxInstance(_secret='abc', public='def')
        faux = FauxResponse()
        response = self._makeOne(faux)

        response.setBody(body)

        body_str = faux._body
        self.assertEqual(type(body_str), type(''))

        as_set, method = xmlrpclib.loads(body_str)
        as_set = as_set[0]

        self.assertEqual(method, None)
        self.failIf('_secret' in as_set.keys())
        self.failUnless('public' in as_set.keys())
        self.assertEqual(as_set['public'], 'def')

def test_suite():
    return unittest.TestSuite((unittest.makeSuite(XMLRPCResponseTests),))

if __name__ == '__main__':
    unittest.main(defaultTest='test_suite')