Files
tryton/trytond/tests/test_protocols.py
T
Cédric Krier 8070360fd8 Use biginteger for integer overflowing XML-RPC integer dumper
issue10732
review363891002
(grafted from 63c5ffec46df34be952103f289bd51b35480112f)

--HG--
branch : 6.0
2021-09-19 00:04:57 +02:00

121 lines
3.5 KiB
Python

# This file is part of Tryton. The COPYRIGHT file at the top level of
# this repository contains the full copyright notices and license terms.
import unittest
import json
import datetime
from decimal import Decimal
from trytond.model.fields.dict import ImmutableDict
from trytond.protocols.jsonrpc import JSONEncoder, JSONDecoder, JSONRequest
from trytond.protocols.xmlrpc import client, XMLRequest
class JSONTestCase(unittest.TestCase):
'Test JSON'
def test_json_request(self):
req = JSONRequest.from_values(
data=b'{"method": "method", "params": ["foo", "bar"]}',
content_type='text/json',
)
self.assertEqual(req.parsed_data,
{'method': 'method', 'params': ['foo', 'bar']})
self.assertEqual(req.rpc_method, 'method')
self.assertEqual(req.rpc_params, ['foo', 'bar'])
def dumps_loads(self, value):
self.assertEqual(json.loads(
json.dumps(value, cls=JSONEncoder),
object_hook=JSONDecoder()), value)
def test_datetime(self):
'Test datetime'
self.dumps_loads(datetime.datetime.now())
def test_date(self):
'Test date'
self.dumps_loads(datetime.date.today())
def test_time(self):
'Test time'
self.dumps_loads(datetime.datetime.now().time())
def test_bytes(self):
'Test bytes'
self.dumps_loads(bytes(b'foo'))
self.dumps_loads(bytearray(b'foo'))
def test_decimal(self):
'Test Decimal'
self.dumps_loads(Decimal('3.141592653589793'))
def test_immutable_dict(self):
"Test ImmutableDict"
self.dumps_loads(ImmutableDict(foo='bar'))
class XMLTestCase(unittest.TestCase):
'Test XML'
def test_xml_request(self):
req = XMLRequest.from_values(
data=b"""<?xml version='1.0'?>
<methodCall>
<methodName>method</methodName>
<params>
<param>
<value><string>foo</string></value>
</param>
<param>
<value><string>bar</string></value>
</param>
</params>
</methodCall>""",
content_type='text/xml')
self.assertEqual(req.parsed_data, (('foo', 'bar'), 'method'))
self.assertEqual(req.rpc_method, 'method')
self.assertEqual(req.rpc_params, ('foo', 'bar'))
def dumps_loads(self, value):
s = client.dumps((value,), allow_none=True)
result, _ = client.loads(s)
result, = result
self.assertEqual(value, result)
def test_decimal(self):
'Test Decimal'
self.dumps_loads(Decimal('3.141592653589793'))
def test_bytes(self):
'Test bytes'
self.dumps_loads(bytes(b'foo'))
self.dumps_loads(bytearray(b'foo'))
def test_date(self):
'Test date'
self.dumps_loads(datetime.date.today())
def test_time(self):
'Test time'
self.dumps_loads(datetime.datetime.now().time())
def test_biginteger(self):
"Test BigInteger"
self.dumps_loads(client.MAXINT + 1)
def test_immutable_dict(self):
"Test ImmutableDict"
self.dumps_loads(ImmutableDict(foo='bar'))
def test_none(self):
'Test None'
self.dumps_loads(None)
def suite():
suite_ = unittest.TestSuite()
suite_.addTests(unittest.TestLoader().loadTestsFromTestCase(JSONTestCase))
suite_.addTests(unittest.TestLoader().loadTestsFromTestCase(XMLTestCase))
return suite_